mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-22 09:36:48 +03:00
adding scoped provider with examples
This commit is contained in:
parent
1631978790
commit
1cdf393f56
67
examples/scoped_provider.py
Normal file
67
examples/scoped_provider.py
Normal file
|
@ -0,0 +1,67 @@
|
|||
"""
|
||||
Scoped provider examples.
|
||||
"""
|
||||
|
||||
from objects import AbstractCatalog
|
||||
from objects.providers import Singleton, Scoped
|
||||
from objects.injections import InitArg, Attribute
|
||||
import sqlite3
|
||||
|
||||
|
||||
class ObjectA(object):
|
||||
def __init__(self, db):
|
||||
self.db = db
|
||||
|
||||
|
||||
# Catalog of objects providers.
|
||||
class Catalog(AbstractCatalog):
|
||||
"""
|
||||
Objects catalog.
|
||||
"""
|
||||
|
||||
database = Singleton(sqlite3.Connection,
|
||||
InitArg('database', ':memory:'),
|
||||
Attribute('row_factory', sqlite3.Row))
|
||||
""" :type: (objects.Provider) -> sqlite3.Connection """
|
||||
|
||||
object_a = Scoped(ObjectA,
|
||||
InitArg('db', database))
|
||||
""" :type: (objects.Provider) -> ObjectA """
|
||||
|
||||
|
||||
with Catalog.object_a as object_a_provider:
|
||||
object_a1 = object_a_provider()
|
||||
object_a2 = object_a_provider()
|
||||
|
||||
assert object_a1 is object_a2
|
||||
assert object_a1.db is object_a2.db
|
||||
|
||||
with Catalog.object_a as object_a_provider:
|
||||
object_a3 = object_a_provider()
|
||||
object_a4 = object_a_provider()
|
||||
|
||||
assert object_a3 is object_a4
|
||||
assert object_a3.db is object_a4.db
|
||||
|
||||
assert (object_a1 is not object_a3) and \
|
||||
(object_a1 is not object_a4)
|
||||
assert (object_a2 is not object_a3) and \
|
||||
(object_a2 is not object_a4)
|
||||
|
||||
|
||||
Catalog.object_a.in_scope()
|
||||
|
||||
object_a5 = Catalog.object_a()
|
||||
object_a6 = Catalog.object_a()
|
||||
|
||||
assert object_a5 is object_a6
|
||||
assert object_a5.db is object_a6.db
|
||||
|
||||
assert (object_a1 is not object_a3) and \
|
||||
(object_a1 is not object_a4) and \
|
||||
(object_a1 is not object_a5) and \
|
||||
(object_a1 is not object_a6)
|
||||
assert (object_a2 is not object_a3) and \
|
||||
(object_a2 is not object_a4) and \
|
||||
(object_a2 is not object_a5) and \
|
||||
(object_a2 is not object_a6)
|
|
@ -108,6 +108,60 @@ class Singleton(NewInstance):
|
|||
self.instance = super(Singleton, self).__call__(*args, **kwargs)
|
||||
return self.instance
|
||||
|
||||
def _reset_instance(self):
|
||||
"""
|
||||
Resets instance.
|
||||
"""
|
||||
self.instance = None
|
||||
|
||||
|
||||
class Scoped(Singleton):
|
||||
"""
|
||||
Scoped provider will create instance once for every scope and return it on every call.
|
||||
"""
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
"""
|
||||
Initializer.
|
||||
"""
|
||||
self.is_in_scope = None
|
||||
super(Scoped, self).__init__(*args, **kwargs)
|
||||
|
||||
def in_scope(self):
|
||||
"""
|
||||
Sets provider in "in scope" state.
|
||||
"""
|
||||
self.is_in_scope = True
|
||||
self._reset_instance()
|
||||
|
||||
def out_of_scope(self):
|
||||
"""
|
||||
Sets provider in "out of scope" state.
|
||||
"""
|
||||
self.is_in_scope = False
|
||||
self._reset_instance()
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
"""
|
||||
Returns provided instance.
|
||||
"""
|
||||
if not self.is_in_scope:
|
||||
raise RuntimeError('Trying to provide {} while provider is not in scope'.format(self.provides))
|
||||
return super(Scoped, self).__call__(*args, **kwargs)
|
||||
|
||||
def __enter__(self):
|
||||
"""
|
||||
With __enter__() implementation. Makes provider to be in scope.
|
||||
"""
|
||||
self.in_scope()
|
||||
return self
|
||||
|
||||
def __exit__(self, *_):
|
||||
"""
|
||||
With __exit__() implementation. Makes provider to be out of scope.
|
||||
"""
|
||||
self.out_of_scope()
|
||||
|
||||
|
||||
class ExternalDependency(Provider):
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue
Block a user