12 from wizard import deploy, util, shell, sset, command
14 def main(argv, baton):
15 options, args = parse_args(argv, baton)
17 base_args = calculate_base_args(options)
18 sh = shell.ParallelShell.make(options.no_parallelize, options.max_processes)
19 seen = sset.make(options.seen)
20 is_root = not os.getuid()
21 warnings_log, errors_log, merge_log = command.open_logs(options.log_dir, ('warnings', 'errors', 'merge'))
25 merge_fails = [0] # otherwise I get a UnboundLocalError later on when I increment
26 deploys = deploy.parse_install_lines(app, options.versions_path)
27 requested_deploys = itertools.islice(deploys, options.limit)
28 for i, d in enumerate(requested_deploys, 1):
29 # check if we want to punt due to --limit
30 if d.location in seen:
32 if is_root and not command.security_check_homedir(d.location):
34 # XXX: we may be able to punt based on detected versions from d
35 logging.info("Processing %s" % d.location)
36 child_args = list(base_args)
37 # calculate the log file, if a log dir was specified
39 log_file = command.calculate_log_name(options.log_dir, i, d.location)
40 child_args.append("--log-file=" + log_file)
42 def make_on_pair(d, i):
43 # we need to make another stack frame so that d and i get specific bindings.
44 def on_success(stdout, stderr):
46 warnings_log.write("%s\n" % d.location)
47 logging.warning("Warnings [%04d] %s:\n%s" % (i, d.location, stderr))
50 if e.name == "wizard.command.upgrade.AlreadyUpgraded" or \
51 e.name == "AlreadyUpgraded":
53 logging.info("Skipped already upgraded %s" % d.location)
54 elif e.name == "MergeFailed":
56 logging.warning("Merge failed: resolve at [%s], source at [%s]" % (e.stdout.rstrip(), d.location))
57 merge_log.write("%s\n" % d.location)
61 if name not in errors: errors[name] = []
62 errors[name].append(d)
63 logging.error("%s in [%04d] %s" % (name, i, d.location))
64 errors_log.write("%s\n" % d.location)
65 return (on_success, on_error)
66 on_success, on_error = make_on_pair(d, i)
67 sh.call("wizard", "upgrade", d.location, *child_args,
68 on_success=on_success, on_error=on_error)
70 for name, deploys in errors.items():
71 logging.warning("%s from %d installs" % (name, len(deploys)))
73 logging.warning("%d out of %d installs (%.1f%%) had merge failure" % (merge_fails[0], i, float(merge_fails[0])/i*100))
75 def parse_args(argv, baton):
76 usage = """usage: %prog mass-upgrade [ARGS] APPLICATION
78 Mass upgrades an application to the latest scripts version.
79 Essentially equivalent to running '%prog upgrade' on all
80 autoinstalls for a particular application found by parallel-find,
81 but with advanced reporting.
83 This command is intended to be run as root on a server with
84 the scripts AFS patch."""
85 parser = command.WizardOptionParser(usage)
86 baton.push(parser, "log_dir")
87 baton.push(parser, "seen")
88 baton.push(parser, "no_parallelize")
89 baton.push(parser, "dry_run")
90 baton.push(parser, "max_processes")
91 baton.push(parser ,"limit")
92 baton.push(parser, "versions_path")
93 baton.push(parser, "srv_path")
94 parser.add_option("--force", dest="force", action="store_true",
95 default=False, help="Force running upgrade even if it's already at latest version.")
96 options, args, = parser.parse_all(argv)
98 parser.error("too many arguments")
100 parser.error("must specify application to upgrade")
103 def calculate_base_args(options):
104 return command.makeBaseArgs(options, dry_run="--dry-run", srv_path="--srv-path", force="--force")