Source code for pymlconf.models

import abc
import copy
from collections import OrderedDict, Iterable
from os import path

from .errors import ConfigurationAlreadyInitializedError, \
from .yamling import load_string, load_yaml

[docs]def isiterable(o): if isinstance(o, (bytes, str, type)): return False return isinstance(o, Iterable)
[docs]class Mergable(metaclass=abc.ABCMeta): """Base class for all configuration nodes, so all configuration nodes are mergable """ def __init__(self, data=None, context=None): """ :param data: Initial value to constract a mergable instance. It can be ``yaml string`` or python dictionary. default: None. :type data: list or dict :param context: dictionary to format the yaml before parsing in pre-processor. :type context: dict """ self.context = context if context else {} if data: self.merge(data)
[docs] @abc.abstractmethod def can_merge(self, data): # pragma: no cover """ Determines whenever can merge with the passed argument or not. :param data: An object to test. :type data: any :returns: bool """ raise NotImplementedError()
@abc.abstractmethod def _merge(self, data): # pragma: no cover raise NotImplementedError()
[docs] @abc.abstractmethod def copy(self): # pragma: no cover """ When implemented, returns copy of current config instance. :returns: :class:`.Mergable` """ raise NotImplementedError()
[docs] @classmethod def make_mergable_if_possible(cls, data, context): """ Makes an object mergable if possible. Returns the virgin object if cannot convert it to a mergable instance. :returns: :class:`.Mergable` or type(data) """ if isinstance(data, dict): return MergableDict(data=data, context=context) elif isiterable(data): return MergableList( data=[cls.make_mergable_if_possible(i, context) for i in data], context=context ) else: return data
[docs] def merge(self, *args): """ Merges this instance with new instances, in-place. :param \*args: Configuration values to merge with current instance. :type \*args: iterable """ for data in args: if isinstance(data, str): to_merge = load_string(data, self.context) if not to_merge: continue else: to_merge = data if not self.can_merge(to_merge): raise TypeError( 'Cannot merge myself:%s with %s. data: %s' \ % (type(self), type(data), data) ) self._merge(to_merge)
[docs]class MergableDict(OrderedDict, Mergable): """ Configuration node that represents python dictionary data. """ def __init__(self, *args, **kwargs): OrderedDict.__init__(self) Mergable.__init__(self, *args, **kwargs)
[docs] def can_merge(self, data): return data is not None and isinstance(data, dict)
def _merge(self, data): for k in list(data.keys()): v = data[k] if k in self \ and isinstance(self[k], Mergable) \ and self[k].can_merge(v): self[k].merge(v) else: if isinstance(v, Mergable): self[k] = v.copy() else: self[k] = Mergable.make_mergable_if_possible( copy.deepcopy(v), self.context ) def __getattr__(self, key): if key in self: return self.get(key) raise AttributeError(key) def __setattr__(self, key, value): if key not in self: self.__dict__[key] = value else: self[key] = value
[docs] def copy(self): return MergableDict(self, context=self.context)
[docs]class ConfigurationNamespace(MergableDict): """ Configuration node that represents the configuration namespace node. """
[docs]class MergableList(list, Mergable): """ Configuration node that represents the python list data. """ def __init__(self, *args, **kwargs): list.__init__(self) Mergable.__init__(self, *args, **kwargs)
[docs] def can_merge(self, data): return data and hasattr(data, '__iter__')
def _merge(self, data): del self[:] self.extend(data)
[docs] def copy(self): return MergableList(self, context=self.context)
[docs]class Root(MergableDict): """ The main class which exposes pymlconf package. Example:: from pymlconf import Root from os import path config = Root(''' server: host: localhost port: 4455 ''') print print config.server.port """
[docs] def load_file(self, filename): """ load file which contains yaml configuration entries.and merge it by current instance :param files: files to load and merge into existing configuration instance :type files: list """ if not path.exists(filename): raise FileNotFoundError(filename) loaded_yaml = load_yaml(filename, self.context) if loaded_yaml: self.merge(loaded_yaml)
[docs]class DeferredRoot: _instance = None def __getattr__(self, key): return getattr( object.__getattribute__(self, '__class__')._get_instance(), key ) def __setattr__(self, key, value): setattr( object.__getattribute__(self, '__class__')._get_instance(), key, value ) @classmethod def _get_instance(cls): if cls._instance is None: raise ConfigurationNotInitializedError( 'Configuration manager object is not initialized yet.' ) return cls._instance
[docs] def initialize(self, *args, force=False, **kw): """ Initialize the configuration manager :param force: force initialization even if it's already initialized :return: """ if not force and self._instance is not None: raise ConfigurationAlreadyInitializedError( 'Configuration manager object is already initialized.' ) self.__class__._instance = Root(*args, **kw)