MVP

app/middlewares/__init__.py

@@ -0,0 +1,4 @@
+from .outer_middlewares import AccessCheckMiddleware
+
+
+__all__ = ["AccessCheckMiddleware"]

app/middlewares/outer_middlewares.py

@@ -0,0 +1,29 @@
+import logging
+from typing import Any, Awaitable, Callable, Dict
+from aiogram import BaseMiddleware, Bot
+from aiogram.types import TelegramObject
+from database import async_session_, Worker
+from sqlalchemy import select
+
+
+
+class AccessCheckMiddleware(BaseMiddleware):
+    sessions_in_memory_db = set()
+    async def __call__(
+            self,
+            handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
+            event: TelegramObject,
+            data: Dict[str, Any]
+    ) -> Any:
+        event_data = event.message or event.callback_query
+        user = event_data.from_user.id
+        if user not in self.sessions_in_memory_db:
+            async with async_session_() as session:
+                async with session.begin():
+                    result = await session.execute(select(Worker).where(Worker.telegram_id == event_data.from_user.id))
+                    user = result.scalars().first()
+        if user:
+            self.sessions_in_memory_db.add(event_data.from_user.id)
+            return await handler(event, data)
+        return None
+