mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +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
95 lines
3.5 KiB
Python
95 lines
3.5 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 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)
|
|
|
|
|
|
def set_fs_states(volume):
|
|
volume.fsm.current = 'detached'
|
|
|
|
p_map = volume.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'
|