source4/setup/upgrade.py

   1 #!/usr/bin/python
   2 #
   3 #       Upgrade from Samba3
   4 #       Copyright Jelmer Vernooij 2005-2007
   5 #       Released under the GNU GPL v3 or later
   6 #
   7 import getopt
   8 import optparse
   9 import os, sys
  10 sys.path.append("scripting/python")
  11 import param
  12 import samba
  13 import samba.getopt as options
  14 from auth import system_session
  15
  16 parser = optparse.OptionParser("upgrade [options] <libdir> <smbconf>")
  17 parser.add_option_group(options.SambaOptions(parser))
  18 parser.add_option_group(options.VersionOptions(parser))
  19 credopts = options.CredentialsOptions(parser)
  20 parser.add_option_group(credopts)
  21 parser.add_option("--setupdir", type="string", metavar="DIR",
  22                 help="directory with setup files")
  23 parser.add_option("--realm", type="string", metavar="REALM", help="set realm")
  24 parser.add_option("--quiet", help="Be quiet")
  25 parser.add_option("--blank",
  26                 help="do not add users or groups, just the structure")
  27 parser.add_option("--targetdir", type="string", metavar="DIR",
  28                           help="Set target directory")
  29
  30 opts, args = parser.parse_args()
  31
  32 def message(text):
  33     """Print a message if quiet is not set."""
  34     if opts.quiet:
  35         print text
  36
  37 if len(args) < 1:
  38     parser.print_usage()
  39     sys.exit(1)
  40 from samba.samba3 import Samba3
  41 message("Reading Samba3 databases and smb.conf\n")
  42 libdir = args[0]
  43 if not os.path.isdir(libdir):
  44     print "error: %s is not a directory"
  45     sys.exit(1)
  46 if len(args) > 1:
  47     smbconf = args[1]
  48 else:
  49     smbconf = os.path.join(libdir, "smb.conf")
  50 samba3 = Samba3(libdir, smbconf)
  51
  52 from samba.upgrade import upgrade_provision
  53 from samba.provision import provision_paths_from_lp
  54
  55 message("Provisioning\n")
  56
  57 setup_dir = opts.setupdir
  58 if setup_dir is None:
  59         setup_dir = "setup"
  60
  61 creds = credopts.get_credentials()
  62 lp = param.LoadParm()
  63 if opts.configfile:
  64     lp.load(opts.configfile)
  65 if opts.targetdir is not None:
  66     if not os.path.exists(opts.targetdir):
  67         os.mkdir(opts.targetdir)
  68     lp.set("private dir", os.path.abspath(opts.targetdir))
  69     lp.set("lock dir", os.path.abspath(opts.targetdir))
  70 paths = provision_paths_from_lp(lp, "")
  71 paths.smbconf = opts.configfile
  72 upgrade_provision(samba3, setup_dir, message, credentials=creds, session_info=system_session(),
  73                   lp=lp, paths=paths)