Index: infra/tools/cros_pin/__main__.py |
diff --git a/infra/tools/restart/__main__.py b/infra/tools/cros_pin/__main__.py |
similarity index 51% |
copy from infra/tools/restart/__main__.py |
copy to infra/tools/cros_pin/__main__.py |
index 3afc7619b7d1b5413dd122578dae25eee69044b3..90275bf78811507c688b45eb529a863f1349327e 100755 |
--- a/infra/tools/restart/__main__.py |
+++ b/infra/tools/cros_pin/__main__.py |
@@ -2,36 +2,30 @@ |
# Copyright 2014 Google Inc. All Rights Reserved. |
# pylint: disable=F0401 |
-"""Restart a master via master-manager.""" |
+"""CrOS pin management/update tool.""" |
import argparse |
import datetime |
import sys |
- |
import infra_libs.logs |
- |
-from infra.tools.restart import restart |
+from infra.tools.cros_pin import cros_pin |
def main(argv): |
parser = argparse.ArgumentParser( |
- prog='restart', |
+ prog='cros-pin', |
description=sys.modules['__main__'].__doc__) |
infra_libs.logs.add_argparse_options(parser) |
- restart.add_argparse_options(parser) |
+ cros_pin.add_argparse_options(parser) |
+ |
args = parser.parse_args(argv) |
infra_libs.logs.process_argparse_options(args) |
- if args.minutes_in_future < 0: |
- parser.error('minutes-in-future must be nonnegative, use 0 for "now"') |
- |
- delta = datetime.timedelta(minutes=args.minutes_in_future) |
- |
- return restart.run(args.masters, delta, args.reviewer, args.bug, args.force, |
- args.no_commit) |
+ # Execute our subcommand (configured by cros_pin.add_argparse_options). |
+ return args.func(args) |
if __name__ == '__main__': |