with open(yaml_file, 'r') as f: doc = yaml.load(f) config.user_count = doc['config']['user_count'] config.bucket_count = doc['config']['bucket_count'] config.objects_count = doc['config']['objects_count'] config.objects_size_range = { 'min': doc['config']['objects_size_range']['min'], 'max': doc['config']['objects_size_range']['max'] } for k, v in doc.items(): if 'shards' in v: config.shards = doc['config']['shards'] print('shard value: %s' % config.shards) if 'max_objects' in v: config.max_objects = doc['config']['max_objects'] if doc['config'].get('dynamic_sharding', None) is True: config.dynamic_sharding = True config.max_objects_per_shard = doc['config'][ 'max_objects_per_shard'] else: config.dynamic_sharding = False log.info('user_count:%s\n' 'bucket_count: %s\n' 'objects_count: %s\n' 'objects_size_range: %s\n' 'shards: %s\n' 'max_objects: %s\n' % (config.user_count, config.bucket_count, config.objects_count, config.objects_size_range, config.shards, config.max_objects)) test_exec(config)
config.user_count = doc["config"]["user_count"] config.bucket_count = doc["config"]["bucket_count"] config.objects_count = doc["config"]["objects_count"] config.objects_size_range = { "min": doc["config"]["objects_size_range"]["min"], "max": doc["config"]["objects_size_range"]["max"], } for k, v in doc.items(): if "shards" in v: config.shards = doc["config"]["shards"] print("shard value: %s" % config.shards) if "max_objects" in v: config.max_objects = doc["config"]["max_objects"] if doc["config"].get("dynamic_sharding", None) is True: config.dynamic_sharding = True config.max_objects_per_shard = doc["config"][ "max_objects_per_shard"] else: config.dynamic_sharding = False log.info("user_count:%s\n" "bucket_count: %s\n" "objects_count: %s\n" "objects_size_range: %s\n" "shards: %s\n" "max_objects: %s\n" % ( config.user_count, config.bucket_count, config.objects_count, config.objects_size_range, config.shards, config.max_objects, ))