mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2025-07-11 00:22:25 +03:00
Add overriding functionality on declarative container initialization
This commit is contained in:
parent
1115e783e0
commit
19aeb45c47
File diff suppressed because it is too large
Load Diff
|
@ -88,6 +88,18 @@ class DynamicContainer(object):
|
||||||
del self.providers[name]
|
del self.providers[name]
|
||||||
super(DynamicContainer, self).__delattr__(name)
|
super(DynamicContainer, self).__delattr__(name)
|
||||||
|
|
||||||
|
def set_providers(self, **providers):
|
||||||
|
"""Set container providers.
|
||||||
|
|
||||||
|
:param providers: Dictionary of providers
|
||||||
|
:type providers:
|
||||||
|
dict[str, :py:class:`dependency_injector.providers.Provider`]
|
||||||
|
|
||||||
|
:rtype: None
|
||||||
|
"""
|
||||||
|
for name, provider in six.iteritems(providers):
|
||||||
|
setattr(self, name, provider)
|
||||||
|
|
||||||
def override(self, object overriding):
|
def override(self, object overriding):
|
||||||
"""Override current container by overriding container.
|
"""Override current container by overriding container.
|
||||||
|
|
||||||
|
@ -111,6 +123,19 @@ class DynamicContainer(object):
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def override_providers(self, **overriding_providers):
|
||||||
|
"""Override container providers.
|
||||||
|
|
||||||
|
:param overriding_providers: Dictionary of providers
|
||||||
|
:type overriding_providers:
|
||||||
|
dict[str, :py:class:`dependency_injector.providers.Provider`]
|
||||||
|
|
||||||
|
:rtype: None
|
||||||
|
"""
|
||||||
|
for name, overriding_provider in six.iteritems(overriding_providers):
|
||||||
|
container_provider = getattr(self, name)
|
||||||
|
container_provider.override(overriding_provider)
|
||||||
|
|
||||||
def reset_last_overriding(self):
|
def reset_last_overriding(self):
|
||||||
"""Reset last overriding provider for each container providers.
|
"""Reset last overriding provider for each container providers.
|
||||||
|
|
||||||
|
@ -265,18 +290,16 @@ class DeclarativeContainer(object):
|
||||||
:type: tuple[:py:class:`DeclarativeContainer`]
|
:type: tuple[:py:class:`DeclarativeContainer`]
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __new__(cls, *args, **kwargs):
|
def __new__(cls, **overriding_providers):
|
||||||
"""Constructor.
|
"""Constructor.
|
||||||
|
|
||||||
:return: Dynamic container with copy of all providers.
|
:return: Dynamic container with copy of all providers.
|
||||||
:rtype: :py:class:`DynamicContainer`
|
:rtype: :py:class:`DynamicContainer`
|
||||||
"""
|
"""
|
||||||
container = cls.instance_type(*args, **kwargs)
|
container = cls.instance_type()
|
||||||
container.provider_type = cls.provider_type
|
container.provider_type = cls.provider_type
|
||||||
|
container.set_providers(**deepcopy(cls.providers))
|
||||||
for name, provider in six.iteritems(deepcopy(cls.providers)):
|
container.override_providers(**overriding_providers)
|
||||||
setattr(container, name, provider)
|
|
||||||
|
|
||||||
return container
|
return container
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
|
|
@ -265,3 +265,12 @@ class DeclarativeContainerTests(unittest.TestCase):
|
||||||
dict(Container1=Container.Container1,
|
dict(Container1=Container.Container1,
|
||||||
Container2=Container.Container2,
|
Container2=Container.Container2,
|
||||||
Container3=Container.Container3))
|
Container3=Container.Container3))
|
||||||
|
|
||||||
|
def test_init_with_overriding_providers(self):
|
||||||
|
p1 = providers.Provider()
|
||||||
|
p2 = providers.Provider()
|
||||||
|
|
||||||
|
container = ContainerA(p11=p1, p12=p2)
|
||||||
|
|
||||||
|
self.assertIs(container.p11.last_overriding, p1)
|
||||||
|
self.assertIs(container.p12.last_overriding, p2)
|
||||||
|
|
|
@ -75,6 +75,16 @@ class DeclarativeContainerInstanceTests(unittest.TestCase):
|
||||||
self.assertIs(ContainerA.provider_type,
|
self.assertIs(ContainerA.provider_type,
|
||||||
containers.DeclarativeContainer.provider_type)
|
containers.DeclarativeContainer.provider_type)
|
||||||
|
|
||||||
|
def test_set_providers(self):
|
||||||
|
p13 = providers.Provider()
|
||||||
|
p14 = providers.Provider()
|
||||||
|
container_a = ContainerA()
|
||||||
|
|
||||||
|
container_a.set_providers(p13=p13, p14=p14)
|
||||||
|
|
||||||
|
self.assertIs(container_a.p13, p13)
|
||||||
|
self.assertIs(container_a.p14, p14)
|
||||||
|
|
||||||
def test_override(self):
|
def test_override(self):
|
||||||
class _Container(containers.DeclarativeContainer):
|
class _Container(containers.DeclarativeContainer):
|
||||||
p11 = providers.Provider()
|
p11 = providers.Provider()
|
||||||
|
@ -108,6 +118,22 @@ class DeclarativeContainerInstanceTests(unittest.TestCase):
|
||||||
with self.assertRaises(errors.Error):
|
with self.assertRaises(errors.Error):
|
||||||
container.override(container)
|
container.override(container)
|
||||||
|
|
||||||
|
def test_override_providers(self):
|
||||||
|
p1 = providers.Provider()
|
||||||
|
p2 = providers.Provider()
|
||||||
|
container_a = ContainerA()
|
||||||
|
|
||||||
|
container_a.override_providers(p11=p1, p12=p2)
|
||||||
|
|
||||||
|
self.assertIs(container_a.p11.last_overriding, p1)
|
||||||
|
self.assertIs(container_a.p12.last_overriding, p2)
|
||||||
|
|
||||||
|
def test_override_providers_with_unknown_provider(self):
|
||||||
|
container_a = ContainerA()
|
||||||
|
|
||||||
|
with self.assertRaises(AttributeError):
|
||||||
|
container_a.override_providers(unknown=providers.Provider())
|
||||||
|
|
||||||
def test_reset_last_overridding(self):
|
def test_reset_last_overridding(self):
|
||||||
class _Container(containers.DeclarativeContainer):
|
class _Container(containers.DeclarativeContainer):
|
||||||
p11 = providers.Provider()
|
p11 = providers.Provider()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user