""" Standard providers. """ from collections import Iterable from .injections import InitArg, Attribute, Method class Provider(object): """ Base provider class. """ __is_objects_provider__ = True __overridden_by__ = list() def __init__(self): """ Initializer. """ self.__overridden_by__ = list() def __call__(self, *args, **kwargs): """ Returns provided instance. """ raise NotImplementedError() def __override__(self, provider): """ Overrides provider with another provider. """ self.__overridden_by__.append(provider) def prepare_injections(injections): """ Prepares injections list to injection. """ return [(injection.name, injection.value) for injection in injections] def fetch_injections(injections, injection_type): """ Fetches injections of injection type from list. """ return tuple([injection for injection in injections if isinstance(injection, injection_type)]) class NewInstance(Provider): """ New instance providers will create and return new instance on every call. """ def __init__(self, provides, *injections): """ Initializer. """ self.provides = provides self.init_injections = fetch_injections(injections, InitArg) self.attribute_injections = fetch_injections(injections, Attribute) self.method_injections = fetch_injections(injections, Method) super(NewInstance, self).__init__() def __call__(self, *args, **kwargs): """ Returns provided instance. """ if self.__overridden_by__: return self.__overridden_by__[-1].__call__(*args, **kwargs) init_injections = prepare_injections(self.init_injections) init_injections = dict(init_injections) init_injections.update(kwargs) provided = self.provides(*args, **init_injections) attribute_injections = prepare_injections(self.attribute_injections) for name, injectable in attribute_injections: setattr(provided, name, injectable) method_injections = prepare_injections(self.method_injections) for name, injectable in method_injections: getattr(provided, name)(injectable) return provided class Singleton(NewInstance): """ Singleton provider will create instance once and return it on every call. """ def __init__(self, *args, **kwargs): """ Initializer. """ self.instance = None super(Singleton, self).__init__(*args, **kwargs) def __call__(self, *args, **kwargs): """ Returns provided instance. """ if not self.instance: self.instance = super(Singleton, self).__call__(*args, **kwargs) return self.instance class ExternalDependency(Provider): """ External dependency provider. """ def __init__(self, instance_of): """ Initializer :param instance_of: type """ if not isinstance(instance_of, Iterable): instance_of = (instance_of,) self.instance_of = instance_of self.dependency = None super(ExternalDependency, self).__init__() def satisfy(self, provider): """ Satisfies an external dependency. :param provider: Provider """ self.dependency = provider def __call__(self, *args, **kwargs): """ Returns provided instance. """ if not self.dependency: raise ValueError('Dependency is not satisfied') result = self.dependency.__call__(*args, **kwargs) if not any((isinstance(result, possible_type) for possible_type in self.instance_of)): raise TypeError('{} is not an instance of {}'.format(result, self.instance_of)) return result class _StaticProvider(Provider): """ Static provider is base implementation that provides exactly the same as it got on input. """ def __init__(self, provides): """ Initializer. """ self.provides = provides super(_StaticProvider, self).__init__() def __call__(self): """ Returns provided instance. """ if self.__overridden_by__: return self.__overridden_by__[-1].__call__() return self.provides class Class(_StaticProvider): """ Class provider provides class. """ class Object(_StaticProvider): """ Object provider provides object. """ class Function(_StaticProvider): """ Function provider provides function. """ class Value(_StaticProvider): """ Value provider provides value. """