mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
55 lines
2.4 KiB
Python
55 lines
2.4 KiB
Python
from base import Task
|
|
from common import phases
|
|
|
|
|
|
class ConfigureGrub(Task):
|
|
description = 'Configuring grub for KVM'
|
|
phase = phases.system_modification
|
|
|
|
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)
|
|
x_all = stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH
|
|
import os.path
|
|
device_map_path = os.path.join(info.root, 'boot/grub/device.map')
|
|
with open(device_map_path, 'w') as device_map:
|
|
device_map.write('(hd0) /dev/sda\n')
|
|
|
|
from common.tools import log_check_call
|
|
|
|
from shutil import copy
|
|
script_src = os.path.normpath(os.path.join(os.path.dirname(__file__), '../assets/grub.d/10_linux'))
|
|
script_dst = os.path.join(info.root, 'etc/grub.d/10_linux')
|
|
copy(script_src, script_dst)
|
|
os.chmod(script_dst, rwxr_xr_x)
|
|
script_src = os.path.normpath(os.path.join(os.path.dirname(__file__), '../assets/grub.d/00_header'))
|
|
script_dst = os.path.join(info.root, 'etc/grub.d/00_header')
|
|
copy(script_src, script_dst)
|
|
os.chmod(script_dst, rwxr_xr_x)
|
|
|
|
if info.manifest.virtualization == 'virtio':
|
|
print "Using virtio"
|
|
modules_path = os.path.join(info.root,
|
|
'etc/initramfs-tools/modules')
|
|
with open(modules_path, 'a') as modules:
|
|
modules.write("\nvirtio_pci\nvirtio_blk\n")
|
|
|
|
log_check_call(['/usr/sbin/chroot', info.root, '/usr/sbin/update-initramfs', '-u'])
|
|
# Install grub in mbr
|
|
log_check_call(['/usr/sbin/grub-install', '--boot-directory='+info.root+"/boot/", info.bootstrap_device['path']])
|
|
|
|
log_check_call(['/usr/sbin/chroot', info.root, '/usr/sbin/update-grub'])
|
|
|
|
if info.manifest.virtualization == 'virtio':
|
|
from common.tools import sed_i
|
|
grub_cfg = os.path.join(info.root, 'boot/grub/grub.cfg')
|
|
sed_i(grub_cfg, 'sda', 'vda')
|
|
device_map = os.path.join(info.root,
|
|
'boot/grub/device.map')
|
|
sed_i(device_map, 'sda', 'vda')
|
|
fstab_file = os.path.join(info.root,
|
|
'etc/fstab')
|
|
sed_i(fstab_file, 'sda', 'vda')
|
|
|