python-dependency-injector/tests/unit/samples/wiringstringids/resourceclosing.py

from dependency_injector import containers, providers
from dependency_injector.wiring import inject, Provide, Closing


class Singleton:
    pass


class Service:
    init_counter: int = 0
    shutdown_counter: int = 0
    dependency: Singleton = None

    @classmethod
    def reset_counter(cls):
        cls.init_counter = 0
        cls.shutdown_counter = 0

    @classmethod
    def init(cls, dependency: Singleton = None):
        if dependency:
            cls.dependency = dependency
        cls.init_counter += 1

    @classmethod
    def shutdown(cls):
        cls.shutdown_counter += 1


class FactoryService:
    def __init__(self, service: Service):
        self.service = service


class NestedService:
    def __init__(self, factory_service: FactoryService):
        self.factory_service = factory_service


def init_service():
    service = Service()
    service.init()
    yield service
    service.shutdown()


def init_service_with_singleton(singleton: Singleton):
    service = Service()
    service.init(singleton)
    yield service
    service.shutdown()


class Container(containers.DeclarativeContainer):

    service = providers.Resource(init_service)
    factory_service = providers.Factory(FactoryService, service)
    factory_service_kwargs = providers.Factory(
        FactoryService,
        service=service
    )
    nested_service = providers.Factory(NestedService, factory_service)


class ContainerSingleton(containers.DeclarativeContainer):

    singleton = providers.Resource(Singleton)
    service = providers.Resource(
        init_service_with_singleton,
        singleton
    )
    factory_service = providers.Factory(FactoryService, service)
    factory_service_kwargs = providers.Factory(
        FactoryService,
        service=service
    )
    nested_service = providers.Factory(NestedService, factory_service)


@inject
def test_function(service: Service = Closing[Provide["service"]]):
    return service


@inject
def test_function_dependency(factory: FactoryService = Closing[Provide["factory_service"]]):
    return factory


@inject
def test_function_dependency_kwargs(factory: FactoryService = Closing[Provide["factory_service_kwargs"]]):
    return factory


def test_function_nested_dependency(
    nested: NestedService = Closing[Provide["nested_service"]]
):
    return nested