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.")