from dataclasses import dataclass
import toml
from models.DatabaseConfig import DatabaseConfig
from models.KeePassConfig import KeePassConfig


@dataclass
class Config:
    _config: dict
    _kee_pass_config: KeePassConfig
    _database_config: DatabaseConfig

    def __init__(self):
        with open('config.toml', 'r') as f:
            self._config: dict = toml.load(f)
            self._kee_pass_config: KeePassConfig = KeePassConfig(self._config['keepass'])
            self._database_config: DatabaseConfig = DatabaseConfig(self._config['database'])

    @property
    def kee_pass(self) -> KeePassConfig:
        return self._kee_pass_config

    @property
    def database(self) -> DatabaseConfig:
        return self._database_config