Update CL description by adding CQ_EXTRA_TRYBOTS for Telemetry changes.

tools/perf/PRESUBMIT.py

 # Copyright 2012 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.
 
 """Presubmit script for changes affecting tools/perf/.
 
 See http://dev.chromium.org/developers/how-tos/depottools/presubmit-scripts
 for more details about the presubmit API built into depot_tools.
 """
 
 import os
+import re
 import sys
 
 PYLINT_BLACKLIST = []
 PYLINT_DISABLED_WARNINGS = [
     'R0923',  # Interface not implemented
     'R0201',  # Method could be a function
     'E1101',  # Non-existent member is accessed.
 ]
 
 
@@ skipping 56 matching lines @@
 def CheckChangeOnUpload(input_api, output_api):
   report = []
   report.extend(_CommonChecks(input_api, output_api))
   return report
 
 
 def CheckChangeOnCommit(input_api, output_api):
   report = []
   report.extend(_CommonChecks(input_api, output_api))
   return report
+
+
+def _IsBenchmarksModified(change_list):

[qyearsley, 2015/04/08 00:07:13]

Should the argument be called change or change_list? It's called change below.

[prasadv, 2015/04/08 00:15:48]

Done.

+  """Checks whether CL contains any modification to Telemetry benchmarks."""
+  for affected_file in change_list.AffectedFiles():
+    affected_file_path = affected_file.LocalPath()
+    file_path, _ = os.path.splitext(affected_file_path)
+    if (os.path.join('tools', 'perf', 'benchmarks') in file_path or
+        os.path.join('tools', 'perf', 'measurements') in file_path):
+      return True
+  return False
+
+
+def PostUploadHook(cl, change, output_api):
+  """git cl upload will call this hook after the issue is created/modified.
+
+  This hook adds extra try bots list to the CL description in order to run
+  Telemetry benchmarks on Perf trybots in addtion to CQ trybots if the CL
+  contains any changes to Telemetry benchmarks.
+  """
+  benchmarks_modified = _IsBenchmarksModified(change)
+  rietveld_obj = cl.RpcServer()
+  issue = cl.issue
+  original_description = rietveld_obj.get_description(issue)
+  if not benchmarks_modified or re.search(
+     r'^CQ_EXTRA_TRYBOTS=.*', original_description, re.M | re.I):
+    return []
+
+  results = []
+  bots = [
+    'linux_perf_bisect',
+    'mac_perf_bisect',
+    'win_perf_bisect',
+    'android_nexus5_perf_bisect'
+  ]
+  bots = ['tryserver.chromium.perf:%s' % s for s in bots]
+  bots_string = ';'.join(bots)
+  description = original_description
+  description += '\nCQ_EXTRA_TRYBOTS=%s' % bots_string
+  results.append(output_api.PresubmitNotifyResult(
+      'Automatically added Perf trybots to run Telemetry benchmarks on CQ.'))
+
+  if description != original_description:
+   rietveld_obj.update_description(issue, description)
+
+  return results