[Findit] Refactor Findit pipeline.

appengine/findit/waterfall/schedule_compile_try_job_pipeline.py

Index: appengine/findit/waterfall/schedule_compile_try_job_pipeline.py
diff --git a/appengine/findit/waterfall/schedule_compile_try_job_pipeline.py b/appengine/findit/waterfall/schedule_compile_try_job_pipeline.py
new file mode 100644
index 0000000000000000000000000000000000000000..6d05bbb558482bbe453a9943f51a6d4422576634
--- /dev/null
+++ b/appengine/findit/waterfall/schedule_compile_try_job_pipeline.py
@@ -0,0 +1,48 @@
+# Copyright 2016 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.
+
+from common.waterfall import failure_type
+from model.wf_try_job import WfTryJob
+from waterfall.schedule_try_job_pipeline import ScheduleTryJobPipeline
+
+
+class ScheduleCompileTryJobPipeline(ScheduleTryJobPipeline):
+  """A pipeline for scheduling a new try job for failed compile build."""
+
+  def _GetBuildProperties(
+      self, master_name, builder_name, build_number, good_revision,
+      bad_revision, try_job_type, suspected_revisions):
+    properties = super(ScheduleCompileTryJobPipeline, self)._GetBuildProperties(
+        master_name, builder_name, build_number, good_revision,
+        bad_revision, try_job_type, suspected_revisions)
+    properties['target_buildername'] = builder_name
+
+    return properties
+
+  # Arguments number differs from overridden method - pylint: disable=W0221
+  def run(
+      self, master_name, builder_name, build_number, good_revision,
+      bad_revision, try_job_type, compile_targets, suspected_revisions):

[Sharu Jiang, 2016/08/09 23:05:04]

nit: Add doc string to explain parameters because the old function did :)

[chanli, 2016/08/11 01:27:54]

Done.

+
+    properties = self._GetBuildProperties(
+        master_name, builder_name, build_number, good_revision, bad_revision,
+        try_job_type, suspected_revisions)
+    additional_parameters = {'compile_targets': compile_targets}
+
+    build_id = self._TriggerTryJob(
+        master_name, builder_name, properties, additional_parameters)
+
+    try_job_result = WfTryJob.Get(master_name, builder_name, build_number)
+    try_job_result.compile_results.append({'try_job_id': build_id})
+    try_job_result.try_job_ids.append(build_id)
+    try_job_result.put()
+
+    # Create a corresponding WfTryJobData entity to capture as much metadata as
+    # early as possible.
+    self._CreateTryJobData(
+        build_id, master_name, builder_name, build_number,
+        failure_type.GetDescriptionForFailureType(try_job_type),
+        bool(compile_targets), bool(suspected_revisions))
+
+    return build_id