mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-22 17:47:02 +03:00
45 lines
989 B
Python
45 lines
989 B
Python
"""Custom `Factory` example."""
|
|
|
|
import dependency_injector.providers as providers
|
|
|
|
|
|
class User(object):
|
|
"""Example class User."""
|
|
|
|
|
|
class UsersFactory(providers.Provider):
|
|
"""Example users factory."""
|
|
|
|
__slots__ = ('_factory',)
|
|
|
|
def __init__(self):
|
|
"""Initializer."""
|
|
self._factory = providers.Factory(User)
|
|
super(UsersFactory, self).__init__()
|
|
|
|
def _provide(self, *args, **kwargs):
|
|
"""Return provided instance.
|
|
|
|
:param args: tuple of context positional arguments
|
|
:type args: tuple[object]
|
|
|
|
:param kwargs: dictionary of context keyword arguments
|
|
:type kwargs: dict[str, object]
|
|
|
|
:rtype: object
|
|
"""
|
|
return self._factory(*args, **kwargs)
|
|
|
|
|
|
# Users factory:
|
|
users_factory = UsersFactory()
|
|
|
|
# Creating several User objects:
|
|
user1 = users_factory()
|
|
user2 = users_factory()
|
|
|
|
# Making some asserts:
|
|
assert isinstance(user1, User)
|
|
assert isinstance(user2, User)
|
|
assert user1 is not user2
|