finance-watcher / helpers / config.py
config.py
Raw
import yaml

DEFAULT_CONFIG_PATH = 'config.yaml'

class Config:
    """
    Class that holds the database configuration
    """
    def __init__(self, file_path: str = DEFAULT_CONFIG_PATH):
         with open(file_path, 'r') as yaml_file:
            data_loaded = yaml.safe_load(yaml_file)

            database_config = data_loaded.get('finance_db')
            self.host = database_config.get('host')
            self.port = database_config.get('port')
            self.database = database_config.get('database')
            self.user = database_config.get('user')
            self.password = database_config.get('password')