Add middlewaer datatabase # TODO

middlewares/admin.py

@@ -14,7 +14,6 @@ class AdminMiddleware(BaseMiddleware):
         event: TelegramObject,
         data: Dict[str, Any],
     ) -> Any:
-        # Проверяем, является ли пользователь администратором
         user_id = None
 
         if isinstance(event, Message):

middlewares/database.py

@@ -0,0 +1,23 @@
+from functools import wraps
+from typing import Any, Awaitable, Callable, Dict
+
+import asyncpg
+from aiogram import BaseMiddleware
+from aiogram.types import CallbackQuery, Message, TelegramObject
+
+from config import DATABASE_URL
+
+
+class AdminMiddleware(BaseMiddleware):
+    async def __call__(
+        self,
+        handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
+        event: TelegramObject,
+        data: Dict[str, Any],
+    ) -> Any:
+        session = await asyncpg.connect(DATABASE_URL)
+        data["session"] = session
+        try:
+            return await handler(event, data)
+        finally:
+            await session.close()