from boto_util import BotoUtil
import argparse
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('region', help='Region')
parser.add_argument('instance_name', help='InstanceName')
args = parser.parse_args()
BUtil = BotoUtil(args.region)
dns_tup, cluster_name = BUtil.get_ec2_instances(args.instance_name)
for dns in dns_tup:
print dns
print "Cluster name: {}".format(cluster_name)
BUtil.write_dns(cluster_name, dns_tup)
parser.add_argument('pem_key_name', help='region specific pem key name to associate instances with')
parser.add_argument('num_instances', help='number of instances to spin up')
parser.add_argument('security_groups', help='array of security group names')
parser.add_argument('instance_type', help='type of instances to spin up')
parser.add_argument('volume_size', help='size in GB of default EBS volume')
parser.add_argument('price', help='bid price of spot instance')
parser.add_argument('image', help='region specific ami to use')
args = parser.parse_args()
BUtil = BotoUtil(args.region)
IConf = InstanceConfig(purchase_type=args.purchase_type,
region=args.region,
az=args.az,
image=args.image,
price=args.price,
num_instances=args.num_instances,
key_name=args.pem_key_name,
security_groups=[args.security_groups],
instance_type=args.instance_type,
tag_name=args.instance_name,
vol_size=args.volume_size)
BUtil.create_ec2(IConf)
dns_tup, cluster_name = BUtil.get_ec2_instances(args.instance_name)
BUtil.write_dns(args.instance_name, dns_tup)
