27 lines
789 B
Python
27 lines
789 B
Python
from typing import Callable, Awaitable, Any
|
|
from aiogram import BaseMiddleware
|
|
from aiogram.types import TelegramObject
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker
|
|
|
|
from dto.di import DependenciesDTO
|
|
|
|
|
|
class DIMiddleware(BaseMiddleware):
|
|
def __init__(
|
|
self, deps: DependenciesDTO, session_factory: async_sessionmaker
|
|
) -> None:
|
|
self.deps = deps
|
|
self.session_factory = session_factory
|
|
|
|
async def __call__(
|
|
self,
|
|
handler: Callable[[TelegramObject, dict[str, Any]], Awaitable[Any]],
|
|
event: TelegramObject,
|
|
data: dict[str, Any],
|
|
) -> Any:
|
|
data["deps"] = self.deps
|
|
|
|
async with self.session_factory() as session:
|
|
data["session"] = session
|
|
return await handler(event, data)
|