estimate_capacity.py: filter by patch project or blamelist

scripts/tools/estimate_capacity.py

 #!/usr/bin/env python
 # Copyright 2014 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.
 
 """
 Estimates capacity needs for all builders of a given master.
 """
 
 import argparse
@@ skipping 64 matching lines @@
   return {
     'hourly_bots': min_bots(hourly_buckets, 3600),
     'daily_bots': min_bots(daily_buckets, 3600 * 24),
   }
 
 
 def main(argv):
   parser = argparse.ArgumentParser()
   parser.add_argument('master')
   parser.add_argument('--days', type=int, default=14)
-  parser.add_argument('--ignore-cls-by', action='append')
+  parser.add_argument('--filter-by-blamelist')
+  parser.add_argument('--filter-by-patch-project')
 
   args = parser.parse_args(argv)
 
   with tempfile.NamedTemporaryFile() as f:
     subprocess.check_call([
         os.path.join(BASE_DIR, 'scripts', 'tools', 'runit.py'),
         os.path.join(BASE_DIR, 'scripts', 'tools', 'dump_master_cfg.py'),
         'masters/%s' % args.master,
         f.name])
     master_config = json.load(f)
@@ skipping 23 matching lines @@
     }
     # TODO(phajdan.jr): use multiprocessing pool to speed this up.
     for builddir, builders in pool['builders'].iteritems():
       print '  builddir "%s":' % builddir
       for builder in builders:
         raw_builds = get_builds(
             args.master.replace('master.', ''), builder, days)
 
         builds = []
         for build in raw_builds:
+          properties = {p[0]: p[1] for p in build.get('properties', [])}
+          if (args.filter_by_patch_project and
+              properties.get('patch_project') != args.filter_by_patch_project):
+            continue
+
           blamelist = build.get('blame', [])
-          ignore_cl = False
-          for entry in blamelist:
-            if entry in args.ignore_cls_by:
-              ignore_cl = True
-              break
-          if not ignore_cl:
-            builds.append(build)
+          if (args.filter_by_blamelist and
+              args.filter_by_blamelist not in blamelist):
+            continue
 
         capacity = estimate_capacity(builds)
         for key in ('hourly_bots', 'daily_bots'):
           pool_capacity[key] += capacity[key]
         print '    %-45s %5.1f %5.1f' % (
             builder,
             capacity['daily_bots'],
             capacity['hourly_bots'])
     print '  %-45s   %5.1f %5.1f %5.1f' % (
         'total',
         pool_capacity['daily_bots'],
         pool_capacity['hourly_bots'],
         len(pool['slave_set']))
 
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv[1:]))