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 samba.provision import provision_default_paths
  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("--verify", help="Verify resulting configuration")
  26 parser.add_option("--blank",
  27                 help="do not add users or groups, just the structure")
  28 parser.add_option("--targetdir", type="string", metavar="DIR",
  29                           help="Set target directory")
  30
  31 opts, args = parser.parse_args()
  32
  33 def message(text):
  34     """Print a message if quiet is not set."""
  35     if opts.quiet:
  36         print text
  37
  38 if len(args) < 1:
  39     parser.print_usage()
  40     sys.exit(1)
  41 from samba.samba3 import Samba3
  42 message("Reading Samba3 databases and smb.conf\n")
  43 libdir = args[0]
  44 if not os.path.isdir(libdir):
  45     print "error: %s is not a directory"
  46     sys.exit(1)
  47 if len(args) > 1:
  48     smbconf = args[1]
  49 else:
  50     smbconf = os.path.join(libdir, "smb.conf")
  51 samba3 = Samba3(libdir, smbconf)
  52
  53 from samba.upgrade import upgrade_provision
  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 lp.load(opts.configfile)
  64 upgrade_provision(samba3, setup_dir, message, credentials=creds, session_info=system_session())
  65
  66 if opts.verify:
  67         message("Verifying...\n")
  68         ret = upgrade_verify(subobj, samba3, paths, message)