first commit
This commit is contained in:
26
middlewares/di.py
Normal file
26
middlewares/di.py
Normal file
@@ -0,0 +1,26 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user