r26370: Import upgrade script in Python.
[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 samba.options
10
11 parser = optparse.OptionParser("upgrade [options]")
12 parser.add_option_group(options.SambaOptions(parser))
13 parser.add_option_group(options.VersionOptions(parser))
14 parser.add_option_group(options.CredentialsOptions(parser))
15 parser.add_option("--realm", type="string", metavar="REALM", help="set realm")
16 parser.add_option("--quiet", help="Be quiet")
17 parser.add_option("--verify", help="Verify resulting configuration")
18 parser.add_option("--blank", 
19                 help="do not add users or groups, just the structure")
20 parser.add_option("--targetdir", type="string", metavar="DIR", 
21                           help="Set target directory")
22
23 def message(text):
24     """Print a message if quiet is not set."""
25         if opts.quiet:
26                 print text
27
28 message("Reading Samba3 databases and smb.conf\n")
29 samba3 = samba3_read(options.ARGV[0], options.ARGV[1])
30
31 message("Provisioning\n")
32 subobj = upgrade_provision(samba3)
33 if options.targetdir is not None:
34         paths = ProvisionPaths()
35         paths.smbconf = os.path.join(options.targetdir, "smb.conf")
36         ldbs = ["hklm","hkcr","hku","hkcu","hkpd","hkpt","samdb","rootdse","secrets","wins"]
37         for n in ldbs:
38                 paths[n] = sprintf("tdb://%s/%s.ldb", options.targetdir, n)
39         paths.dns = os.path.join(options.targetdir, "dns.zone")
40 else:
41         paths = provision_default_paths(subobj)
42
43 creds = options.get_credentials()
44 system_session = system_session()
45 paths = provision_default_paths(subobj)
46
47 if options.realm:
48         subobj.realm = options.realm
49
50 provision(lp, subobj, message, options.blank, paths, system_session, creds, undefined)
51
52 ret = upgrade(subobj,samba3,message,paths, system_session, creds)
53 if ret > 0:
54         message("Failed to import %d entries\n", ret)
55 else:
56         provision_dns(subobj, message, paths, system_session, creds)
57         message("All OK\n")
58
59 if options.verify:
60         message("Verifying...\n")
61         ret = upgrade_verify(subobj, samba3, paths, message)