Fix and test python scripts and kerberos
[ab/samba.git/.git] / 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)