1 #!/usr/bin/env python 2 # 3 # Copyright 2017 - The Android Open Source Project 4 # 5 # Licensed under the Apache License, Version 2.0 (the "License"); 6 # you may not use this file except in compliance with the License. 7 # You may obtain a copy of the License at 8 # 9 # http://www.apache.org/licenses/LICENSE-2.0 10 # 11 # Unless required by applicable law or agreed to in writing, software 12 # distributed under the License is distributed on an "AS IS" BASIS, 13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 # See the License for the specific language governing permissions and 15 # limitations under the License. 16 import re 17 18 # Handles edge case of acronyms then another word, eg. CPUMetric -> CPU_Metric 19 # or lower camel case, cpuMetric -> cpu_Metric. 20 first_cap_re = re.compile('(.)([A-Z][a-z]+)') 21 # Handles CpuMetric -> Cpu_Metric 22 all_cap_re = re.compile('([a-z0-9])([A-Z])') 23 24 25 class Runner: 26 """Calls metrics and passes response to reporters. 27 28 Attributes: 29 metric_list: a list of metric objects 30 reporter_list: a list of reporter objects 31 object and value is dictionary returned by that response 32 """ 33 34 def __init__(self, metric_list, reporter_list): 35 self.metric_list = metric_list 36 self.reporter_list = reporter_list 37 38 def run(self): 39 """Calls metrics and passes response to reporters.""" 40 raise NotImplementedError() 41 42 43 class InstantRunner(Runner): 44 def convert_to_snake(self, name): 45 """Converts a CamelCaseName to snake_case_name 46 47 Args: 48 name: The string you want to convert. 49 Returns: 50 snake_case_format of name. 51 """ 52 temp_str = first_cap_re.sub(r'\1_\2', name) 53 return all_cap_re.sub(r'\1_\2', temp_str).lower() 54 55 def run(self): 56 """Calls all metrics, passes responses to reporters.""" 57 responses = {} 58 for metric in self.metric_list: 59 # [:-7] removes the ending '_metric'. 60 key_name = self.convert_to_snake(metric.__class__.__name__)[:-7] 61 responses[key_name] = metric.gather_metric() 62 for reporter in self.reporter_list: 63 reporter.report(responses) 64
