Added tools to retrieve CQ builders from a CQ config

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'
 
 import functools
+import json
 import logging
 import optparse
 import os
 import sys
 import urllib2
 
 import breakpad  # pylint: disable=W0611
 
 import auth
 import fix_encoding
 import rietveld
 
+THIRD_PARTY_DIR = os.path.join(os.path.dirname(__file__), 'third_party')
+sys.path.append(THIRD_PARTY_DIR)
+
+from cq_client import cq_pb2
+from google.protobuf import text_format
 
 def usage(more):
   def hook(fn):
     fn.func_usage_more = more
     return fn
   return hook
 
 
 def need_issue(fn):
   """Post-parse args to create a Rietveld object."""
@@ skipping 61 matching lines @@
 
 @need_issue
 def CMDclear(parser, args):
   """Clears 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, '0')
 
 
+def CMDbuilders(parser, args):
+  """Prints json-formatted list of builders given a path to cq.cfg file.
+
+  The output is a dictionary in the following format:
+    {
+      'master_name': {
+        'builder_name': {
+          'custom_property': 'value',
+          'testfilter': 'compile'
+        },
+        'another_builder': {}
+      },
+      '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].setdefault(builder.name, {})
+          for prop in builder.properties:
+            masters[bucket.name][builder.name][prop.name] = prop.value
+  print json.dumps(masters)
+
+CMDbuilders.func_usage_more = '<path-to-cq-config>'
+
 ###############################################################################
 ## Boilerplate code
 
 
 class OptionParser(optparse.OptionParser):
   """An OptionParser instance with default options.
 
   It should be then processed with gen_usage() before being used.
   """
   def __init__(self, *args, **kwargs):
@@ skipping 67 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)