| import sys |
| import argparse |
| from twisted.internet import defer, reactor |
| from twisted.python import log |
| from slavealloc import exceptions |
| |
| # subcommands |
| from slavealloc.scripts import dbinit, gettac, lock, disable, enable, dbdump, dbimport, notes |
| subcommands = [dbinit, gettac, lock, disable, enable, dbdump, dbimport, notes] |
| |
| |
| def parse_options(): |
| parser = argparse.ArgumentParser(description="Runs slavealloc subcommands") |
| parser.set_defaults(_module=None) |
| |
| parser.add_argument('-A', '--api', dest='apiurl', |
| default='http://slavealloc.build.mozilla.org/api', |
| help="""URL of the REST API to use for most subcommands""") |
| |
| subparsers = parser.add_subparsers(title='subcommands') |
| |
| for module in subcommands: |
| subparser = module.setup_argparse(subparsers) |
| subparser.set_defaults(module=module, subparser=subparser) |
| |
| # parse the args |
| args = parser.parse_args() |
| |
| # make sure we got a subcommand |
| if not args.module: |
| parser.error("No subcommand specified") |
| |
| # let it process its own args |
| args.module.process_args(args.subparser, args) |
| |
| # and return the results |
| return args.module.main, args |
| |
| |
| def main(): |
| errors = [] |
| |
| func, args = parse_options() |
| |
| def do_command(): |
| d = defer.maybeDeferred(func, args) |
| |
| # catch command-line errors and don't show the traceback |
| def cmdline_error(f): |
| f.trap(exceptions.CmdlineError) |
| errors.append(str(f.value)) |
| d.addErrback(cmdline_error) |
| |
| # but catch everything else.. |
| d.addErrback(log.err, "while executing subcommand") |
| |
| # before unconditionally stopping the reactor |
| d.addBoth(lambda _: reactor.stop()) |
| reactor.callWhenRunning(do_command) |
| reactor.run() |
| |
| # handle any errors after the reactor is done |
| if errors: |
| for error in errors: |
| print >>sys.stderr, error |
| sys.exit(1) |
