mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from bootstrapvz.base import Task
|
|
from .. import phases
|
|
import host
|
|
import volume
|
|
|
|
|
|
class AddRequiredCommands(Task):
|
|
description = 'Adding commands required for creating loopback volumes'
|
|
phase = phases.preparation
|
|
successors = [host.CheckExternalCommands]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from ..fs.loopbackvolume import LoopbackVolume
|
|
if isinstance(info.volume, LoopbackVolume):
|
|
info.host_dependencies['qemu-img'] = 'qemu-utils'
|
|
info.host_dependencies['losetup'] = 'mount'
|
|
from ..fs.qemuvolume import QEMUVolume
|
|
if isinstance(info.volume, QEMUVolume):
|
|
info.host_dependencies['losetup'] = 'mount'
|
|
|
|
|
|
class Create(Task):
|
|
description = 'Creating a loopback volume'
|
|
phase = phases.volume_creation
|
|
successors = [volume.Attach]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
import os.path
|
|
image_path = os.path.join(info.workspace, 'volume.{ext}'.format(ext=info.volume.extension))
|
|
info.volume.create(image_path)
|
|
|
|
|
|
class MoveImage(Task):
|
|
description = 'Moving volume image'
|
|
phase = phases.image_registration
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
image_name = info.manifest.image['name'].format(**info.manifest_vars)
|
|
filename = '{image_name}.{ext}'.format(image_name=image_name, ext=info.volume.extension)
|
|
|
|
import os.path
|
|
destination = os.path.join(info.manifest.bootstrapper['workspace'], filename)
|
|
import shutil
|
|
shutil.move(info.volume.image_path, destination)
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
log.info('The volume image has been moved to {image_path}'.format(image_path=destination))
|