mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
168 lines
6.8 KiB
Python
168 lines
6.8 KiB
Python
from base import Task
|
|
from common import phases
|
|
from common.exceptions import TaskError
|
|
from common.tools import log_check_call
|
|
from ebs import Snapshot
|
|
from common.tasks import workspace
|
|
from connection import Connect
|
|
from . import assets
|
|
import os.path
|
|
|
|
cert_ec2 = os.path.join(assets, 'certs/cert-ec2.pem')
|
|
|
|
|
|
class AMIName(Task):
|
|
description = 'Determining the AMI name'
|
|
phase = phases.preparation
|
|
predecessors = [Connect]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
ami_name = info.manifest.image['name'].format(**info.manifest_vars)
|
|
ami_description = info.manifest.image['description'].format(**info.manifest_vars)
|
|
|
|
images = info.connection.get_all_images()
|
|
for image in images:
|
|
if ami_name == image.name:
|
|
msg = 'An image by the name {ami_name} already exists.'.format(ami_name=ami_name)
|
|
raise TaskError(msg)
|
|
info.ami_name = ami_name
|
|
info.ami_description = ami_description
|
|
|
|
|
|
class BundleImage(Task):
|
|
description = 'Bundling the image'
|
|
phase = phases.image_registration
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
bundle_name = 'bundle-{id}'.format(id=info.run_id)
|
|
info.bundle_path = os.path.join(info.workspace, bundle_name)
|
|
log_check_call(['/usr/bin/euca-bundle-image',
|
|
'--image', info.volume.image_path,
|
|
'--user', info.credentials['user-id'],
|
|
'--privatekey', info.credentials['private-key'],
|
|
'--cert', info.credentials['certificate'],
|
|
'--ec2cert', cert_ec2,
|
|
'--destination', info.bundle_path,
|
|
'--prefix', info.ami_name])
|
|
|
|
|
|
class UploadImage(Task):
|
|
description = 'Uploading the image bundle'
|
|
phase = phases.image_registration
|
|
predecessors = [BundleImage]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
manifest_file = os.path.join(info.bundle_path, info.ami_name + '.manifest.xml')
|
|
if info.host['region'] == 'us-east-1':
|
|
s3_url = 'https://s3.amazonaws.com/'
|
|
elif info.host['region'] == 'cn-north-1':
|
|
s3_url = 'https://s3.cn-north-1.amazonaws.com.cn'
|
|
else:
|
|
s3_url = 'https://s3-{region}.amazonaws.com/'.format(region=info.host['region'])
|
|
log_check_call(['/usr/bin/euca-upload-bundle',
|
|
'--bucket', info.manifest.image['bucket'],
|
|
'--manifest', manifest_file,
|
|
'--access-key', info.credentials['access-key'],
|
|
'--secret-key', info.credentials['secret-key'],
|
|
'--url', s3_url,
|
|
'--region', info.host['region'],
|
|
'--ec2cert', cert_ec2])
|
|
|
|
|
|
class RemoveBundle(Task):
|
|
description = 'Removing the bundle files'
|
|
phase = phases.cleaning
|
|
successors = [workspace.DeleteWorkspace]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from shutil import rmtree
|
|
rmtree(info.bundle_path)
|
|
del info.bundle_path
|
|
|
|
|
|
class RegisterAMI(Task):
|
|
description = 'Registering the image as an AMI'
|
|
phase = phases.image_registration
|
|
predecessors = [Snapshot, UploadImage]
|
|
|
|
# Source: http://docs.aws.amazon.com/AWSEC2/latest/UserGuide/UserProvidedKernels.html#AmazonKernelImageIDs
|
|
kernel_mapping = {'ap-northeast-1': # Asia Pacific (Tokyo) Region
|
|
{'i386': 'aki-136bf512', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-176bf516' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'ap-southeast-1': # Asia Pacific (Singapore) Region
|
|
{'i386': 'aki-ae3973fc', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-503e7402' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'ap-southeast-2': # Asia Pacific (Sydney) Region
|
|
{'i386': 'aki-cd62fff7', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-c362fff9' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'eu-west-1': # EU (Ireland) Region
|
|
{'i386': 'aki-68a3451f', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-52a34525' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'sa-east-1': # South America (Sao Paulo) Region
|
|
{'i386': 'aki-5b53f446', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-5553f448' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'us-east-1': # US East (Northern Virginia) Region
|
|
{'i386': 'aki-8f9dcae6', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-919dcaf8' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'us-gov-west-1': # AWS GovCloud (US)
|
|
{'i386': 'aki-1fe98d3c', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-1de98d3e' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'us-west-1': # US West (Northern California) Region
|
|
{'i386': 'aki-8e0531cb', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-880531cd' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'us-west-2': # US West (Oregon) Region
|
|
{'i386': 'aki-f08f11c0', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-fc8f11cc' # pv-grub-hd0_1.04-x86_64.gz
|
|
},
|
|
'cn-north-1': # China North (Beijing) Region
|
|
{'i386': 'aki-908f1da9', # pv-grub-hd0_1.04-i386.gz
|
|
'amd64': 'aki-9e8f1da7' # pv-grub-hd0_1.04-x86_64.gz
|
|
}
|
|
}
|
|
@classmethod
|
|
def run(cls, info):
|
|
registration_params = {'name': info.ami_name,
|
|
'description': info.ami_description}
|
|
registration_params['architecture'] = {'i386': 'i386',
|
|
'amd64': 'x86_64'}.get(info.manifest.system['architecture'])
|
|
|
|
if info.manifest.volume['backing'] == 's3':
|
|
grub_boot_device = 'hd0'
|
|
else:
|
|
root_dev_name = {'pvm': '/dev/sda',
|
|
'hvm': '/dev/xvda'}.get(info.manifest.data['virtualization'])
|
|
registration_params['root_device_name'] = root_dev_name
|
|
from base.fs.partitionmaps.none import NoPartitions
|
|
if isinstance(info.volume.partition_map, NoPartitions):
|
|
grub_boot_device = 'hd0'
|
|
else:
|
|
grub_boot_device = 'hd00'
|
|
|
|
from boto.ec2.blockdevicemapping import BlockDeviceType
|
|
from boto.ec2.blockdevicemapping import BlockDeviceMapping
|
|
block_device = BlockDeviceType(snapshot_id=info.snapshot.id, delete_on_termination=True,
|
|
size=info.volume.size.get_qty_in('GiB'))
|
|
registration_params['block_device_map'] = BlockDeviceMapping()
|
|
registration_params['block_device_map'][root_dev_name] = block_device
|
|
|
|
if info.manifest.data['virtualization'] == 'hvm':
|
|
registration_params['virtualization_type'] = 'hvm'
|
|
else:
|
|
registration_params['virtualization_type'] = 'paravirtual'
|
|
registration_params['kernel_id'] = (cls.kernel_mapping
|
|
.get(info.host['region'])
|
|
.get(info.manifest.system['architecture']))
|
|
|
|
info.image = info.connection.register_image(**registration_params)
|