mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +00:00
123 lines
4.6 KiB
Python
123 lines
4.6 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
from bootstrapvz.common.tasks import volume
|
|
from bootstrapvz.common.tasks import packages
|
|
from bootstrapvz.providers.virtualbox.tasks import guest_additions
|
|
from bootstrapvz.providers.ec2.tasks import ebs
|
|
from bootstrapvz.common.fs import unmounted
|
|
from bootstrapvz.common.tools import log_check_call
|
|
from shutil import copyfile
|
|
import os.path
|
|
import time
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Snapshot(Task):
|
|
description = 'Creating a snapshot of the bootstrapped volume'
|
|
phase = phases.package_installation
|
|
predecessors = [packages.InstallPackages, guest_additions.InstallGuestAdditions]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
snapshot = None
|
|
with unmounted(info.volume):
|
|
snapshot = info.volume.snapshot()
|
|
msg = 'A snapshot of the bootstrapped volume was created. ID: ' + snapshot.id
|
|
log.info(msg)
|
|
|
|
|
|
class CreateFromSnapshot(Task):
|
|
description = 'Creating EBS volume from a snapshot'
|
|
phase = phases.volume_creation
|
|
successors = [ebs.Attach]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
snapshot = info.manifest.plugins['prebootstrapped']['snapshot']
|
|
ebs_volume = info._ec2['connection'].create_volume(info.volume.size.bytes.get_qty_in('GiB'),
|
|
info._ec2['host']['availabilityZone'],
|
|
snapshot=snapshot)
|
|
while ebs_volume.volume_state() != 'available':
|
|
time.sleep(5)
|
|
ebs_volume.update()
|
|
|
|
info.volume.volume = ebs_volume
|
|
set_fs_states(info.volume)
|
|
|
|
|
|
class CopyImage(Task):
|
|
description = 'Creating a snapshot of the bootstrapped volume'
|
|
phase = phases.package_installation
|
|
predecessors = [packages.InstallPackages, guest_additions.InstallGuestAdditions]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
loopback_backup_name = 'volume-{id}.{ext}.backup'.format(id=info.run_id, ext=info.volume.extension)
|
|
destination = os.path.join(info.manifest.bootstrapper['workspace'], loopback_backup_name)
|
|
|
|
with unmounted(info.volume):
|
|
copyfile(info.volume.image_path, destination)
|
|
msg = 'A copy of the bootstrapped volume was created. Path: ' + destination
|
|
log.info(msg)
|
|
|
|
|
|
class CreateFromImage(Task):
|
|
description = 'Creating loopback image from a copy'
|
|
phase = phases.volume_creation
|
|
successors = [volume.Attach]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info.volume.image_path = os.path.join(info.workspace, 'volume.' + info.volume.extension)
|
|
loopback_backup_path = info.manifest.plugins['prebootstrapped']['image']
|
|
copyfile(loopback_backup_path, info.volume.image_path)
|
|
|
|
set_fs_states(info.volume)
|
|
|
|
|
|
class CopyFolder(Task):
|
|
description = 'Creating a copy of the bootstrap folder'
|
|
phase = phases.package_installation
|
|
predecessors = [packages.InstallPackages, guest_additions.InstallGuestAdditions]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
folder_backup_name = '{id}.{ext}.backup'.format(id=info.run_id, ext=info.volume.extension)
|
|
destination = os.path.join(info.manifest.bootstrapper['workspace'], folder_backup_name)
|
|
log_check_call(['cp', '-a', info.volume.path, destination])
|
|
msg = 'A copy of the bootstrapped volume was created. Path: ' + destination
|
|
log.info(msg)
|
|
|
|
|
|
class CreateFromFolder(Task):
|
|
description = 'Creating bootstrap folder from a copy'
|
|
phase = phases.volume_creation
|
|
successors = [volume.Attach]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info.root = os.path.join(info.workspace, 'root')
|
|
log_check_call(['cp', '-a', info.manifest.plugins['prebootstrapped']['folder'], info.root])
|
|
info.volume.path = info.root
|
|
info.volume.fsm.current = 'attached'
|
|
|
|
|
|
def set_fs_states(vol):
|
|
vol.fsm.current = 'detached'
|
|
|
|
p_map = vol.partition_map
|
|
from bootstrapvz.base.fs.partitionmaps.none import NoPartitions
|
|
if not isinstance(p_map, NoPartitions):
|
|
p_map.fsm.current = 'unmapped'
|
|
|
|
from bootstrapvz.base.fs.partitions.unformatted import UnformattedPartition
|
|
from bootstrapvz.base.fs.partitions.single import SinglePartition
|
|
for partition in p_map.partitions:
|
|
if isinstance(partition, UnformattedPartition):
|
|
partition.fsm.current = 'unmapped'
|
|
continue
|
|
if isinstance(partition, SinglePartition):
|
|
partition.fsm.current = 'formatted'
|
|
continue
|
|
partition.fsm.current = 'unmapped_fmt'
|