mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
44 lines
1.3 KiB
Python
44 lines
1.3 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)
|
||
|
|
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']
|