music-kraken-core/src/music_kraken/utils/config/config.py

from typing import Any, Tuple, Union
from pathlib import Path
import logging

import toml

from .attributes.attribute import Attribute, Description, EmptyLine


class ConfigDict(dict):
    def __init__(self, config_reference: "Config", *args, **kwargs):
        self.config_reference: Config = config_reference

        super().__init__(*args, **kwargs)

    def __getattribute__(self, __name: str) -> Any:
        return super().__getattribute__(__name)
    
    def __setitem__(self, __key: Any, __value: Any) -> None:
        attribute: Attribute = self.config_reference.attribute_map[__key]
        attribute.load_toml({attribute.name: __value})
        self.config_reference.write()

        return super().__setitem__(__key, attribute.value)


class Config:
    def __init__(self, componet_list: Tuple[Union[Attribute, Description, EmptyLine]], config_file: Path) -> None:
        self.config_file: Path = config_file

        self.component_list: Tuple[Union[Attribute, Description, EmptyLine]] = componet_list
        self.loaded_settings: dict = {}

        self.attribute_map = {}
        for component in self.component_list:
            if not isinstance(component, Attribute):
                continue
            
            self.attribute_map[component.name] = component
            self.loaded_settings[component.name] = component.value

    @property
    def toml_string(self):
        "\n\n".join(component.toml_string for component in self.component_list)

    def write(self):
        with self.config_file.open("w") as conf_file:
            conf_file.write(self.toml_string)

    def read(self):
        if not self.config_file.is_file():
            logging.info(f"Config file at '{self.config_file}' doesn't exist => generating")
            self.write()
            return
        
        toml_data = {}
        with self.config_file.open("r") as conf_file:
            toml_data = toml.load(conf_file)

        for component in self.component_list:
            if isinstance(component, Attribute):
                component.load_toml(toml_data, self.loaded_settings)