| Index: cit.py
|
| diff --git a/cit.py b/cit.py
|
| index b99411bc922a2b054e829fe44d7647d73dcc1f3a..f08ddddc5b057e9d86cbcc6617a32cdbdd558238 100755
|
| --- a/cit.py
|
| +++ b/cit.py
|
| @@ -99,27 +99,7 @@ def get_available_tools():
|
| return (sorted(infra_tools), sorted(cipd_tools))
|
|
|
|
|
| -def run(args):
|
| - if args:
|
| - tool_name = args[0]
|
| - # Check to see if it is a infra tool first.
|
| - infra_dir = os.path.join(
|
| - TARGET_DIR, 'infra', 'infra', 'tools', *tool_name.split('.'))
|
| - cipd_file = os.path.join(TARGET_DIR, 'infra', 'cipd', tool_name)
|
| - if sys.platform.startswith('win'):
|
| - cipd_file += '.exe'
|
| - if (os.path.isdir(infra_dir)
|
| - and os.path.isfile(os.path.join(infra_dir, '__main__.py'))):
|
| - cmd = [
|
| - sys.executable, os.path.join(TARGET_DIR, 'infra', 'run.py'),
|
| - 'infra.tools.%s' % tool_name]
|
| - elif os.path.isfile(cipd_file) and is_exe(cipd_file):
|
| - cmd = [cipd_file]
|
| -
|
| - # Add the remaining arguments.
|
| - cmd.extend(args[1:])
|
| - return subprocess.call(cmd)
|
| -
|
| +def usage():
|
| infra_tools, cipd_tools = get_available_tools()
|
| print """usage: cit.py <name of tool> [args for tool]
|
|
|
| @@ -137,6 +117,32 @@ def run(args):
|
| print ' * %s' % tool
|
|
|
|
|
| +def run(args):
|
| + if not args:
|
| + return usage()
|
| +
|
| + tool_name = args[0]
|
| + # Check to see if it is a infra tool first.
|
| + infra_dir = os.path.join(
|
| + TARGET_DIR, 'infra', 'infra', 'tools', *tool_name.split('.'))
|
| + cipd_file = os.path.join(TARGET_DIR, 'infra', 'cipd', tool_name)
|
| + if sys.platform.startswith('win'):
|
| + cipd_file += '.exe'
|
| + if (os.path.isdir(infra_dir)
|
| + and os.path.isfile(os.path.join(infra_dir, '__main__.py'))):
|
| + cmd = [
|
| + sys.executable, os.path.join(TARGET_DIR, 'infra', 'run.py'),
|
| + 'infra.tools.%s' % tool_name]
|
| + elif os.path.isfile(cipd_file) and is_exe(cipd_file):
|
| + cmd = [cipd_file]
|
| + else:
|
| + print >>sys.stderr, 'Unknown tool "%s"' % tool_name
|
| + return usage()
|
| +
|
| + # Add the remaining arguments.
|
| + cmd.extend(args[1:])
|
| + return subprocess.call(cmd)
|
| +
|
|
|
| def main():
|
| parser = argparse.ArgumentParser("Chrome Infrastructure CLI.")
|
|
|