[Findit] Refactoring WfTryJobData into BaseTryJobData, WfTryJobData, and FlakeTryJobData

appengine/findit/model/flake/flake_try_job.py

 # Copyright 2015 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
 
 from google.appengine.ext import ndb
 
 from model.base_try_job import BaseTryJob
 
 
 class FlakeTryJob(BaseTryJob):
   """Represents a try job results for a check flake try job."""
   # A list of dict containing results and urls of each flake try job.
   # For example:
   # [
   #      {
   #           'report': (dict) The 'result' dict of the try job,
   #           'url': (str) The url to the try job,
   #           'try_job_id': (str) The try job id.
   #      },
   #      ...
   # ]
   flake_results = ndb.JsonProperty(indexed=False, compressed=True)
 
+  @classmethod
+  def GetMasterName(cls, key):
+    return key.pairs()[0][1].split('/')[0]
+
+  @classmethod
+  def GetBuilderName(cls, key):
+    return key.pairs()[0][1].split('/')[1]
+
+  @classmethod
+  def GetStepName(cls, key):
+    return key.pairs()[0][1].split('/')[2]
+
+  @classmethod
+  def GetTestName(cls, key):
+    return base64.b64decode(key.pairs()[0][1].split('/')[3])
+
+  @classmethod
+  def GetGitHash(cls, key):
+    return key.pairs()[0][1].split('/')[4]
+
+  @ndb.ComputedProperty
+  def master_name(self):
+    return self.GetMasterName(self.key)
+
+  @ndb.ComputedProperty
+  def builder_name(self):
+    return self.GetBuilderName(self.key)
+
+  @ndb.ComputedProperty
+  def step_name(self):
+    return self.GetStepName(self.key)
+
+  @ndb.ComputedProperty
+  def test_name(self):
+    return self.GetTestName(self.key)
+
+  @ndb.ComputedProperty
+  def git_hash(self):
+    return self.GetGitHash(self.key)
+
   @staticmethod
   def _CreateTryJobId(master_name, builder_name, step_name, test_name,
                       git_hash):  # pragma: no cover
     """Creates an ID to associate with this try job.
 
     Args:
       master_name (str): The name of the master the flake was detected on.
       builder_name (str): the name of the builder the flake was detected on.
       step_name (str): The name of the step the flake was detected on.
       test_name (str): The original name of the test that flaked. This will be
@@ skipping 21 matching lines @@
         key=FlakeTryJob._CreateKey(master_name, builder_name, step_name,
                                    test_name, git_hash))
     flake_try_job.flake_results = flake_try_job.flake_results or []
     flake_try_job.try_job_ids = flake_try_job.try_job_ids or []
     return flake_try_job
 
   @staticmethod
   def Get(master_name, builder_name, step_name, test_name, git_hash):
     return FlakeTryJob._CreateKey(
         master_name, builder_name, step_name, test_name, git_hash).get()