mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 18:00:35 +00:00
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
from bootstrapvz.common.tasks import apt
|
|
from bootstrapvz.common.tasks import bootstrap
|
|
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['foldermounts'] = os.path.join(info.workspace, 'minimize_size')
|
|
os.mkdir(info._minimize_size['foldermounts'])
|
|
for folder in folders:
|
|
temp_path = os.path.join(info._minimize_size['foldermounts'], folder.replace('/', '_'))
|
|
os.mkdir(temp_path)
|
|
|
|
full_path = os.path.join(info.root, folder)
|
|
os.chmod(temp_path, os.stat(full_path).st_mode)
|
|
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['foldermounts'], 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['foldermounts'])
|
|
del info._minimize_size['foldermounts']
|