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 20def 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 39generate_args() 40