Init commit

Mousey/Bot/Middlewares/database.py

@@ -0,0 +1,25 @@
+from typing import Callable, Awaitable, Dict, Any
+
+from aiogram import BaseMiddleware
+from aiogram.types import TelegramObject
+
+from sqlalchemy.ext.asyncio import async_sessionmaker
+
+
+class DatabaseSessionMiddleware(BaseMiddleware):
+    """
+        Мидлвари для получения сессии в БД для фильтров, сообщений и прочего.
+    """
+    def __init__(self, session_pool: async_sessionmaker) -> None:
+        self.session_pool = session_pool
+
+    async def __call__(
+        self,
+        handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
+        event: TelegramObject,
+        data: Dict[str, Any],
+    ) -> Any:
+        async with self.session_pool() as session:
+            data["session"] = session
+            return await handler(event, data)
+