-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathrequest.py
68 lines (57 loc) · 2.65 KB
/
request.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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import boto.ec2
import boto.utils
import jinja2
import json
import time
def _create_master(id):
try:
# -- load configuration data via JSON
with open('%s.json' % id, 'r') as f:
config = json.load(f)
# user_data (cloud-init) from template
jinja2_environment = jinja2.Environment(loader=jinja2.FileSystemLoader('templates'))
jinja2_environment.filters['jsonify'] = json.dumps
USER_DATA = jinja2_environment.get_template('master.user_data').render(config=config)
# create a NetworkInterface in our VPC
interface = boto.ec2.networkinterface.NetworkInterfaceSpecification(
subnet_id = config['SUBNET_ID'],
groups = config['SECURITY_GROUP_IDS'],
associate_public_ip_address = True
)
interfaces = boto.ec2.networkinterface.NetworkInterfaceCollection(interface)
# create the Instance reservation
reservation = connection.run_instances(
image_id = config['IMAGE_ID'],
key_name = config['KEY_NAME'],
user_data = USER_DATA,
instance_type = config['MASTER_INSTANCE_TYPE'],
instance_initiated_shutdown_behavior = 'terminate',
placement_group = config['PLACEMENT_GROUP'],
network_interfaces = interfaces,
instance_profile_name = config['IAM_ROLE']
)
# -- loop until instance is running and update tags
for instance in reservation.instances:
status = instance.update()
while status == 'pending':
time.sleep(5)
status = instance.update()
instance.add_tag('Name', config['INSTANCE_NAME'])
return reservation
except:
pass
def main():
id = 'NECOFS' # name of JSON file to read, eg. NECOFS.json
reservation = _create_master(id)
print reservation
print [i.private_ip_address for i in reservation.instances]
# EC2 Instance metadata (can be run outside AWS with credentials)
instance_metadata = boto.utils.get_instance_metadata(timeout=5, num_retries=1)
instance_id = instance_metadata.get('instance-id')
placement = instance_metadata.get('placement')
availability_zone = placement.get('availability-zone')
region = availability_zone[:-1]
connection = boto.ec2.connect_to_region(region) # TODO, we can change region
local_ipv4 = instance_metadata.get('local-ipv4') # TODO: get port the Flask application is running on
if __name__ == "__main__":
main()