mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
d85ad7f598
It requires vmware workstation to be installed (and zerofree optionally, but highly recommended)
101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
from base import Task
|
|
from common import phases
|
|
from common.tasks import apt
|
|
from common.tasks import bootstrap
|
|
from common.tasks import filesystem
|
|
from common.tasks import partitioning
|
|
from common.tasks import volume
|
|
import os
|
|
|
|
folders = ['tmp', 'var/lib/apt/lists']
|
|
|
|
|
|
class AddFolderMounts(Task):
|
|
description = 'Mounting folders for writing temporary and cache data'
|
|
phase = phases.os_installation
|
|
predecessors = [bootstrap.Bootstrap]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info.minimize_size_folder = os.path.join(info.workspace, 'minimize_size')
|
|
os.mkdir(info.minimize_size_folder)
|
|
for folder in folders:
|
|
temp_path = os.path.join(info.minimize_size_folder, folder.replace('/', '_'))
|
|
os.mkdir(temp_path)
|
|
|
|
full_path = os.path.join(info.root, folder)
|
|
info.volume.partition_map.root.add_mount(temp_path, full_path, ['--bind'])
|
|
|
|
|
|
class RemoveFolderMounts(Task):
|
|
description = 'Removing folder mounts for temporary and cache data'
|
|
phase = phases.system_cleaning
|
|
successors = [apt.AptClean]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
import shutil
|
|
for folder in folders:
|
|
temp_path = os.path.join(info.minimize_size_folder, folder.replace('/', '_'))
|
|
full_path = os.path.join(info.root, folder)
|
|
|
|
info.volume.partition_map.root.remove_mount(full_path)
|
|
shutil.rmtree(temp_path)
|
|
|
|
os.rmdir(info.minimize_size_folder)
|
|
del info.minimize_size_folder
|
|
|
|
|
|
class CheckZerofreePath(Task):
|
|
description = 'Checking path to zerofree tool'
|
|
phase = phases.preparation
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from common.exceptions import TaskError
|
|
import os
|
|
zerofree = info.manifest.plugins['minimize_size']['zerofree']
|
|
if not os.path.isfile(zerofree):
|
|
raise TaskError('The path `{path}\' does not exist or is not a file'.format(path=zerofree))
|
|
if not os.access(zerofree, os.X_OK):
|
|
raise TaskError('The path `{path}\' is not executable'.format(path=zerofree))
|
|
|
|
|
|
# Get zerofree here: http://intgat.tigress.co.uk/rmy/uml/index.html
|
|
class Zerofree(Task):
|
|
description = 'Zeroing unused blocks on the volume'
|
|
phase = phases.volume_unmounting
|
|
predecessors = [filesystem.UnmountRoot, partitioning.UnmapPartitions]
|
|
successors = [volume.Detach]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from common.tools import log_check_call
|
|
zerofree = info.manifest.plugins['minimize_size']['zerofree']
|
|
log_check_call([zerofree, info.volume.device_path])
|
|
|
|
|
|
class CheckVMWareDMCommand(Task):
|
|
description = 'Checking path to vmware-vdiskmanager tool'
|
|
phase = phases.preparation
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from common.exceptions import TaskError
|
|
import os
|
|
vdiskmngr = '/usr/bin/vmware-vdiskmanager'
|
|
if not os.path.isfile(vdiskmngr):
|
|
raise TaskError('Unable to find vmware-vdiskmanager at `{path}\''.format(path=vdiskmngr))
|
|
if not os.access(vdiskmngr, os.X_OK):
|
|
raise TaskError('vmware-vdiskmanager at `{path}\' is not executable'.format(path=vdiskmngr))
|
|
|
|
|
|
class ShrinkVolume(Task):
|
|
description = 'Shrinking the volume'
|
|
phase = phases.volume_unmounting
|
|
predecessors = [volume.Detach]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from common.tools import log_check_call
|
|
log_check_call(['/usr/bin/vmware-vdiskmanager', '-k', info.volume.image_path])
|