[Findit] Adding logic to force try jobs regardless of bailout or previous results

appengine/findit/waterfall/build_failure_analysis_pipelines.py

 # 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.
 
 from datetime import datetime
 import logging
 
 from google.appengine.ext import ndb
 
 from common import appengine_util
 from common import constants
+from model import analysis_status
 from model.wf_analysis import WfAnalysis
-from model import analysis_status
 from waterfall import analyze_build_failure_pipeline
 
+
 @ndb.transactional
 def NeedANewAnalysis(
     master_name, builder_name, build_number, failed_steps,
     build_completed, force):
   """Checks status of analysis for the build and decides if a new one is needed.
 
   A WfAnalysis entity for the given build will be created if none exists.
   When a new analysis is needed, this function will create and save a WfAnalysis
   entity to the datastore, or it will reset the existing one but still keep the
   result of last analysis.
@@ skipping 42 matching lines @@
   # TODO: support following cases
   # * Automatically retry if last analysis failed with errors.
   # * Analysis is not complete and no update in the last 5 minutes.
   return False
 
 
 def ScheduleAnalysisIfNeeded(master_name, builder_name, build_number,
                              failed_steps=None,
                              build_completed=False,
                              force=False,
+                             force_try_job=False,
                              queue_name=constants.DEFAULT_QUEUE):
   """Schedules an analysis if needed and returns the build analysis.
 
   When the build failure was already analyzed and a new analysis is scheduled,
   the returned WfAnalysis will still have the result of last completed analysis.
 
   Args:
     master_name (str): The master name of the failed build.
     builder_name (str): The builder name of the failed build.
     build_number (int): The build number of the failed build.
     failed_steps (list): The names of all failed steps reported for the build.
     build_completed (bool): Indicate whether the build is completed.
     force (bool): If True, a fresh new analysis will be triggered even when an
         old one was completed already; otherwise bail out.
+    force_try_job (bool): Indicate whether or not a try job should be run or
+        rerun as long as a corresponding try bot is configured.
     queue_name (str): The task queue to be used for pipeline tasks.
 
   Returns:
     A WfAnalysis instance.
   """
   if NeedANewAnalysis(
       master_name, builder_name, build_number, failed_steps,
       build_completed, force):
     pipeline_job = analyze_build_failure_pipeline.AnalyzeBuildFailurePipeline(
-        master_name, builder_name, build_number, build_completed)
+        master_name, builder_name, build_number, build_completed,
+        force_try_job)
     # Explicitly run analysis in the backend module "waterfall-backend".
     # Note: Just setting the target in queue.yaml does NOT work for pipeline
     # when deployed to App Engine, but it does work in dev-server locally.
     # A possible reason is that pipeline will pick a default target if none is
     # specified explicitly, and the default target is used rather than the one
     # in the queue.yaml file, but this contradicts the documentation in
     # https://cloud.google.com/appengine/docs/python/taskqueue/tasks#Task.
     pipeline_job.target = appengine_util.GetTargetNameForModule(
         constants.WATERFALL_BACKEND)
     pipeline_job.start(queue_name=queue_name)
 
     logging.info('An analysis was scheduled for build %s, %s, %s: %s',
                  master_name, builder_name, build_number,
                  pipeline_job.pipeline_status_path())
 
   return WfAnalysis.Get(master_name, builder_name, build_number)