| 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 |