config.py

#!/usr/bin/python3 -u
"""
Configuration for various MantisBT utility scripts
"""
from collections.abc import Mapping
from pathlib import Path
import yaml


# Constants
ORG_MANTIS = 'mantisbt'
ORG_PLUGINS = 'mantisbt-plugins'


class Config(dict):
    """
    The Config class extends dict wrapper allowing its keys to be accessed as
    attributes for convenience.
    """
    def __getattr__(self, name):
        if name in self:
            return self[name]
        raise AttributeError(name)


def __update(base, upd):
    """
    Recursively update 'base' dict with corresponding values in 'upd'
    """
    if (isinstance(base, Mapping) and
            isinstance(upd, Mapping)):
        for key, value in upd.items():
            base[key] = __update(base.get(key, {}), value)
    elif upd is not None:
        return upd
    return base


def __read_config():
    """
    Read YML config files and return the data
    """
    path = Path(__file__).parent.resolve()

    # Read default configuration values
    with open(Path.joinpath(path, "config_defaults.yml"), 'r') as ymlfile:
        cfg = yaml.safe_load(ymlfile)

    # Read local settings
    try:
        with open(Path.joinpath(path, "config.yml"), 'r') as ymlfile:
            cfg = __update(cfg, yaml.safe_load(ymlfile))
    except FileNotFoundError:
        pass

    return Config(cfg)


cfg = __read_config()