mirror of https://github.com/cutefishos/calamares
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
# encoding: utf-8
|
|
# === This file is part of Calamares - <http://github.com/calamares> ===
|
|
#
|
|
# Copyright 2014, Aurélien Gâteau <agateau@kde.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 os
|
|
import subprocess
|
|
|
|
import libcalamares
|
|
|
|
|
|
# FIXME: Duplicated between mount and grub
|
|
def mount(device_path, mount_point, fs=None, options=None):
|
|
assert device_path
|
|
assert mount_point
|
|
if not os.path.exists(mount_point):
|
|
os.makedirs(mount_point)
|
|
cmd = ["mount", device_path, mount_point]
|
|
if fs:
|
|
cmd += ("-t", fs)
|
|
if options:
|
|
cmd += ("-o", options)
|
|
subprocess.check_call(cmd)
|
|
|
|
|
|
def mount_partitions(root_mount_point, partitions):
|
|
for partition in partitions:
|
|
if not partition["mountPoint"]:
|
|
continue
|
|
# Create mount point with `+` rather than `os.path.join()` because
|
|
# `partition["mountPoint"]` starts with a '/'.
|
|
mount_point = root_mount_point + partition["mountPoint"]
|
|
mount(partition["device"], mount_point,
|
|
fs=partition.get("fs"),
|
|
options=partition.get("options")
|
|
)
|
|
|
|
|
|
def umount_partitions(root_mount_point, partitions):
|
|
for partition in partitions:
|
|
if not partition["mountPoint"]:
|
|
continue
|
|
subprocess.call(["umount", root_mount_point + partition["mountPoint"]])
|
|
|
|
|
|
def chroot_call(root_mount_point, cmd):
|
|
subprocess.check_call(["chroot", root_mount_point] + cmd)
|
|
|
|
|
|
def install_grub(root_mount_point, boot_loader):
|
|
install_path = boot_loader["installPath"]
|
|
chroot_call(root_mount_point, ["grub-install", install_path])
|
|
chroot_call(root_mount_point, ["grub-mkconfig", "-o", "/boot/grub/grub.cfg"])
|
|
|
|
|
|
def run():
|
|
root_mount_point = libcalamares.globalstorage.value("rootMountPoint")
|
|
boot_loader = libcalamares.globalstorage.value("bootLoader")
|
|
extra_mounts = libcalamares.job.configuration["extraMounts"]
|
|
mount_partitions(root_mount_point, extra_mounts)
|
|
try:
|
|
install_grub(root_mount_point, boot_loader)
|
|
finally:
|
|
umount_partitions(root_mount_point, extra_mounts)
|
|
return None
|