File size: 1,553 Bytes
21c7db9 | 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 | """Configuration loading."""
from __future__ import annotations
import os
from pathlib import Path
from typing import Any
import yaml
def _read_yaml(path: Path) -> dict[str, Any]:
if not path.exists():
return {}
with path.open("r", encoding="utf-8") as handle:
return yaml.safe_load(handle) or {}
def load_config(config_name: str = "base.yaml") -> dict[str, Any]:
root = Path(__file__).resolve().parents[2]
config_path = root / "configs" / config_name
return _read_yaml(config_path)
def load_project_env(path: Path | None = None) -> None:
"""Load simple KEY=VALUE pairs from .env without overriding shell env."""
root = Path(__file__).resolve().parents[2]
env_path = path or root / ".env"
if not env_path.exists():
return
for raw_line in env_path.read_text(encoding="utf-8").splitlines():
line = raw_line.strip()
if not line or line.startswith("#") or "=" not in line:
continue
key, value = line.split("=", 1)
key = key.strip()
if not key or key in os.environ:
continue
os.environ[key] = value.strip().strip('"').strip("'")
def env_bool(name: str, default: bool = False) -> bool:
raw = os.getenv(name)
if raw is None:
return default
return raw.strip().lower() in {"1", "true", "yes", "on"}
def env_int(name: str, default: int) -> int:
raw = os.getenv(name)
if raw is None:
return default
try:
return int(raw)
except ValueError:
return default
|