blob: 38254c99dc30bf6245262f755f42969fe00ff459 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from __future__ import annotations
import logging
import sys
from enum import Enum
from typing import List
from ..hardware import has_uefi
from ..output import log
class Bootloader(Enum):
Systemd = 'Systemd-boot'
Grub = 'Grub'
Efistub = 'Efistub'
def json(self):
return self.value
@classmethod
def values(cls) -> List[str]:
return [e.value for e in cls]
@classmethod
def get_default(cls) -> Bootloader:
if has_uefi():
return Bootloader.Systemd
else:
return Bootloader.Grub
@classmethod
def from_arg(cls, bootloader: str) -> Bootloader:
# to support old configuration files
bootloader = bootloader.capitalize()
if bootloader not in cls.values():
values = ', '.join(cls.values())
log(f'Invalid bootloader value "{bootloader}". Allowed values: {values}', level=logging.WARN)
sys.exit(1)
return Bootloader(bootloader)
|