1 #!/usr/bin/env python2.7 2 3 # Copyright 2018 gRPC authors. 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 17 import gen_build_yaml as gen 18 import json 19 20 def generate_args(): 21 all_scenario_set = gen.generate_yaml() 22 all_scenario_set = all_scenario_set['tests'] 23 qps_json_driver_scenario_set = \ 24 [item for item in all_scenario_set if item['name'] == 'qps_json_driver'] 25 qps_json_driver_arg_set = \ 26 [item['args'][2] for item in qps_json_driver_scenario_set \ 27 if 'args' in item and len(item['args']) > 2] 28 deserialized_scenarios = [json.loads(item)['scenarios'][0] \ 29 for item in qps_json_driver_arg_set] 30 all_scenarios = {scenario['name'].encode('ascii', 'ignore'): \ 31 '\'{\'scenarios\' : [' + json.dumps(scenario) + ']}\'' \ 32 for scenario in deserialized_scenarios} 33 34 serialized_scenarios_str = str(all_scenarios).encode('ascii', 'ignore') 35 with open('qps_json_driver_scenarios.bzl', 'w') as f: 36 f.write('"""Scenarios of qps driver."""\n\n') 37 f.write('QPS_JSON_DRIVER_SCENARIOS = ' + serialized_scenarios_str + '\n') 38 39 generate_args() 40
