mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 18:15:28 +03:00
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
__all__ = ["inject", "setup_dishka", "FromDishka"]
|
|
|
|
from typing import Any, Callable, TypeVar
|
|
|
|
from dishka import Container, FromDishka
|
|
from dishka.integrations.base import is_dishka_injected, wrap_injection
|
|
|
|
from argenta.app.models import App
|
|
from argenta.response.entity import Response
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
def inject(func: Callable[..., T]) -> Callable[..., T]:
|
|
return wrap_injection(
|
|
func=func,
|
|
is_async=False,
|
|
container_getter=_get_container_from_response,
|
|
)
|
|
|
|
|
|
def setup_dishka(app: App, container: Container, *, auto_inject: bool = False) -> None:
|
|
Response.patch_by_container(container)
|
|
if auto_inject:
|
|
_auto_inject_handlers(app)
|
|
|
|
|
|
def _get_container_from_response(args: tuple[Any, ...], kwargs: dict[str, Any]) -> Container:
|
|
for arg in args:
|
|
if isinstance(arg, Response):
|
|
if hasattr(arg, "__dishka_container__"):
|
|
return arg.__dishka_container__ # pyright: ignore[reportPrivateUsage]
|
|
break
|
|
raise RuntimeError("dishka container not found in Response")
|
|
|
|
|
|
def _auto_inject_handlers(app: App) -> None:
|
|
for router in app.registered_routers:
|
|
for command_handler in router.command_handlers:
|
|
if not is_dishka_injected(command_handler.handler_as_func):
|
|
injected_handler = inject(command_handler.handler_as_func)
|
|
command_handler.handler_as_func = injected_handler
|