Fix and test python scripts and kerberos

[samba.git] / source4 / setup / upgrade.py

diff --git a/source4/setup/upgrade.py b/source4/setup/upgrade.py
index 96584a1026d306896c7df3dfa32740fdc83e5c12..3bcc57ab64770169ca7d13b7fd2ee5a8259d1cf4 100755 (executable)
--- a/source4/setup/upgrade.py
+++ b/source4/setup/upgrade.py
@@ -6,61 +6,60 @@
 #
 import getopt
 import optparse
-import sys
-sys.path.append("scripting/python")
+import os, sys
+import param
 import samba
-import samba.getopt
+import samba.getopt as options
+from auth import system_session
 
-parser = optparse.OptionParser("upgrade [options]")
-parser.add_option_group(options.SambaOptions(parser))
+parser = optparse.OptionParser("upgrade [options] <libdir> <smbconf>")
+sambaopts = options.SambaOptions(parser)
+parser.add_option_group(sambaopts)
 parser.add_option_group(options.VersionOptions(parser))
-parser.add_option_group(options.CredentialsOptions(parser))
+credopts = options.CredentialsOptions(parser)
+parser.add_option_group(credopts)
+parser.add_option("--setupdir", type="string", metavar="DIR", 
+               help="directory with setup files")
 parser.add_option("--realm", type="string", metavar="REALM", help="set realm")
 parser.add_option("--quiet", help="Be quiet")
-parser.add_option("--verify", help="Verify resulting configuration")
 parser.add_option("--blank", 
                help="do not add users or groups, just the structure")
 parser.add_option("--targetdir", type="string", metavar="DIR", 
                          help="Set target directory")
 
-opts = parser.parse_args()[0]
+opts, args = parser.parse_args()
 
 def message(text):
     """Print a message if quiet is not set."""
     if opts.quiet:
         print text
 
+if len(args) < 1:
+    parser.print_usage()
+    sys.exit(1)
+from samba.samba3 import Samba3
 message("Reading Samba3 databases and smb.conf\n")
-samba3 = samba3_read(options.ARGV[0], options.ARGV[1])
-
-message("Provisioning\n")
-subobj = upgrade_provision(samba3)
-if options.targetdir is not None:
-       paths = ProvisionPaths()
-       paths.smbconf = os.path.join(options.targetdir, "smb.conf")
-       ldbs = ["hklm","hkcr","hku","hkcu","hkpd","hkpt","samdb","rootdse","secrets","wins"]
-       for n in ldbs:
-               paths[n] = sprintf("tdb://%s/%s.ldb", options.targetdir, n)
-       paths.dns = os.path.join(options.targetdir, "dns.zone")
+libdir = args[0]
+if not os.path.isdir(libdir):
+    print "error: %s is not a directory"
+    sys.exit(1)
+if len(args) > 1:
+    smbconf = args[1]
 else:
-       paths = provision_default_paths(subobj)
+    smbconf = os.path.join(libdir, "smb.conf")
+samba3 = Samba3(libdir, smbconf)
 
-creds = options.get_credentials()
-system_session = system_session()
-paths = provision_default_paths(subobj)
+from samba.upgrade import upgrade_provision
 
-if options.realm:
-       subobj.realm = options.realm
+message("Provisioning\n")
 
-provision(lp, subobj, message, options.blank, paths, system_session, creds, undefined)
+setup_dir = opts.setupdir
+if setup_dir is None:
+       setup_dir = "setup"
 
-ret = upgrade(subobj,samba3,message,paths, system_session, creds)
-if ret > 0:
-       message("Failed to import %d entries\n", ret)
-else:
-       provision_dns(subobj, message, paths, system_session, creds)
-       message("All OK\n")
+lp = sambaopts.get_loadparm()
+smbconf = lp.configfile()
+creds = credopts.get_credentials(lp)
 
-if options.verify:
-       message("Verifying...\n")
-       ret = upgrade_verify(subobj, samba3, paths, message)
+upgrade_provision(samba3, setup_dir, message, credentials=creds, session_info=system_session(), 
+                  smbconf=smbconf, targetdir=opts.targetdir)