4cf9641ef2aca3f0164b08639fa2b41c45a1b330
[ira/wip.git] / source / 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 sambaopts = options.SambaOptions(parser)
18 parser.add_option_group(sambaopts)
19 parser.add_option_group(options.VersionOptions(parser))
20 credopts = options.CredentialsOptions(parser)
21 parser.add_option_group(credopts)
22 parser.add_option("--setupdir", type="string", metavar="DIR", 
23                 help="directory with setup files")
24 parser.add_option("--realm", type="string", metavar="REALM", help="set realm")
25 parser.add_option("--quiet", help="Be quiet")
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 from samba.provision import provision_paths_from_lp
55
56 message("Provisioning\n")
57
58 setup_dir = opts.setupdir
59 if setup_dir is None:
60         setup_dir = "setup"
61
62 creds = credopts.get_credentials()
63 lp = sambaopts.get_loadparm()
64 if opts.targetdir is not None:
65     if not os.path.exists(opts.targetdir):
66         os.mkdir(opts.targetdir)
67     lp.set("private dir", os.path.abspath(opts.targetdir))
68     lp.set("lock dir", os.path.abspath(opts.targetdir))
69 paths = provision_paths_from_lp(lp, "")
70 paths.smbconf = sambaopts.get_loadparm_path()
71 upgrade_provision(samba3, setup_dir, message, credentials=creds, session_info=system_session(), 
72                   lp=lp, paths=paths)