bootstrap-vz/bootstrapvz/providers/ec2/tasks/ebs.py
2017-01-19 18:53:26 +00:00

37 lines
1.1 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):
info.volume.create(info._ec2['connection'], info._ec2['host']['availabilityZone'])
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']
# tags = {k: v.format(**info.manifest_vars) for k, v in raw_tags.items()}
# info._ec2['connection'](Resources=[info._ec2['snapshot']],
# Tags=[tags])