File size: 1,683 Bytes
026f63e | 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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | """YAML configs with `extends:` inheritance and dotted CLI overrides."""
from __future__ import annotations
import copy
from pathlib import Path
from typing import Any
import yaml
def deep_merge(base: dict, override: dict) -> dict:
out = copy.deepcopy(base)
for k, v in override.items():
if isinstance(v, dict) and isinstance(out.get(k), dict):
out[k] = deep_merge(out[k], v)
else:
out[k] = copy.deepcopy(v)
return out
def _load(path: Path) -> dict:
cfg = yaml.safe_load(path.read_text()) or {}
parent = cfg.pop("extends", None)
return deep_merge(_load(path.parent / parent), cfg) if parent else cfg
def load_config(path: str | Path, overrides: list[str] | None = None) -> dict:
path = Path(path)
raw = yaml.safe_load(path.read_text()) or {}
cfg = _load(path)
cfg["name"] = raw.get("name", path.stem)
for ov in overrides or []:
key, _, val = ov.partition("=")
set_dotted(cfg, key.strip(), _parse_value(val))
return cfg
def _parse_value(val: str) -> Any:
v = yaml.safe_load(val)
if isinstance(v, str): # YAML 1.1 reads "3e-4" as a string
try:
return float(v)
except ValueError:
pass
return v
def set_dotted(cfg: dict, key: str, value: Any) -> None:
node = cfg
parts = key.split(".")
for p in parts[:-1]:
node = node.setdefault(p, {})
node[parts[-1]] = value
def get_dotted(cfg: dict, key: str, default: Any = None) -> Any:
node = cfg
for p in key.split("."):
if not isinstance(node, dict) or p not in node:
return default
node = node[p]
return node
|