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 import param
  11 import samba
  12 import samba.getopt as options
  13 from auth import system_session
  14
  15 parser = optparse.OptionParser("upgrade [options] <libdir> <smbconf>")
  16 sambaopts = options.SambaOptions(parser)
  17 parser.add_option_group(sambaopts)
  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
  54 message("Provisioning\n")
  55
  56 setup_dir = opts.setupdir
  57 if setup_dir is None:
  58         setup_dir = "setup"
  59
  60 lp = sambaopts.get_loadparm()
  61 smbconf = lp.configfile()
  62 creds = credopts.get_credentials(lp)
  63
  64 upgrade_provision(samba3, setup_dir, message, credentials=creds, session_info=system_session(),
  65                   smbconf=smbconf, targetdir=opts.targetdir)