mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
0092e1c2c2
In practice they are just typed functions with attributes, having a reference to an object is just confusing. So: Task.run() is now a classmethod
129 lines
4.1 KiB
Python
129 lines
4.1 KiB
Python
from base import Task
|
|
from common import phases
|
|
from common.tools import log_check_call
|
|
import locale
|
|
import os
|
|
|
|
|
|
class AddDefaultSources(Task):
|
|
description = 'Adding default release sources'
|
|
phase = phases.preparation
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
if info.source_lists.target_exists('{system.release}'):
|
|
import logging
|
|
msg = ('{system.release} target already exists').format(**info.manifest_vars)
|
|
logging.getLogger(__name__).info(msg)
|
|
else:
|
|
info.source_lists.add('main', 'deb {apt_mirror} {system.release} main')
|
|
info.source_lists.add('main', 'deb-src {apt_mirror} {system.release} main')
|
|
info.source_lists.add('main', 'deb {apt_mirror} {system.release}-updates main')
|
|
info.source_lists.add('main', 'deb-src {apt_mirror} {system.release}-updates main')
|
|
|
|
|
|
class WriteSources(Task):
|
|
description = 'Writing aptitude sources to disk'
|
|
phase = phases.package_installation
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
for name, sources in info.source_lists.sources.iteritems():
|
|
if name == 'main':
|
|
list_path = os.path.join(info.root, 'etc/apt/sources.list')
|
|
else:
|
|
list_path = os.path.join(info.root, 'etc/apt/sources.list.d/', name + '.list')
|
|
with open(list_path, 'w') as source_list:
|
|
for source in sources:
|
|
source_list.write('{line}\n'.format(line=str(source)))
|
|
|
|
|
|
class DisableDaemonAutostart(Task):
|
|
description = 'Disabling daemon autostart'
|
|
phase = phases.package_installation
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
rc_policy_path = os.path.join(info.root, 'usr/sbin/policy-rc.d')
|
|
with open(rc_policy_path, 'w') as rc_policy:
|
|
rc_policy.write(('#!/bin/sh\n'
|
|
'exit 101'))
|
|
import stat
|
|
os.chmod(rc_policy_path,
|
|
stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR |
|
|
stat.S_IRGRP | stat.S_IXGRP |
|
|
stat.S_IROTH | stat.S_IXOTH)
|
|
|
|
|
|
class AptUpdate(Task):
|
|
description = 'Updating the package cache'
|
|
phase = phases.package_installation
|
|
predecessors = [locale.GenerateLocale, WriteSources]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
log_check_call(['/usr/sbin/chroot', info.root,
|
|
'/usr/bin/apt-get', 'update'])
|
|
|
|
|
|
class AptUpgrade(Task):
|
|
description = 'Upgrading packages and fixing broken dependencies'
|
|
phase = phases.package_installation
|
|
predecessors = [AptUpdate, DisableDaemonAutostart]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from subprocess import CalledProcessError
|
|
try:
|
|
log_check_call(['/usr/sbin/chroot', info.root,
|
|
'/usr/bin/apt-get', 'install',
|
|
'--fix-broken',
|
|
'--no-install-recommends',
|
|
'--assume-yes'])
|
|
log_check_call(['/usr/sbin/chroot', info.root,
|
|
'/usr/bin/apt-get', 'upgrade',
|
|
'--no-install-recommends',
|
|
'--assume-yes'])
|
|
except CalledProcessError as e:
|
|
if e.returncode == 100:
|
|
import logging
|
|
msg = ('apt exited with status code 100. '
|
|
'This can sometimes occur when package retrieval times out or a package extraction failed. '
|
|
'apt might succeed if you try bootstrapping again.')
|
|
logging.getLogger(__name__).warn(msg)
|
|
raise e
|
|
|
|
|
|
class PurgeUnusedPackages(Task):
|
|
description = 'Removing unused packages'
|
|
phase = phases.system_cleaning
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
log_check_call(['/usr/sbin/chroot', info.root,
|
|
'/usr/bin/apt-get', 'autoremove',
|
|
'--purge'])
|
|
|
|
|
|
class AptClean(Task):
|
|
description = 'Clearing the aptitude cache'
|
|
phase = phases.system_cleaning
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
log_check_call(['/usr/sbin/chroot', info.root,
|
|
'/usr/bin/apt-get', 'clean'])
|
|
|
|
lists = os.path.join(info.root, 'var/lib/apt/lists')
|
|
for list_file in [os.path.join(lists, f) for f in os.listdir(lists)]:
|
|
if os.path.isfile(list_file):
|
|
os.remove(list_file)
|
|
|
|
|
|
class EnableDaemonAutostart(Task):
|
|
description = 'Re-enabling daemon autostart after installation'
|
|
phase = phases.system_cleaning
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
os.remove(os.path.join(info.root, 'usr/sbin/policy-rc.d'))
|