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.
68 lines
3.0 KiB
68 lines
3.0 KiB
import argparse
|
|
import logging
|
|
import os
|
|
import random
|
|
import sys
|
|
import threading
|
|
|
|
from utils import utils
|
|
|
|
|
|
logging.basicConfig(level=logging.INFO, format='%(threadName)s %(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
LOGGER = logging.getLogger(__file__)
|
|
LOGGER.setLevel(logging.INFO)
|
|
|
|
def terminate_instances_by_region(region_number, region_config, node_name_tag):
|
|
ec2_client, _ = utils.create_ec2_client(region_number, region_config)
|
|
filters = [{'Name': 'tag:Name','Values': [node_name_tag]}]
|
|
instance_ids = utils.get_instance_ids(ec2_client.describe_instances(Filters=filters))
|
|
if instance_ids:
|
|
ec2_client.terminate_instances(InstanceIds=instance_ids)
|
|
LOGGER.info("waiting until instances with tag %s died." % node_name_tag)
|
|
waiter = ec2_client.get_waiter('instance_terminated')
|
|
waiter.wait(InstanceIds=instance_ids)
|
|
LOGGER.info("instances with node name tag %s terminated." % node_name_tag)
|
|
else:
|
|
pass
|
|
LOGGER.warn("there is no instances to terminate")
|
|
|
|
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('--node_name_tag', type=str, dest='node_name_tag')
|
|
parser.add_argument('--region_number', type=str, dest='region_number')
|
|
parser.add_argument('--region_config', type=str,
|
|
dest='region_config', default='configuration.txt')
|
|
args = parser.parse_args()
|
|
|
|
if not args.instance_output or not os.path.isfile(args.instance_output):
|
|
LOGGER.info("%s is not existed" % args.instance_output)
|
|
sys.exit(1)
|
|
if args.region_number and args.node_name_tag:
|
|
region_number_items = args.region_number.split(",")
|
|
node_name_tag_items = args.node_name_tag.split(",")
|
|
thread_pool = []
|
|
for i in range(len(region_number_items)):
|
|
region_number = region_number_items[i]
|
|
node_name_tag = node_name_tag_items[i]
|
|
t = threading.Thread(target=terminate_instances_by_region, args=(region_number, args.region_config, node_name_tag))
|
|
t.start()
|
|
thread_pool.append(t)
|
|
for t in thread_pool:
|
|
t.join()
|
|
LOGGER.info("done.")
|
|
elif args.instance_output:
|
|
with open(args.instance_output, "r") as fin:
|
|
thread_pool = []
|
|
for line in fin.readlines():
|
|
items = line.split(" ")
|
|
region_number = items[1].strip()
|
|
node_name_tag = items[0].strip()
|
|
t = threading.Thread(target=terminate_instances_by_region, args=(region_number, args.region_config, node_name_tag))
|
|
t.start()
|
|
thread_pool.append(t)
|
|
for t in thread_pool:
|
|
t.join()
|
|
LOGGER.info("done.")
|
|
|