mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
71 lines
2.3 KiB
Python
71 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()
|