mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
from base import Task
|
|
from common import phases
|
|
from common.tools import log_check_call
|
|
import os.path
|
|
|
|
|
|
class ResolveInitScripts(Task):
|
|
description = 'Determining which startup scripts to install or disable'
|
|
phase = phases.system_modification
|
|
|
|
def run(self, info):
|
|
init_scripts = {}
|
|
init_scripts['expand-volume'] = 'expand-volume'
|
|
|
|
from subprocess import CalledProcessError
|
|
try:
|
|
log_check_call(['/usr/sbin/chroot', info.root,
|
|
'/usr/bin/dpkg-query', '-W', 'openssh-server'])
|
|
init_scripts['generate-ssh-hostkeys'] = 'generate-ssh-hostkeys'
|
|
if info.manifest.system['release'] == 'squeeze':
|
|
init_scripts['generate-ssh-hostkeys'] = 'squeeze/generate-ssh-hostkeys'
|
|
except CalledProcessError:
|
|
pass
|
|
|
|
disable_scripts = ['hwclock.sh']
|
|
if info.manifest.system['release'] == 'squeeze':
|
|
disable_scripts.append('hwclockfirst.sh')
|
|
|
|
init_scripts_dir = os.path.normpath(os.path.join(os.path.dirname(__file__), '../assets/init.d'))
|
|
for name, path in init_scripts.iteritems():
|
|
init_scripts[name] = os.path.join(init_scripts_dir, path)
|
|
|
|
info.initd = {'install': init_scripts,
|
|
'disable': disable_scripts}
|
|
|
|
|
|
class InstallInitScripts(Task):
|
|
description = 'Installing startup scripts'
|
|
phase = phases.system_modification
|
|
predecessors = [ResolveInitScripts]
|
|
|
|
def run(self, info):
|
|
import stat
|
|
rwxr_xr_x = (stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR |
|
|
stat.S_IRGRP | stat.S_IXGRP |
|
|
stat.S_IROTH | stat.S_IXOTH)
|
|
from shutil import copy
|
|
for name, src in info.initd['install'].iteritems():
|
|
dst = os.path.join(info.root, 'etc/init.d', name)
|
|
copy(src, dst)
|
|
os.chmod(dst, rwxr_xr_x)
|
|
log_check_call(['/usr/sbin/chroot', info.root, '/sbin/insserv', '--default', name])
|
|
|
|
for name in info.initd['disable']:
|
|
log_check_call(['/usr/sbin/chroot', info.root, '/sbin/insserv', '--remove', name])
|