mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +00:00
f62c8ade99
Up until now I didn't see the point of using spaces for indentation. However, the previous commit (a18bec3) was quite eye opening. Given that python is an indentation aware language, the amount of mistakes that went unnoticed because tabs and spaces were used at the same time (tabs for indentation and spaces for alignment) were unacceptable. E101,W191 have been re-enable in the tox flake8 checker and the documentation has been modified accordingly. The following files have been left as-is: * bootstrapvz/common/assets/extlinux/extlinux.conf * bootstrapvz/common/assets/init.d/expand-root * bootstrapvz/common/assets/init.d/generate-ssh-hostkeys * bootstrapvz/common/assets/init.d/squeeze/generate-ssh-hostkeys * bootstrapvz/plugins/docker_daemon/assets/init.d/docker * bootstrapvz/providers/ec2/assets/bin/growpart * bootstrapvz/providers/ec2/assets/grub.d/40_custom * bootstrapvz/providers/ec2/assets/init.d/ec2-get-credentials * bootstrapvz/providers/ec2/assets/init.d/ec2-run-user-data * docs/_static/taskoverview.coffee * docs/_static/taskoverview.less * tests/unit/subprocess.sh
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
from bootstrapvz.base import Task
|
|
from .. import phases
|
|
from ..tasks import packages
|
|
import logging
|
|
|
|
|
|
class AddDKMSPackages(Task):
|
|
description = 'Adding DKMS and kernel header packages'
|
|
phase = phases.package_installation
|
|
successors = [packages.InstallPackages]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info.packages.add('dkms')
|
|
kernel_pkg_arch = {'i386': '686-pae', 'amd64': 'amd64'}[info.manifest.system['architecture']]
|
|
info.packages.add('linux-headers-' + kernel_pkg_arch)
|
|
|
|
|
|
class UpdateInitramfs(Task):
|
|
description = 'Rebuilding initramfs'
|
|
phase = phases.system_modification
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from bootstrapvz.common.tools import log_check_call
|
|
# Update initramfs (-u) for all currently installed kernel versions (-k all)
|
|
log_check_call(['chroot', info.root, 'update-initramfs', '-u', '-k', 'all'])
|
|
|
|
|
|
class DetermineKernelVersion(Task):
|
|
description = 'Determining kernel version'
|
|
phase = phases.package_installation
|
|
predecessors = [packages.InstallPackages]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
# Snatched from `extlinux-update' in wheezy
|
|
# list the files in boot/ that match vmlinuz-*
|
|
# sort what the * matches, the first entry is the kernel version
|
|
import os.path
|
|
import re
|
|
regexp = re.compile('^vmlinuz-(?P<version>.+)$')
|
|
|
|
def get_kernel_version(vmlinuz_path):
|
|
vmlinux_basename = os.path.basename(vmlinuz_path)
|
|
return regexp.match(vmlinux_basename).group('version')
|
|
from glob import glob
|
|
boot = os.path.join(info.root, 'boot')
|
|
vmlinuz_paths = glob('{boot}/vmlinuz-*'.format(boot=boot))
|
|
kernels = map(get_kernel_version, vmlinuz_paths)
|
|
info.kernel_version = sorted(kernels, reverse=True)[0]
|
|
logging.getLogger(__name__).debug('Kernel version is {version}'.format(version=info.kernel_version))
|