27 lines
1.6 KiB
Python
27 lines
1.6 KiB
Python
from typing import Dict, Any
|
||
import html
|
||
|
||
from aiogram import BaseMiddleware, types
|
||
from helper_bot.utils.base_dependency_factory import get_global_instance
|
||
from logs.custom_logger import logger
|
||
|
||
bdf = get_global_instance()
|
||
BotDB = bdf.get_db()
|
||
|
||
|
||
class BlacklistMiddleware(BaseMiddleware):
|
||
async def __call__(self, handler, event: types.Message, data: Dict[str, Any]) -> Any:
|
||
logger.info(f'Вызов BlacklistMiddleware для пользователя {event.from_user.username}')
|
||
# Используем асинхронную версию для предотвращения блокировки
|
||
if await BotDB.check_user_in_blacklist_async(user_id=event.from_user.id):
|
||
logger.info(f'BlacklistMiddleware результат для пользователя: {event.from_user.username} заблокирован!')
|
||
user_info = await BotDB.get_blacklist_users_by_id_async(event.from_user.id)
|
||
# Экранируем потенциально проблемные символы
|
||
reason = html.escape(str(user_info[2])) if user_info[2] else "Не указана"
|
||
date_unban = html.escape(str(user_info[3])) if user_info[3] else "Не указана"
|
||
await event.answer(
|
||
f"<b>Ты заблокирован.</b>\n<b>Причина блокировки:</b> {reason}\n<b>Дата разбана:</b> {date_unban}")
|
||
return False
|
||
logger.info(f'BlacklistMiddleware результат для пользователя: {event.from_user.username} доступ разрешен')
|
||
return await handler(event, data)
|