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 __getitem__(self, __name: str) -> Any:
        return super().__getitem__(__name)
    
    def __setitem__(self, __key: Any, __value: Any, from_attribute: bool = False, is_parsed: bool = False) -> None:
        if not from_attribute:
            attribute: Attribute = self.config_reference.attribute_map[__key]
            if is_parsed:
                attribute.value = __value
            else:
                attribute.parse(__value)
            self.config_reference.write()

            __value = attribute.value

        return super().__setitem__(__key, __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: ConfigDict = ConfigDict(self)

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

    @property
    def toml_string(self):
        return "\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, encoding="utf-8")

    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)