[Dashboard] Get or create ancestors in add_histograms_queue

dashboard/dashboard/add_point_queue.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.
 
 """URL endpoint to add new graph data to the datastore."""
 
 import datetime
 import json
 import logging
 
@@ skipping 160 matching lines @@
 
   Raises:
     RuntimeError: Something went wrong when trying to get the parent test.
   """
   master_name = row_dict.get('master')
   bot_name = row_dict.get('bot')
   test_name = row_dict.get('test').strip('/')
   units = row_dict.get('units')
   higher_is_better = row_dict.get('higher_is_better')
   improvement_direction = _ImprovementDirection(higher_is_better)
-  internal_only = _BotInternalOnly(bot_name, bot_whitelist)
+  internal_only = BotInternalOnly(bot_name, bot_whitelist)
   benchmark_description = row_dict.get('benchmark_description')
 
-  parent_test = _GetOrCreateAncestors(
-      master_name, bot_name, test_name, units=units,
-      improvement_direction=improvement_direction,
-      internal_only=internal_only,
-      benchmark_description=benchmark_description)
+  parent_test = GetOrCreateAncestors(
+      master_name, bot_name, test_name, internal_only=internal_only,
+      benchmark_description=benchmark_description, units=units,
+      improvement_direction=improvement_direction)
 
   return parent_test
 
 
 def _ImprovementDirection(higher_is_better):
   """Returns an improvement direction (constant from alerts_data) or None."""
   if higher_is_better is None:
     return None
   return anomaly.UP if higher_is_better else anomaly.DOWN
 
 
-def _BotInternalOnly(bot_name, bot_whitelist):
+def BotInternalOnly(bot_name, bot_whitelist):
   """Checks whether a given bot name is internal-only.
 
   If a bot name is internal only, then new data for that bot should be marked
   as internal-only.
   """
   if not bot_whitelist:
     logging.warning(
         'No bot whitelist available. All data will be internal-only. If this '
         'is not intended, please add a bot whitelist using /edit_site_config.')
     return True
   return bot_name not in bot_whitelist
 
 
-def _GetOrCreateAncestors(
-    master_name, bot_name, test_name, units=None,
-    improvement_direction=None, internal_only=True, benchmark_description=''):
+def GetOrCreateAncestors(
+    master_name, bot_name, test_name, internal_only=True,
+    benchmark_description='', units=None, improvement_direction=None):
   """Gets or creates all parent Master, Bot, TestMetadata entities for a Row."""
 
   master_entity = _GetOrCreateMaster(master_name)
   _GetOrCreateBot(bot_name, master_entity.key, internal_only)
 
   # Add all ancestor tests to the datastore in order.
   ancestor_test_parts = test_name.split('/')
 
   test_path = '%s/%s' % (master_name, bot_name)
   suite = None
@@ skipping 119 matching lines @@
 
   if properties_changed:
     existing.put()
   return existing
 
 
 def _IsRefBuild(test_key):
   """Checks whether a TestMetadata is for a reference build test run."""
   test_parts = test_key.id().split('/')
   return test_parts[-1] == 'ref' or test_parts[-1].endswith('_ref')