From ad260fe709ae5d98ef4980aa4f3b1ad883a69611 Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Thu, 3 Sep 2020 17:56:54 -0400 Subject: [PATCH] Add container usage to the custom provider example --- examples/providers/custom_factory.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/examples/providers/custom_factory.py b/examples/providers/custom_factory.py index 706ed04c..c73ab041 100644 --- a/examples/providers/custom_factory.py +++ b/examples/providers/custom_factory.py @@ -1,6 +1,6 @@ """Custom provider example.""" -from dependency_injector import providers +from dependency_injector import containers, providers class CustomFactory(providers.Provider): @@ -29,14 +29,18 @@ class CustomFactory(providers.Provider): return self._factory(*args, **kwargs) -factory = CustomFactory(object) +class Container(containers.DeclarativeContainer): + + factory = CustomFactory(object) if __name__ == '__main__': - object1 = factory() + container = Container() + + object1 = container.factory() assert isinstance(object1, object) - object2 = factory() + object2 = container.factory() assert isinstance(object1, object) assert object1 is not object2