Andrey
1c6a37bc12
- Added `ca-certificates` installation to Dockerfile for improved network security. - Updated health check command in Dockerfile to include better timeout handling. - Refactored `run_helper.py` to implement proper signal handling and logging during shutdown. - Transitioned database operations to an asynchronous model in `async_db.py`, improving performance and responsiveness. - Updated database schema to support new foreign key relationships and optimized indexing for better query performance. - Enhanced various bot handlers to utilize async database methods, improving overall efficiency and user experience. - Removed obsolete database and fix scripts to streamline the project structure.
59 lines
3.0 KiB
Python
59 lines
3.0 KiB
Python
from typing import Dict, Any
|
||
import html
|
||
from datetime import datetime
|
||
|
||
from aiogram import BaseMiddleware, types
|
||
from aiogram.types import TelegramObject, Message, CallbackQuery
|
||
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: TelegramObject, data: Dict[str, Any]) -> Any:
|
||
# Проверяем тип события и получаем пользователя
|
||
user = None
|
||
if isinstance(event, Message):
|
||
user = event.from_user
|
||
elif isinstance(event, CallbackQuery):
|
||
user = event.from_user
|
||
|
||
# Если это не сообщение или callback, пропускаем проверку
|
||
if not user:
|
||
return await handler(event, data)
|
||
|
||
logger.info(f'Вызов BlacklistMiddleware для пользователя {user.username}')
|
||
|
||
# Используем асинхронную версию для предотвращения блокировки
|
||
if await BotDB.check_user_in_blacklist(user.id):
|
||
logger.info(f'BlacklistMiddleware результат для пользователя: {user.username} заблокирован!')
|
||
user_info = await BotDB.get_blacklist_users_by_id(user.id)
|
||
# Экранируем потенциально проблемные символы
|
||
reason = html.escape(str(user_info[1])) if user_info and user_info[1] else "Не указана"
|
||
|
||
# Преобразуем timestamp в человекочитаемый формат
|
||
if user_info and user_info[2]:
|
||
try:
|
||
timestamp = int(user_info[2])
|
||
date_unban = datetime.fromtimestamp(timestamp).strftime("%d-%m-%Y %H:%M")
|
||
except (ValueError, TypeError):
|
||
date_unban = "Не указана"
|
||
else:
|
||
date_unban = "Не указана"
|
||
|
||
# Отправляем сообщение в зависимости от типа события
|
||
if isinstance(event, Message):
|
||
await event.answer(
|
||
f"<b>Ты заблокирован.</b>\n<b>Причина блокировки:</b> {reason}\n<b>Дата разбана:</b> {date_unban}")
|
||
elif isinstance(event, CallbackQuery):
|
||
await event.answer(
|
||
f"<b>Ты заблокирован.</b>\n<b>Причина блокировки:</b> {reason}\n<b>Дата разбана:</b> {date_unban}",
|
||
show_alert=True)
|
||
|
||
return False
|
||
|
||
logger.info(f'BlacklistMiddleware результат для пользователя: {user.username} доступ разрешен')
|
||
return await handler(event, data)
|