67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# encoding: utf-8
|
||
|
|
# === This file is part of Calamares - <http://github.com/calamares> ===
|
||
|
|
#
|
||
|
|
# Copyright 2014, Philip Müller <philm@manjaro.org>
|
||
|
|
#
|
||
|
|
# Calamares is free software: you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU General Public License as published by
|
||
|
|
# the Free Software Foundation, either version 3 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
#
|
||
|
|
# Calamares is distributed in the hope that it will be useful,
|
||
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
# GNU General Public License for more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU General Public License
|
||
|
|
# along with Calamares. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
import libcalamares
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def modify_grub_default(partitions, root_mount_point, distributor):
|
||
|
|
|
||
|
|
default_dir = os.path.join(root_mount_point, "etc/default")
|
||
|
|
default_grub = os.path.join(default_dir, "grub")
|
||
|
|
plymouth_bin = os.path.join(root_mount_point, "usr/bin/plymouth")
|
||
|
|
use_splash = ''
|
||
|
|
|
||
|
|
if os.path.exists(plymouth_bin):
|
||
|
|
use_splash = 'splash'
|
||
|
|
|
||
|
|
for partition in partitions:
|
||
|
|
if partition["fs"] == "linuxswap":
|
||
|
|
swap_uuid = partition["uuid"]
|
||
|
|
|
||
|
|
if swap_uuid != "":
|
||
|
|
kernel_cmd = 'GRUB_CMDLINE_LINUX_DEFAULT="resume=UUID=%s quiet %s"' % (swap_uuid, use_splash)
|
||
|
|
else:
|
||
|
|
kernel_cmd = 'GRUB_CMDLINE_LINUX_DEFAULT="quiet %s"' % use_splash
|
||
|
|
|
||
|
|
if not os.path.exists(default_dir):
|
||
|
|
os.mkdir(default_dir)
|
||
|
|
|
||
|
|
with open(default_grub, 'r') as grub_file:
|
||
|
|
lines = [x.strip() for x in grub_file.readlines()]
|
||
|
|
|
||
|
|
for i in range(len(lines)):
|
||
|
|
if lines[i].startswith("#GRUB_CMDLINE_LINUX_DEFAULT"):
|
||
|
|
lines[i] = kernel_cmd
|
||
|
|
elif lines[i].startswith("GRUB_CMDLINE_LINUX_DEFAULT"):
|
||
|
|
lines[i] = kernel_cmd
|
||
|
|
elif lines[i].startswith("#GRUB_DISTRIBUTOR") or lines[i].startswith("GRUB_DISTRIBUTOR"):
|
||
|
|
lines[i] = "GRUB_DISTRIBUTOR=%s" % distributor
|
||
|
|
|
||
|
|
with open(default_grub, 'w') as grub_file:
|
||
|
|
grub_file.write("\n".join(lines) + "\n")
|
||
|
|
|
||
|
|
|
||
|
|
def run():
|
||
|
|
partitions = libcalamares.globalstorage.value("partitions")
|
||
|
|
root_mount_point = libcalamares.globalstorage.value("rootMountPoint")
|
||
|
|
distributor = libcalamares.job.configuration["distributor"]
|
||
|
|
modify_grub_default(partitions, root_mount_point, distributor)
|
||
|
|
return None
|