log.info('test data dir not exists, creating.. ') os.makedirs(TEST_DATA_PATH) parser = argparse.ArgumentParser(description='RGW S3 Automation') parser.add_argument('-c', dest="config", help='RGW Test yaml configuration') args = parser.parse_args() yaml_file = args.config config = Config() with open(yaml_file, 'r') as f: doc = yaml.load(f) 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'] } config.sharding_type = doc['config']['sharding_type'] log.info('objects_count: %s\n' 'objects_size_range: %s\n' 'sharding_type: %s\n' % (config.objects_count, config.objects_size_range, config.sharding_type)) test_exec(config)
if __name__ == "__main__": project_dir = os.path.abspath(os.path.join(__file__, "../../..")) test_data_dir = "test_data" TEST_DATA_PATH = os.path.join(project_dir, test_data_dir) log.info("TEST_DATA_PATH: %s" % TEST_DATA_PATH) if not os.path.exists(TEST_DATA_PATH): log.info("test data dir not exists, creating.. ") os.makedirs(TEST_DATA_PATH) parser = argparse.ArgumentParser(description="RGW S3 Automation") parser.add_argument("-c", dest="config", help="RGW Test yaml configuration") args = parser.parse_args() yaml_file = args.config config = Config() with open(yaml_file, "r") as f: doc = yaml.load(f) 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"], } config.sharding_type = doc["config"]["sharding_type"] log.info("objects_count: %s\n" "objects_size_range: %s\n" "sharding_type: %s\n" % (config.objects_count, config.objects_size_range, config.sharding_type)) test_exec(config)