Arthur K.
bad4a355e8
* feat: lrclib and genius providers * revert `prepend_header` to true * change providers order --------- Co-authored-by: mrsobakin <68982655+mrsobakin@users.noreply.github.com>
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
from typing import Self
|
|
import os
|
|
import tomllib
|
|
|
|
from lrc_dl.logger import DefaultLogger
|
|
|
|
|
|
def _get_config_file() -> Path | None:
|
|
config_dir = os.environ.get("XDG_CONFIG_HOME")
|
|
|
|
if config_dir is None:
|
|
return None
|
|
|
|
return Path(config_dir) / "lrc-dl" / "config.toml"
|
|
|
|
|
|
CONFIG_PATH = _get_config_file()
|
|
|
|
|
|
@dataclass
|
|
class LyricsDlConfig:
|
|
order: list[str] = field(default_factory=lambda: ["kugou", "lrclib", "genius"])
|
|
delay: float | None = 10
|
|
prepend_header: bool = True
|
|
providers_configs: dict[str, dict] = field(default_factory=lambda: {})
|
|
|
|
@classmethod
|
|
def from_file(cls, path: Path) -> Self:
|
|
with open(path, "rb") as f:
|
|
config = tomllib.load(f)
|
|
|
|
cfg = {
|
|
"order": config["providers"].pop("order", None),
|
|
"delay": config["providers"].pop("delay", None),
|
|
"prepend_header": config["providers"].pop("prepend_header", None),
|
|
"providers_configs": config.get("providers"),
|
|
}
|
|
|
|
# Remove unset keys
|
|
cfg = {k: v for k, v in cfg.items() if v is not None}
|
|
|
|
return cls(**cfg)
|
|
|
|
@classmethod
|
|
def default(cls) -> Self:
|
|
try:
|
|
if CONFIG_PATH is not None:
|
|
return cls.from_file(CONFIG_PATH)
|
|
except FileNotFoundError:
|
|
DefaultLogger().warning(
|
|
f"Warning: Missing config file ({CONFIG_PATH})."
|
|
" Falling back to default parameters."
|
|
)
|
|
|
|
return cls()
|