diff --git a/aws-experiment-launch/generate_distribution_config.py b/aws-experiment-launch/generate_distribution_config.py new file mode 100644 index 000000000..bc5570145 --- /dev/null +++ b/aws-experiment-launch/generate_distribution_config.py @@ -0,0 +1,33 @@ +import argparse +import sys + +from utils import utils + +if __name__ == "__main__": + parser = argparse.ArgumentParser(description='This script helps you to genereate distribution config') + parser.add_argument('--ip_list_file', type=str, dest='ip_list_file', + default=None, help="the file containing available raw ips") + # If the raw_ip_file is None we need to use the region, node_name_tag and region_config to collect raw_ip + parser.add_argument('--region', type=str, dest='region_number', + default="4", help="region number") + parser.add_argument('--node_name_tag', type=str, + dest='node_name_tag', default='4-NODE-23-36-01-2018-07-05') + parser.add_argument('--region_config', type=str, + dest='region_config', default='configuration.txt') + + parser.add_argument('--shard_num', type=int, dest='shard_num', default=1) + parser.add_argument('--client_num', type=int, dest='client_number', default=1) + parser.add_argument('--distribution_config', type=str, + dest='distribution_config', default='distribution_config.txt') + args = parser.parse_args() + + if args.raw_ip_file == None: + utils.generate_distribution_config2( + args.region_number, args.node_name_tag, args.region_config, + args.shard_num, args.client_num, args.distribution_config) + else: + with open(args.ip_list_file, "r") as fin: + lines = fin.readlines() + ip_list = [line.strip() for line in lines] + + ip_list = utils.collect_public_ips(args.region_number, args.node_name_tag, args.region_config) \ No newline at end of file