git cl try: load trybot list from cq.cfg by default.

commit_queue.py

 #!/usr/bin/env python
 # Copyright (c) 2011 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Access the commit queue from the command line.
 """
 
 __version__ = '0.1'
 
@@ skipping 86 matching lines @@
     return 0
   _apply_on_issue(_get_commit, obj, issue)
 
 def set_commit(obj, issue, flag):
   """Sets the commit bit flag on an issue."""
   def _set_commit(properties):
     print obj.set_flag(issue, properties['patchsets'][-1], 'commit', flag)
     return 0
   _apply_on_issue(_set_commit, obj, issue)
 
+
+def get_master_builder_map(config_path):
+  """Returns a map of master -> [builders] from cq config."""
+  with open(config_path) as config_file:
+    cq_config = config_file.read()
+
+  config = cq_pb2.Config()
+  text_format.Merge(cq_config, config)
+  masters = {}
+  if config.HasField('verifiers') and config.verifiers.HasField('try_job'):
+    for bucket in config.verifiers.try_job.buckets:
+      masters.setdefault(bucket.name, [])
+      for builder in bucket.builders:
+        if not builder.HasField('experiment_percentage'):
+          masters[bucket.name].append(builder.name)
+  return masters
+
+
 @need_issue
 def CMDset(parser, args):
   """Sets the commit bit."""
   options, args, obj = parser.parse_args(args)
   if args:
     parser.error('Unrecognized args: %s' % ' '.join(args))
   return set_commit(obj, options.issue, '1')
 
 @need_issue
 def CMDget(parser, args):
@@ skipping 23 matching lines @@
       ],
       'another_master': [
         'third_builder'
       ]
     }
   """
   _, args = parser.parse_args(args)
   if len(args) != 1:
     parser.error('Expected a single path to CQ config. Got: %s' %
                  ' '.join(args))
-
-  with open(args[0]) as config_file:
-    cq_config = config_file.read()
-
-  config = cq_pb2.Config()
-  text_format.Merge(cq_config, config)
-  masters = {}
-  if config.HasField('verifiers') and config.verifiers.HasField('try_job'):
-    for bucket in config.verifiers.try_job.buckets:
-      masters.setdefault(bucket.name, [])
-      for builder in bucket.builders:
-        if not builder.HasField('experiment_percentage'):
-          masters[bucket.name].append(builder.name)
-  print json.dumps(masters)
+  print json.dumps(get_master_builder_map(args[0]))
 
 CMDbuilders.func_usage_more = '<path-to-cq-config>'
 
 
 def CMDvalidate(parser, args):
   """Validates a CQ config.
 
   Takes a single argument - path to the CQ config to be validated. Returns 0 on
   valid config, non-zero on invalid config. Errors and warnings are printed to
   screen.
@@ skipping 89 matching lines @@
   return CMDhelp(parser, args)
 
 
 if __name__ == "__main__":
   fix_encoding.fix_encoding()
   try:
     sys.exit(main())
   except KeyboardInterrupt:
     sys.stderr.write('interrupted\n')
     sys.exit(1)