source4/setup/upgrade_from_s3

   1 #!/usr/bin/python
   2 #
   3 # Upgrade from Samba3
   4 # Copyright Jelmer Vernooij 2005-2007
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 import getopt
  20 import optparse
  21 import os, sys
  22
  23 # Find right directory when running from source tree
  24 sys.path.insert(0, "bin/python")
  25
  26 import samba
  27 import samba.getopt as options
  28 from samba import param
  29 from samba.auth import system_session
  30 from samba.provision import find_setup_dir
  31 from samba.upgrade import upgrade_provision
  32 from samba.samba3 import Samba3
  33
  34 parser = optparse.OptionParser("upgrade_from_s3 [options] <libdir> <smbconf>")
  35 sambaopts = options.SambaOptions(parser)
  36 parser.add_option_group(sambaopts)
  37 parser.add_option_group(options.VersionOptions(parser))
  38 credopts = options.CredentialsOptions(parser)
  39 parser.add_option_group(credopts)
  40 parser.add_option("--setupdir", type="string", metavar="DIR",
  41                   help="directory with setup files")
  42 parser.add_option("--realm", type="string", metavar="REALM", help="set realm")
  43 parser.add_option("--quiet", help="Be quiet")
  44 parser.add_option("--blank",
  45                   help="do not add users or groups, just the structure")
  46 parser.add_option("--targetdir", type="string", metavar="DIR",
  47                   help="Set target directory")
  48
  49 opts, args = parser.parse_args()
  50
  51 def message(text):
  52     """Print a message if quiet is not set."""
  53     if not opts.quiet:
  54         print text
  55
  56 if len(args) < 1:
  57     parser.print_usage()
  58     sys.exit(1)
  59
  60 message("Reading Samba3 databases and smb.conf")
  61
  62 libdir = args[0]
  63 if not os.path.isdir(libdir):
  64     print "error: %s is not a directory"
  65     sys.exit(1)
  66
  67 if len(args) > 1:
  68     smbconf = args[1]
  69 else:
  70     smbconf = os.path.join(libdir, "smb.conf")
  71
  72 samba3 = Samba3(libdir, smbconf)
  73
  74 message("Provisioning")
  75
  76 setup_dir = opts.setupdir
  77 if setup_dir is None:
  78     setup_dir = find_setup_dir()
  79
  80 lp = sambaopts.get_loadparm()
  81 smbconf = lp.configfile
  82 creds = credopts.get_credentials(lp)
  83
  84 upgrade_provision(samba3, setup_dir, message, credentials=creds,
  85                   session_info=system_session(), smbconf=smbconf,
  86                   targetdir=opts.targetdir)