mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-22 17:47:02 +03:00
38 lines
743 B
Python
38 lines
743 B
Python
from dependency_injector import containers, providers
|
|
from dependency_injector.wiring import inject, Provide, Closing
|
|
|
|
|
|
class Service:
|
|
init_counter: int = 0
|
|
shutdown_counter: int = 0
|
|
|
|
@classmethod
|
|
def reset_counter(cls):
|
|
cls.init_counter = 0
|
|
cls.shutdown_counter = 0
|
|
|
|
@classmethod
|
|
def init(cls):
|
|
cls.init_counter += 1
|
|
|
|
@classmethod
|
|
def shutdown(cls):
|
|
cls.shutdown_counter += 1
|
|
|
|
|
|
def init_service():
|
|
service = Service()
|
|
service.init()
|
|
yield service
|
|
service.shutdown()
|
|
|
|
|
|
class Container(containers.DeclarativeContainer):
|
|
|
|
service = providers.Resource(init_service)
|
|
|
|
|
|
@inject
|
|
def test_function(service: Service = Closing[Provide["service"]]):
|
|
return service
|