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