You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.6 KiB
36 lines
1.6 KiB
6 years ago
|
import argparse
|
||
|
|
import os
|
||
|
|
import random
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from utils import utils
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
parser = argparse.ArgumentParser(description='This script helps you to collect public ips')
|
||
|
|
parser.add_argument('--instance_output', type=str, dest='instance_output',
|
||
|
|
default='instance_output.txt',
|
||
|
|
help='the file contains node_name_tag and region number of created instances.')
|
||
|
|
parser.add_argument('--region_config', type=str,
|
||
|
|
dest='region_config', default='configuration.txt')
|
||
|
|
parser.add_argument('--file_output', type=str,
|
||
|
|
dest='file_output', default='raw_ip.txt')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
if not args.instance_output or not os.path.isfile(args.instance_output):
|
||
|
|
print "%s or %s are not existed" % (args.file_output, args.instance_output)
|
||
|
|
sys.exit(1)
|
||
|
|
if args.instance_output:
|
||
|
|
with open(args.instance_output, "r") as fin, open(args.file_output, "w") as fout:
|
||
|
|
total_ips = []
|
||
|
|
node_name_tag_list = []
|
||
|
|
for line in fin.readlines():
|
||
|
|
items = line.split(" ")
|
||
|
|
region_number = items[1].strip()
|
||
|
|
node_name_tag = items[0].strip()
|
||
|
|
ip_list = utils.collect_public_ips(region_number, node_name_tag, args.region_config)
|
||
|
|
total_ips.extend([(ip, node_name_tag) for ip in ip_list])
|
||
|
|
random.shuffle(total_ips)
|
||
|
|
for tuple in total_ips:
|
||
|
|
fout.write(tuple[0] + " " + tuple[1] + "\n")
|
||
|
|
print "Done collecting public ips %s" % args.file_output
|