bootstrap-vz/bootstrapvz/common/tasks/cleanup.py
2014-05-03 16:13:07 +02:00

32 lines
720 B
Python

from bootstrapvz.base import Task
from .. import phases
import os
import shutil
class ClearMOTD(Task):
description = 'Clearing the MOTD'
phase = phases.system_cleaning
@classmethod
def run(cls, info):
with open('/var/run/motd', 'w'):
pass
class CleanTMP(Task):
description = 'Removing temporary files'
phase = phases.system_cleaning
@classmethod
def run(cls, info):
tmp = os.path.join(info.root, 'tmp')
for tmp_file in [os.path.join(tmp, f) for f in os.listdir(tmp)]:
if os.path.isfile(tmp_file):
os.remove(tmp_file)
else:
shutil.rmtree(tmp_file)
log = os.path.join(info.root, 'var/log/')
os.remove(os.path.join(log, 'bootstrap.log'))
os.remove(os.path.join(log, 'dpkg.log'))