forked from Mirror/pmbootstrap
7a8deb0f5e
Add a new config section "mirrors", to replace the mirrors_alpine and mirrors_postmarketos options. This will allow for more flexibility since we can then handle the systemd staging repo (and others like plasma nightly) with relative ease. The loading/saving is fixed and now properly avoids writing out default values, this way if the defaults are changed the user won't be stuck with old values in their pmbootstrap.cfg. Signed-off-by: Caleb Connolly <caleb@postmarketos.org>
105 lines
3.7 KiB
Python
105 lines
3.7 KiB
Python
|
|
from copy import deepcopy
|
|
import multiprocessing
|
|
from typing import Any, List, Dict, TypedDict
|
|
from pathlib import Path
|
|
import os
|
|
|
|
class Mirrors(TypedDict):
|
|
alpine: str
|
|
pmaports: str
|
|
systemd: str
|
|
|
|
|
|
class Config():
|
|
aports: List[Path] = [Path(os.path.expanduser("~") +
|
|
"/.local/var/pmbootstrap/cache_git/pmaports")]
|
|
boot_size: int = 256
|
|
build_default_device_arch: bool = False
|
|
build_pkgs_on_install: bool = True
|
|
ccache_size: str = "5G" # yeahhhh this one has a suffix
|
|
device: str = "qemu-amd64"
|
|
extra_packages: str = "none"
|
|
extra_space: int = 0
|
|
hostname: str = ""
|
|
is_default_channel: bool = True
|
|
jobs: str = str(multiprocessing.cpu_count() + 1)
|
|
kernel: str = "stable"
|
|
keymap: str = ""
|
|
locale: str = "en_US.UTF-8"
|
|
mirrors: Mirrors = {
|
|
"alpine": "http://dl-cdn.alpinelinux.org/alpine/",
|
|
"pmaports": "http://mirror.postmarketos.org/postmarketos/",
|
|
"systemd": "http://mirror.postmarketos.org/postmarketos/staging/systemd/"
|
|
}
|
|
# NOTE: mirrors use http by default to leverage caching
|
|
mirror_alpine: str = "http://dl-cdn.alpinelinux.org/alpine/"
|
|
# NOTE: mirrors_postmarketos variable type is supposed to be
|
|
# comma-separated string, not a python list or any other type!
|
|
mirrors_postmarketos: List[str] = ["http://mirror.postmarketos.org/postmarketos/"]
|
|
qemu_redir_stdio: bool = False
|
|
ssh_key_glob: str = "~/.ssh/id_*.pub"
|
|
ssh_keys: bool = False
|
|
sudo_timer: bool = False
|
|
systemd: str = "default"
|
|
timezone: str = "GMT"
|
|
ui: str = "console"
|
|
ui_extras: bool = False
|
|
user: str = "user"
|
|
work: Path = Path(os.path.expanduser("~") + "/.local/var/pmbootstrap")
|
|
|
|
providers: Dict[str, str] = { }
|
|
|
|
|
|
def __init__(self):
|
|
# Make sure we aren't modifying the class defaults
|
|
for key in Config.__annotations__.keys():
|
|
setattr(self, key, deepcopy(Config.get_default(key)))
|
|
|
|
|
|
@staticmethod
|
|
def keys() -> List[str]:
|
|
keys = list(Config.__annotations__.keys())
|
|
keys.remove("mirrors")
|
|
keys += [f"mirrors.{k}" for k in Mirrors.__annotations__.keys()]
|
|
return sorted(keys)
|
|
|
|
|
|
@staticmethod
|
|
def get_default(dotted_key: str) -> Any:
|
|
"""Get the default value for a config option, supporting
|
|
nested dictionaries (e.g. "mirrors.alpine")."""
|
|
keys = dotted_key.split(".")
|
|
if len(keys) == 1:
|
|
return getattr(Config, keys[0])
|
|
elif len(keys) == 2:
|
|
return getattr(Config, keys[0])[keys[1]]
|
|
else:
|
|
raise ValueError(f"Invalid dotted key: {dotted_key}")
|
|
|
|
|
|
def __setattr__(self, key: str, value: str):
|
|
"""Allow for setattr() to be used with a dotted key
|
|
to set nested dictionaries (e.g. "mirrors.alpine")."""
|
|
keys = key.split(".")
|
|
if len(keys) == 1:
|
|
super(Config, self).__setattr__(key, value)
|
|
elif len(keys) == 2:
|
|
#print(f"cfgset, before: {super(Config, self).__getattribute__(keys[0])[keys[1]]}")
|
|
super(Config, self).__getattribute__(keys[0])[keys[1]] = value
|
|
#print(f"cfgset, after: {super(Config, self).__getattribute__(keys[0])[keys[1]]}")
|
|
else:
|
|
raise ValueError(f"Invalid dotted key: {key}")
|
|
|
|
|
|
def __getattribute__(self, key: str) -> str:
|
|
#print(repr(self))
|
|
"""Allow for getattr() to be used with a dotted key
|
|
to get nested dictionaries (e.g. "mirrors.alpine")."""
|
|
keys = key.split(".")
|
|
if len(keys) == 1:
|
|
return super(Config, self).__getattribute__(key)
|
|
elif len(keys) == 2:
|
|
return super(Config, self).__getattribute__(keys[0])[keys[1]]
|
|
else:
|
|
raise ValueError(f"Invalid dotted key: {key}")
|