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(grub_boot_device) .get(info.manifest.system['architecture'])) info.image = info.connection.register_image(**registration_params)