[Findit] Post-process findit-to-fracas results.

appengine/findit/crash/fracas_crash_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.
 
 import base64
+import copy
 import datetime
 import json
 import logging
 
 from google.appengine.ext import ndb
 
 from common import appengine_util
 from common import constants
 from common import pubsub_util
 from common.pipeline_wrapper import BasePipeline
 from common.pipeline_wrapper import pipeline
 from crash import fracas
 from model import analysis_status
 from model.crash.crash_config import CrashConfig
 from model.crash.fracas_crash_analysis import FracasCrashAnalysis
 
 
 # TODO(katesonia): Move these to config page.
 _SIGNATURE_BLACKLIST_MARKERS = ['[Android Java Exception]']
 _PLATFORM_RENAME = {'linux': 'unix'}
+_FINDIT_FEEDBACK_URL_TEMPLATE = ('https://findit-for-me.googleplex.com/crash/'

[stgao, 2016/08/18 18:30:46]

The hostname should go to the config instead.

+                                 'fracas-result-feedback?key=%s')
 
 
 class FracasBasePipeline(BasePipeline):
   def __init__(self, crash_identifiers):
     super(FracasBasePipeline, self).__init__(crash_identifiers)
     self.crash_identifiers = crash_identifiers
 
   def run(self, *args, **kwargs):
     raise NotImplementedError()
 
@@ skipping 37 matching lines @@
     analysis.status = analysis_status.COMPLETED
     analysis.put()
 
 
 class PublishResultPipeline(FracasBasePipeline):
   def finalized(self):
     if self.was_aborted:  # pragma: no cover.
       logging.error('Failed to publish analysis result for %s',
                     repr(self.crash_identifiers))
 
+  def PostProcessResults(self, analysis, crash_identifiers):
+    analysis_result = copy.deepcopy(analysis.result)
+    analysis_result['feedback_url'] = (_FINDIT_FEEDBACK_URL_TEMPLATE %
+                                       analysis.key.urlsafe())
+    if analysis_result['found']:
+      for cl in analysis_result['suspected_cls']:
+        cl['confidence'] = round(cl['confidence'], 2)
+        cl.pop('reason', None)
+
+    return {
+        'crash_identifiers': crash_identifiers,
+        'client_id': analysis.client_id,
+        'result': analysis_result,
+    }
+
   # Arguments number differs from overridden method - pylint: disable=W0221
   def run(self, crash_identifiers):
     analysis = FracasCrashAnalysis.Get(crash_identifiers)
-    result = {
-        'crash_identifiers': crash_identifiers,
-        'client_id': analysis.client_id,
-        'result': analysis.result,
-    }
+    result = self.PostProcessResults(analysis, crash_identifiers)
     messages_data = [json.dumps(result, sort_keys=True)]
 
     crash_config = CrashConfig.Get()
     topic = crash_config.fracas['analysis_result_pubsub_topic']
     pubsub_util.PublishMessagesToTopic(messages_data, topic)
     logging.info('Published analysis result for %s', repr(crash_identifiers))
 
 
 class FracasCrashWrapperPipeline(BasePipeline):
   # Arguments number differs from overridden method - pylint: disable=W0221
@@ skipping 70 matching lines @@
                        platform, stack_trace, channel, historical_metadata):
     analysis_pipeline = FracasCrashWrapperPipeline(crash_identifiers)
     # Attribute defined outside __init__ - pylint: disable=W0201
     analysis_pipeline.target = appengine_util.GetTargetNameForModule(
         constants.CRASH_BACKEND_FRACAS)
     analysis_pipeline.start(queue_name=queue_name)
     logging.info('New analysis is scheduled for %s', repr(crash_identifiers))
     return True
 
   return False