mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
f62c8ade99
Up until now I didn't see the point of using spaces for indentation. However, the previous commit (a18bec3) was quite eye opening. Given that python is an indentation aware language, the amount of mistakes that went unnoticed because tabs and spaces were used at the same time (tabs for indentation and spaces for alignment) were unacceptable. E101,W191 have been re-enable in the tox flake8 checker and the documentation has been modified accordingly. The following files have been left as-is: * bootstrapvz/common/assets/extlinux/extlinux.conf * bootstrapvz/common/assets/init.d/expand-root * bootstrapvz/common/assets/init.d/generate-ssh-hostkeys * bootstrapvz/common/assets/init.d/squeeze/generate-ssh-hostkeys * bootstrapvz/plugins/docker_daemon/assets/init.d/docker * bootstrapvz/providers/ec2/assets/bin/growpart * bootstrapvz/providers/ec2/assets/grub.d/40_custom * bootstrapvz/providers/ec2/assets/init.d/ec2-get-credentials * bootstrapvz/providers/ec2/assets/init.d/ec2-run-user-data * docs/_static/taskoverview.coffee * docs/_static/taskoverview.less * tests/unit/subprocess.sh
127 lines
5.4 KiB
Python
127 lines
5.4 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
from bootstrapvz.common.exceptions import TaskError
|
|
from bootstrapvz.common.tools import log_check_call
|
|
from ebs import Snapshot
|
|
from bootstrapvz.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.name.format(**info.manifest_vars)
|
|
ami_description = info.manifest.provider['description'].format(**info.manifest_vars)
|
|
|
|
images = info._ec2['connection'].get_all_images(owners=['self'])
|
|
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._ec2['ami_name'] = ami_name
|
|
info._ec2['ami_description'] = ami_description
|
|
|
|
|
|
class BundleImage(Task):
|
|
description = 'Bundling the image'
|
|
phase = phases.image_registration
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
bundle_name = 'bundle-' + info.run_id
|
|
info._ec2['bundle_path'] = os.path.join(info.workspace, bundle_name)
|
|
arch = {'i386': 'i386', 'amd64': 'x86_64'}.get(info.manifest.system['architecture'])
|
|
log_check_call(['euca-bundle-image',
|
|
'--image', info.volume.image_path,
|
|
'--arch', arch,
|
|
'--user', info.credentials['user-id'],
|
|
'--privatekey', info.credentials['private-key'],
|
|
'--cert', info.credentials['certificate'],
|
|
'--ec2cert', cert_ec2,
|
|
'--destination', info._ec2['bundle_path'],
|
|
'--prefix', info._ec2['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._ec2['bundle_path'], info._ec2['ami_name'] + '.manifest.xml')
|
|
if info._ec2['region'] == 'us-east-1':
|
|
s3_url = 'https://s3.amazonaws.com/'
|
|
elif info._ec2['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._ec2['region'])
|
|
info._ec2['manifest_location'] = info.manifest.provider['bucket'] + '/' + info._ec2['ami_name'] + '.manifest.xml'
|
|
log_check_call(['euca-upload-bundle',
|
|
'--bucket', info.manifest.provider['bucket'],
|
|
'--manifest', manifest_file,
|
|
'--access-key', info.credentials['access-key'],
|
|
'--secret-key', info.credentials['secret-key'],
|
|
'--url', s3_url,
|
|
'--region', info._ec2['region']])
|
|
|
|
|
|
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._ec2['bundle_path'])
|
|
del info._ec2['bundle_path']
|
|
|
|
|
|
class RegisterAMI(Task):
|
|
description = 'Registering the image as an AMI'
|
|
phase = phases.image_registration
|
|
predecessors = [Snapshot, UploadImage]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
registration_params = {'name': info._ec2['ami_name'],
|
|
'description': info._ec2['ami_description']}
|
|
registration_params['architecture'] = {'i386': 'i386',
|
|
'amd64': 'x86_64'}.get(info.manifest.system['architecture'])
|
|
|
|
if info.manifest.volume['backing'] == 's3':
|
|
registration_params['image_location'] = info._ec2['manifest_location']
|
|
else:
|
|
root_dev_name = {'pvm': '/dev/sda',
|
|
'hvm': '/dev/xvda'}.get(info.manifest.provider['virtualization'])
|
|
registration_params['root_device_name'] = root_dev_name
|
|
|
|
from boto.ec2.blockdevicemapping import BlockDeviceType
|
|
from boto.ec2.blockdevicemapping import BlockDeviceMapping
|
|
block_device = BlockDeviceType(snapshot_id=info._ec2['snapshot'].id, delete_on_termination=True,
|
|
size=info.volume.size.bytes.get_qty_in('GiB'), volume_type='gp2')
|
|
registration_params['block_device_map'] = BlockDeviceMapping()
|
|
registration_params['block_device_map'][root_dev_name] = block_device
|
|
|
|
if info.manifest.provider['virtualization'] == 'hvm':
|
|
registration_params['virtualization_type'] = 'hvm'
|
|
else:
|
|
registration_params['virtualization_type'] = 'paravirtual'
|
|
akis_path = os.path.join(os.path.dirname(__file__), 'ami-akis.yml')
|
|
from bootstrapvz.common.tools import config_get
|
|
registration_params['kernel_id'] = config_get(akis_path, [info._ec2['region'],
|
|
info.manifest.system['architecture']])
|
|
|
|
if info.manifest.provider.get('enhanced_networking', None) == 'simple':
|
|
registration_params['sriov_net_support'] = 'simple'
|
|
|
|
info._ec2['image'] = info._ec2['connection'].register_image(**registration_params)
|