def create_instance(service_uuid, instance_type, user_name, image_id): service = Service.get(uuid=service_uuid) subnet_id = config.PRIVATE_SUBNET_ID if 'deploy' in service.name.lower(): subnet_id = config.PUBLIC_SUBNET_ID conn = connect_ec2() reservation = conn.run_instances(image_id, key_name=service.key_name, instance_type=instance_type, security_group_ids=config.INSTANCE_SG, subnet_id=subnet_id) instances = reservation.instances instance = instances[0] host = Host.create_or_update(instance.id, private_ip=instance.private_ip_address, public_ip=instance.ip_address, public_dns=instance.public_dns_name, image_type=instance_type, image_id=image_id, username=user_name) host.add_service(service.uuid) instance.add_tag("Name", "{}-{}".format(service.name.lower(), host.uuid)) queue().enqueue_call(func=check_instance_for_configuration, args=[host.uuid, service_uuid], timeout=120) return host
def terminate_instance(host_uuid): host = Host.get(uuid=host_uuid) conn = connect_ec2() terminated_instances = conn.terminate_instances(instance_ids=[host.instance_id]) addresses = conn.get_all_addresses(filters={'private_ip_address': [host.private_ip]}) for address in addresses: address.disassociate() address.release() for instance in terminated_instances: Host.soft_delete(instance_id=instance.id)
def get_all_security_groups(): conn = connect_ec2() sg = conn.get_all_security_groups() return_values = [] for s in sg: return_values.append( {'name': s.name, 'label': s.name, 'id': s.id, 'value': s.id, 'vpc_id': s.vpc_id}) return return_values
def check_instance_for_configuration(host_uuid, service_uuid): conn = connect_ec2() host = Host.get(uuid=host_uuid) reservations = conn.get_all_instances(instance_ids=[host.instance_id]) if len(reservations) == 0 or len(reservations[0].instances) == 0: raise Exception('There are no AWS instances for this host.') instance = reservations[0].instances[0] if instance.state == 'running': queue().enqueue_call(func=configure_instance, args=[host_uuid, service_uuid]) return time.sleep(5) queue().enqueue_call(func=check_instance_for_configuration, args=[host_uuid, service_uuid], timeout=120)
def update_instances_from_ids(service_uuid, instance_ids, configure=True): service = Service.get(uuid=service_uuid) conn = connect_ec2() reservation = conn.get_all_instances(instance_ids=instance_ids) for r in reservation: for instance in r.instances: host = Host.create_or_update(instance.id, private_ip=instance.private_ip_address, public_ip=instance.ip_address, public_dns=instance.public_dns_name) host.add_service(service.uuid) instance.add_tag("Name", "{}-{}".format(service.name.lower(), host.uuid)) queue().enqueue_call(func=check_instance_for_configuration, args=[host.uuid, service.uuid], timeout=120)
def get_all_keypairs(): conn = connect_ec2() keys = conn.get_all_key_pairs() keys = [{'label': key.name, 'value': key.name} for key in keys] return keys