mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
from base import Task
|
|
from common import phases
|
|
from common.tasks import apt
|
|
from common.fs.loopbackvolume import LoopbackVolume
|
|
|
|
|
|
class ConfigureGrub(Task):
|
|
description = 'Configuring grub'
|
|
phase = phases.system_modification
|
|
after = [apt.AptUpgrade]
|
|
|
|
def run(self, info):
|
|
import os
|
|
from common.tools import log_check_call
|
|
|
|
boot_dir = os.path.join(info.root, 'boot')
|
|
grub_dir = os.path.join(boot_dir, 'grub')
|
|
|
|
if isinstance(info.volume, LoopbackVolume):
|
|
# GRUB cannot deal with installing to loopback devices
|
|
# so we fake a real harddisk with dmsetup.
|
|
# Guide here: http://ebroder.net/2009/08/04/installing-grub-onto-a-disk-image/
|
|
info.volume.unmount()
|
|
info.volume.unmap()
|
|
info.volume.link_dm_node()
|
|
info.volume.map()
|
|
info.volume.mount_root(info.root)
|
|
info.volume.mount_boot()
|
|
info.volume.mount_specials()
|
|
[device_path] = log_check_call(['readlink', '-f', info.volume.device_path])
|
|
device_map_path = os.path.join(grub_dir, 'device.map')
|
|
with open(device_map_path, 'w') as device_map:
|
|
device_map.write('(hd0) {device_path}\n'.format(device_path=device_path))
|
|
|
|
# Install grub
|
|
log_check_call(['/usr/sbin/grub-install',
|
|
'--root-directory=' + info.root,
|
|
'--boot-directory=' + boot_dir,
|
|
device_path])
|
|
log_check_call(['/usr/sbin/chroot', info.root, '/usr/sbin/update-grub'])
|
|
# log_check_call(['/usr/sbin/chroot', info.root, '/usr/sbin/update-initramfs', '-u'])
|
|
|
|
if isinstance(info.volume, LoopbackVolume):
|
|
info.volume.unmount()
|
|
info.volume.unmap()
|
|
info.volume.unlink_dm_node()
|
|
info.volume.map()
|
|
info.volume.mount_root(info.root)
|
|
info.volume.mount_boot()
|
|
info.volume.mount_specials()
|
|
|
|
# Best guess right now...
|
|
device_map_path = os.path.join(grub_dir, 'device.map')
|
|
with open(device_map_path, 'w') as device_map:
|
|
device_map.write('(hd0) /dev/mapper/vda\n')
|