29
app/middlewares/outer_middlewares.py
Normal file
29
app/middlewares/outer_middlewares.py
Normal file
@@ -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
|
||||
|
Reference in New Issue
Block a user