client = boto3.client('ec2', region_name=ec2_region)

allocation_id = client.describe_addresses(PublicIps=[eip]).get("Addresses")[0]["AllocationId"]

try:

association_id = client.describe_addresses(PublicIps=[eip]).get("Addresses")[0]["AssociationId"]

except IndexError:

association_id = ""

# In VPC, if the address is already associated we must disassociate it first

if association_id:

client.disassociate_address(AssociationId=association_id)

try:

client.associate_address(InstanceId=instance_id, AllocationId=allocation_id)

except Exception as e:

print "[-] Error assigning rollback EIP to original instance!"

print e

sys.exit(2)

print "[+] Assigned {0} back to owner {1}.".format(eip, instance_id)

#conn = boto.connect_ec2()

#try:

# conn.associate_address(instance_id, eip)

#except Exception as e:

# print "[-] Error assigning rollback EIP to instance!"

# print e

# sys.exit(2)

if re.search(judgement_txt, response.json()["operation"]):

print '[+] Deployment is now awaiting judgement.'

return True

try:

if len(regex) > 0:

if re.search(regex, response.text):

print "[+] qa(): found string \"{0}\"".format(regex)

return True

else:

if response.status_code == 200:

print "[+] qa(): got status code 200"

return True

except Exception:

return False

token = get_token(deployment_id)

payload = {'id': deployment_id, 'token': token}

post_url = base_url + '/deployment/rollback'

headers = {'Content-type': 'application/json', 'Accept': 'text/plain'}

requests.post(post_url, data=json.dumps(payload), headers=headers)

conn = boto.connect_ec2()

try:

instance_id = [address.instance_id for address in conn.get_all_addresses(filters={"public-ip": ip})]

except Exception as e:

print "[-] Error:"

print "{0}".format(e)

sys.exit(2)

if len(instance_id[0]) < 9:

print "[-] Error: IP is not currently assigned to any instance."

sys.exit(2)

token = get_token(deployment_id)

payload = {'id': deployment_id, 'token': token}

post_url = base_url + '/deployment/proceed'

headers = {'Content-type': 'application/json', 'Accept': 'text/plain'}

requests.post(post_url, data=json.dumps(payload), headers=headers)

version = '1.3'

parser = argparse.ArgumentParser(description="AMI Asgard Deployment Script.")

parser.add_argument("-n", "--nofollow", action='store_true', help="Do not follow redirects for the test URL.")

parser.add_argument("--eip", help="Elastic IP address for rollback. (EIP based deployments only).", required=False)

parser.add_argument("-b", "--bypass", action="store_true", help="Bypass the URL test")

parser.add_argument("-j", "--judgment", help="Judgment timeout Default: 150 seconds", required=False, const=150,

nargs='?', type=int, default=150)

parser.add_argument("-s", "--startup", help="Asgard in-service startup timeout. Default: 10 minutes",

required=False, const=10, nargs='?', type=int, default=10)

parser.add_argument('-t', '--timeout', help='Asgard polling timeout. (How long to wait for deployment to enter'

'judgment phase before rolling back). '

'Default: 600 seconds', required=False,

const=600, nargs='?', type=int, default=600)

parser.add_argument('asg_id')

parser.add_argument('ami_id')

parser.add_argument('test_url', nargs='?')

parser.add_argument('regex', nargs='?')

args = parser.parse_args()

print "AMI Asgard Deployment Script {0}".format(version)

asg_id = args.asg_id

ami_id = args.ami_id

poll_timeout = args.timeout

judge_timeout = args.judgment

startup_timeout = args.startup

if not args.bypass:

test_url = args.test_url

if args.regex:

global regex

regex = args.regex

print "[+] Asgard Host: {0}".format(asgard_host)

print "[+] EC2 Region: {0}".format(ec2_region)

print "[+] ASG: {0}".format(asg_id)

print "[+] AMI to Launch: {0}".format(ami_id)

if args.eip:

print "[+] Determining rollback instance_id for: {0}".format(args.eip)

rbinstance_id = search_ip(args.eip)

print "{0}".format(rbinstance_id)

query = base_url + '/deployment/prepare?id=' + asg_id

f = urllib2.urlopen(query)

deflcjson = f.read()

f.close()

deflc = json.loads(deflcjson)

deflc['lcOptions']['imageId'] = ami_id

deflc['deploymentOptions'] = {

"clusterName": asg_id,

"notificationDestination": notify,

"steps": [

{"type": "CreateAsg"},

{"type": "Resize", "targetAsg": "Next", "capacity": deflc['asgOptions']['minSize'],

"startUpTimeoutMinutes": startup_timeout},

{"type": "DisableAsg", "targetAsg": "Previous"},

{"type": "Judgment", "durationMinutes": 30},

{"type": "DeleteAsg", "targetAsg": "Previous"},

]

}

post_url = base_url + '/deployment/start'

headers = {'Content-type': 'application/json', 'Accept': 'text/plain'}

response = requests.post(post_url, data=json.dumps(deflc), headers=headers)

print response

print response.text

print "[+] Waiting for deployment to enter judgement state. Timeout is: {0}".format(poll_timeout)

deployment_id = response.json()["deploymentId"]

poll_url = base_url + '/task/show/' + deployment_id + '.json'

time.sleep(15)

try:

polling.poll(

lambda: requests.get(poll_url, allow_redirects=True, timeout=requests_timeout),

check_success=judgement_ready,

step=30,

timeout=poll_timeout

)

except polling.TimeoutException, te:

print "[-] Error waiting for judgement."

while not te.values.empty():

print "{0}".format(te.values.get())

print "{0}".format(get_error(poll_url))

sys.exit(2)

if args.eip:

time.sleep(30)

if rbinstance_id == search_ip(args.eip):

print "[-] Error: New instance did not assign itself EIP: {0}. Cannot proceed with judgment. " \

"Initiating rollback.".format(args.eip)

rollback(deployment_id)

assign_eip(args.eip, rbinstance_id)

sys.exit(2)

if not args.bypass:

if args.nofollow:

allow_redirects = False

else:

allow_redirects = True

try:

polling.poll(

lambda: requests.get(test_url, allow_redirects=allow_redirects, timeout=requests_timeout),

check_success=qa,

ignore_exceptions=(requests.exceptions.Timeout, requests.exceptions.ConnectionError),

step=3,

timeout=judge_timeout

)

except polling.TimeoutException, te:

print "[-] Error running test: Test url failed to meet test criteria."

while not te.values.empty():

print "{0}".format(te.values.get())

print "[+] Initiating rollback of deployment."

rollback(deployment_id)

if args.eip:

assign_eip(args.eip, rbinstance_id)

sys.exit(2)

print "[+] Tests passed. Proceeding deployment."

proceed(deployment_id)

else:

print "[+] Test bypassed. Proceeding deployment."