mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
af68be15c6
It adds the ability to automatically launch an EC2 instance after AMI registration. It has no mandatory configuration, only optional ones, like instance type, security groups, etc. They should be documented in later documentation patches.
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
from bootstrapvz.providers.ec2.tasks import ami
|
|
import time
|
|
import logging
|
|
|
|
|
|
class LaunchEC2Instance(Task):
|
|
description = 'Launching EC2 instance'
|
|
phase = phases.image_registration
|
|
predecessors = [ami.RegisterAMI]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
conn = info._ec2['connection']
|
|
r = conn.run_instances(info._ec2['image'],
|
|
security_group_ids=info.manifest.plugins['ec2_launch'].get('security_group_ids'),
|
|
instance_type=info.manifest.plugins['ec2_launch'].get('instance_type', 't2.micro'))
|
|
info._ec2['instance_id'] = r.instances[0].id
|
|
|
|
if 'tags' in info.manifest.plugins['ec2_launch']:
|
|
def apply_format(v):
|
|
return v.format(**info.manifest_vars)
|
|
tags = info.manifest.plugins['ec2_launch']['tags']
|
|
r = {k: apply_format(v) for k, v in tags.items()}
|
|
conn.create_tags([info._ec2['instance_id']], r)
|
|
|
|
|
|
class PrintPublicIPAddress(Task):
|
|
description = 'Waiting for the instance to launch'
|
|
phase = phases.image_registration
|
|
predecessors = [LaunchEC2Instance]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
ec2 = info._ec2
|
|
logger = logging.getLogger(__name__)
|
|
filename = info.manifest.plugins['ec2_launch']['print_public_ip']
|
|
if not filename:
|
|
filename = '/dev/null'
|
|
f = open(filename, 'w')
|
|
|
|
i = 0
|
|
instance = None
|
|
while True:
|
|
logger.debug('Waiting a bit to get instance metadata...')
|
|
time.sleep(5)
|
|
|
|
i += 1
|
|
if i > 10:
|
|
logger.error('Waited too much, giving up')
|
|
break
|
|
|
|
r = ec2['connection'].get_only_instances([ec2['instance_id']])
|
|
if not r and not r[0]:
|
|
logger.error('Could not get instance metadata')
|
|
break
|
|
|
|
instance = r[0]
|
|
if instance.ip_address:
|
|
break
|
|
|
|
if instance and instance.ip_address:
|
|
logger.info('******* EC2 IP ADDRESS: %s *******' % instance.ip_address)
|
|
f.write(instance.ip_address)
|
|
else:
|
|
logger.error('Could not get IP address for the instance')
|
|
f.write('')
|
|
|
|
f.close()
|