domserver-ami/createami.py at master · MallowTech/domserver-ami · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#!/usr/bin/env python
from __future__ import print_function
import boto.ec2
import sys

import time
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

# From http://cloud-images.ubuntu.com/locator/ec2/
# Choose the hvm:ebs-ssd "Instance Type" for say trusty us-east
#baseami = 'ami-fdb9fc98'  # Trusty 14.04 amd64 hvm:ebs-ssd 2015-09-28
baseami = 'ami-5c207736'  # Trusty 14.04 amd64 hvm:ebs-ssd 2015-12-18
region = 'us-east-1'
#ssh_keypair = 'domjudge-aws'
#ssh_securitygroup = 'open-ssh'

cconfig = """#cloud-config
power_state:
  mode: poweroff
  timeout: 900
  condition: True
"""

udscript = """#!/bin/bash -ex

# install needed packages for ansible
apt-get install -y -q software-properties-common git-core
apt-add-repository -y ppa:ansible/ansible
apt-get update
apt-get install -y -q ansible

ansible-pull -U http://github.com/ubergeek42/domserver-ami.git -d /mnt/playbooks -i "localhost,"

exit 0
"""
combined_message = MIMEMultipart()
sub_message = MIMEText(cconfig, "cloud-config", sys.getdefaultencoding())
sub_message.add_header('Content-Disposition', 'attachment; filename="00cloudconfig.txt"')
combined_message.attach(sub_message)

sub_message = MIMEText(udscript, "x-shellscript", sys.getdefaultencoding())
sub_message.add_header('Content-Disposition', 'attachment; filename="01bootstrap.txt"')
combined_message.attach(sub_message)

conn = boto.ec2.connect_to_region(region)
reservation = conn.run_instances(
    baseami,
    #key_name=ssh-keypair,
    #security_groups=[ssh_securitygroup],
    instance_type='t2.micro',
    user_data=combined_message.as_string()
)
instance = reservation.instances[0]

print("Waiting for instance to boot")
while instance.state != 'running':
    print(".", end='')
    sys.stdout.flush()
    time.sleep(5)
    instance.update()
print()

print("Instance provisioning...")
print("Waiting for instance to stop")
while instance.state != 'stopped':
    print(".", end='')
    sys.stdout.flush()
    time.sleep(5)
    instance.update()
print()

ts = int(time.time())
newami_id = conn.create_image(instance.id, "DOMjudge-domserver-{0}".format(ts), description="DOMjudge DOMserver {0}".format(ts))
instance.terminate()

print("DOMjudge DOMserver Created")
print("AMI ID: " + newami_id)

print("Waiting for image creation to finish")
image = conn.get_all_images(image_ids=[newami_id])[0]
while image.state == 'pending':
    print(".", end='')
    sys.stdout.flush()
    time.sleep(5)
    image.update()
if image.state == 'available':
    print("Image created successfully!")
    print("AMI ID: " + newami_id)
else:
    print("Error creating image. Check AWS console for details")