python-dependency-injector/examples/providers/factory_delegation.py

"""`Factory` provider delegation example."""

from typing import Callable, List

from dependency_injector import providers


class User:
    def __init__(self, uid: int) -> None:
        self.uid = uid


class UserRepository:
    def __init__(self, user_factory: Callable[..., User]) -> None:
        self.user_factory = user_factory

    def get_all(self) -> List[User]:
        return [
            self.user_factory(**user_data)
            for user_data in [{'uid': 1}, {'uid': 2}]
        ]


user_factory = providers.Factory(User)
user_repository_factory = providers.Factory(
    UserRepository,
    user_factory=user_factory.provider,
)


if __name__ == '__main__':
    user_repository = user_repository_factory()

    user1, user2 = user_repository.get_all()

    assert user1.uid == 1
    assert user2.uid == 2