ec2-init.py


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

#!/usr/bin/python

import subprocess
from Cheetah.Template import Template
import sys

import ec2init

def warn(str):
    sys.stderr.write(str)

def main():
    cloud = ec2init.EC2Init()

    data = None
    try:
        cloud.get_data_source()
    except Exception as e:
        print e
        sys.stderr.write("Failed to get instance data")
        sys.exit(1)

    #print "user data is:" + cloud.get_user_data()

    # store the metadata
    cloud.update_cache()

    # parse the user data (ec2-run-userdata.py)
    try:
        cloud.sem_and_run("consume_userdata", "once-per-instance",
            cloud.consume_userdata,[],False)
    except:
        warn("consuming user data failed!")
        raise

    # set the defaults (like what ec2-set-defaults.py did)
    # TODO: cloud.set_defaults()

    # set the ssh keys up
    cloud.apply_credentials()

    # finish, send the cloud-config event
    cloud.initctl_emit()

    sys.exit(0)

if __name__ == '__main__':
    main()