init

middleware/telegram/auth.py

@@ -0,0 +1,40 @@
+from aiogram import BaseMiddleware
+from aiogram.types import TelegramObject, Message, CallbackQuery
+from aiogram.utils.i18n import gettext as _
+from babel.support import LazyProxy
+from typing import Any, Awaitable, Callable, Dict
+
+from ...model.user import UserBase
+
+
+class AuthMiddleware(BaseMiddleware):
+
+    def __init__[UserType: UserBase](self,
+                                     user_class: type[UserType],
+                                     not_authenticated_msg: LazyProxy | str = "not authenticated",
+                                     not_active_msg: LazyProxy | str = "not active"):
+        
+        self.user_class = user_class
+        self.not_authenticated_msg = not_authenticated_msg
+        self.not_active_msg = not_active_msg
+
+    
+    async def __call__(self,
+                       handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
+                       event: TelegramObject,
+                       data: Dict[str, Any]) -> Any:
+        
+        user = await self.user_class.get(id = event.from_user.id, session = data["db_session"])
+            
+        if user and user.is_active:
+            data["user"] = user
+            return await handler(event, data)
+        
+        if type(event) in [Message, CallbackQuery]:
+            if user and not user.is_active:
+                return await event.answer(self.not_active_msg)
+            return await event.answer(self.not_authenticated_msg)
+            
+            
+
+