mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 18:00:35 +00:00
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
|
|
|
|
class Create(Task):
|
|
description = 'Creating the EBS volume'
|
|
phase = phases.volume_creation
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
tags = []
|
|
|
|
# Setting up tags on the EBS volume
|
|
if 'tags' in info.manifest.data:
|
|
raw_tags = info.manifest.data['tags']
|
|
formatted_tags = {k: v.format(**info.manifest_vars) for k, v in raw_tags.items()}
|
|
tags = [{'Key': k, 'Value': v} for k, v in formatted_tags.items()]
|
|
|
|
# EBS volumes support encryption. KMS key id is optional and default key
|
|
# is used when it is not defined.
|
|
encrypted = info.manifest.data['provider'].get('encrypted', False)
|
|
kms_key_id = info.manifest.data['provider'].get('kms_key_id')
|
|
|
|
info.volume.create(info._ec2['connection'], info._ec2['host']['availabilityZone'], tags=tags, encrypted=encrypted, kms_key_id=kms_key_id)
|
|
|
|
|
|
class Attach(Task):
|
|
description = 'Attaching the volume'
|
|
phase = phases.volume_creation
|
|
predecessors = [Create]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info.volume.attach(info._ec2['host']['instanceId'])
|
|
|
|
|
|
class Snapshot(Task):
|
|
description = 'Creating a snapshot of the EBS volume'
|
|
phase = phases.image_registration
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info._ec2['snapshot'] = info.volume.snapshot()
|
|
|
|
# Setting up tags on the snapshot
|
|
if 'tags' in info.manifest.data:
|
|
raw_tags = info.manifest.data['tags']
|
|
formatted_tags = {k: v.format(**info.manifest_vars) for k, v in raw_tags.items()}
|
|
tags = [{'Key': k, 'Value': v} for k, v in formatted_tags.items()]
|
|
info._ec2['connection'].create_tags(Resources=[info._ec2['snapshot']],
|
|
Tags=tags)
|