wally/config.py

from typing import Any, Dict
from .storage import IStorable

ConfigBlock = Dict[str, Any]


class Config(IStorable):
    # make mypy happy
    run_uuid = None  # type: str
    storage_url = None  # type: str
    comment = None  # type: str
    keep_vm = None  # type: bool
    dont_discover_nodes = None  # type: bool
    build_id = None  # type: str
    build_description = None  # type: str
    build_type = None  # type: str
    default_test_local_folder = None  # type: str
    settings_dir = None  # type: str
    connect_timeout = 30  # type: int
    no_tests = False  # type: bool

    def __init__(self, dct: ConfigBlock) -> None:
        self.__dict__['_dct'] = dct

    def get(self, path: str, default: Any = None) -> Any:
        curr = self

        while path:
            if '/' in path:
                name, path = path.split('/', 1)
            else:
                name = path
                path = ""

            try:
                curr = getattr(curr, name)
            except AttributeError:
                return default

        return curr

    def __getattr__(self, name: str) -> Any:
        try:
            val = self.__dct[name]
        except KeyError:
            raise AttributeError(name)

        if isinstance(val, dict):
            val = self.__class__(val)

        return val

    def __setattr__(self, name: str, val: Any):
        self.__dct[name] = val

    def __contains__(self, name: str) -> bool:
        return self.get(name) is not None