[Findit] Updating flakiness dashboard to be able to triage results

appengine/findit/waterfall/flake/recursive_flake_pipeline.py

 # 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 datetime import timedelta
 import logging
-from datetime import timedelta
 import random
 
 from common import appengine_util
 from common import constants
 from common import time_util
 from common.pipeline_wrapper import BasePipeline
 
 from model import analysis_status
+from model import result_status
 from model.flake.flake_swarming_task import FlakeSwarmingTask
 from model.flake.master_flake_analysis import MasterFlakeAnalysis
 from waterfall import waterfall_config
 from waterfall.process_flake_swarming_task_result_pipeline import (
     ProcessFlakeSwarmingTaskResultPipeline)
 from waterfall.trigger_flake_swarming_task_pipeline import (
     TriggerFlakeSwarmingTaskPipeline)
 
 
 def _UpdateAnalysisStatusUponCompletion(master_flake_analysis, status, error):
   master_flake_analysis.end_time = time_util.GetUTCNow()
   master_flake_analysis.status = status
 
   if error:
     master_flake_analysis.error = error
 
+  if master_flake_analysis.suspected_flake_build_number is not None:
+    master_flake_analysis.result_status = result_status.FOUND_UNTRIAGED
+  else:
+    master_flake_analysis.result_status = result_status.NOT_FOUND_UNTRIAGED
+
   master_flake_analysis.put()
 
 
 def _GetETAToStartAnalysis(manually_triggered):
   """Returns an ETA as of a UTC datetime.datetime to start the analysis.
 
   If not urgent, Swarming tasks should be run off PST peak hours from 11am to
   6pm on workdays.
 
   Args:
@@ skipping 187 matching lines @@
   # Unused argument - pylint: disable=W0613
   def run(self, master_name, builder_name, master_build_number,
           run_build_number, step_name, test_name, version_number,
           test_result_future, flakiness_algorithm_results_dict,
           manually_triggered=False):
 
     # Get MasterFlakeAnalysis success list corresponding to parameters.
     master_flake_analysis = MasterFlakeAnalysis.GetVersion(
         master_name, builder_name, master_build_number, step_name, test_name,
         version=version_number)
-    # Don't call another pipeline if we fail.
+
     flake_swarming_task = FlakeSwarmingTask.Get(
         master_name, builder_name, run_build_number, step_name, test_name)
 
+    # Don't call another pipeline if we fail.
     if flake_swarming_task.status == analysis_status.ERROR:
       # TODO(lijeffrey): Implement more detailed error detection and reporting,
       # such as timeouts, dead bots, etc.
       error = {
           'error': 'Swarming task failed',
           'message': 'Swarming task failed'
       }
       _UpdateAnalysisStatusUponCompletion(
           master_flake_analysis, analysis_status.ERROR, error)
       return
@@ skipping 19 matching lines @@
           flakiness_algorithm_results_dict=flakiness_algorithm_results_dict,
           manually_triggered=manually_triggered)
       # pylint: disable=W0201
       pipeline_job.target = appengine_util.GetTargetNameForModule(
           constants.WATERFALL_BACKEND)
       pipeline_job.StartOffPSTPeakHours(
           queue_name=self.queue_name or constants.DEFAULT_QUEUE)
     else:
       _UpdateAnalysisStatusUponCompletion(
           master_flake_analysis, analysis_status.COMPLETED, None)