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