Andrey
8cee629e28
- Introduced `DependenciesMiddleware` and `BlacklistMiddleware` for enhanced request handling across all routers. - Refactored admin handlers to utilize new middleware, improving access control and error handling. - Updated the `admin_router` to include middleware for access checks and streamlined the process of banning users. - Enhanced the structure of admin handler imports for better organization and maintainability. - Improved error handling in various admin functions to ensure robust user interactions.
62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
import html
|
||
from typing import Optional
|
||
from aiogram import types
|
||
from aiogram.fsm.context import FSMContext
|
||
|
||
from helper_bot.keyboards.keyboards import get_reply_keyboard_admin
|
||
from helper_bot.handlers.admin.exceptions import AdminError
|
||
from logs.custom_logger import logger
|
||
|
||
|
||
def escape_html(text: str) -> str:
|
||
"""Экранирование HTML для безопасного использования в сообщениях"""
|
||
return html.escape(str(text)) if text else ""
|
||
|
||
|
||
async def return_to_admin_menu(message: types.Message, state: FSMContext,
|
||
additional_message: Optional[str] = None) -> None:
|
||
"""Универсальная функция для возврата в админ-меню"""
|
||
await state.set_data({})
|
||
await state.set_state("ADMIN")
|
||
markup = get_reply_keyboard_admin()
|
||
|
||
if additional_message:
|
||
await message.answer(additional_message)
|
||
|
||
await message.answer('Вернулись в меню', reply_markup=markup)
|
||
|
||
|
||
async def handle_admin_error(message: types.Message, error: Exception,
|
||
state: FSMContext, error_context: str = "") -> None:
|
||
"""Централизованная обработка ошибок административных операций"""
|
||
logger.error(f"Ошибка в {error_context}: {error}")
|
||
|
||
if isinstance(error, AdminError):
|
||
await message.answer(f"Ошибка: {str(error)}")
|
||
else:
|
||
await message.answer("Произошла внутренняя ошибка. Попробуйте позже.")
|
||
|
||
await return_to_admin_menu(message, state)
|
||
|
||
|
||
def format_user_info(user_id: int, username: str, full_name: str) -> str:
|
||
"""Форматирование информации о пользователе для отображения"""
|
||
safe_username = escape_html(username)
|
||
safe_full_name = escape_html(full_name)
|
||
|
||
return (f"<b>Выбран пользователь:</b>\n"
|
||
f"<b>ID:</b> {user_id}\n"
|
||
f"<b>Username:</b> {safe_username}\n"
|
||
f"<b>Имя:</b> {safe_full_name}")
|
||
|
||
|
||
def format_ban_confirmation(user_id: int, reason: str, ban_days: Optional[int]) -> str:
|
||
"""Форматирование подтверждения бана"""
|
||
safe_reason = escape_html(reason)
|
||
ban_text = "Навсегда" if ban_days is None else f"{ban_days} дней"
|
||
|
||
return (f"<b>Необходимо подтверждение:</b>\n"
|
||
f"<b>Пользователь:</b> {user_id}\n"
|
||
f"<b>Причина бана:</b> {safe_reason}\n"
|
||
f"<b>Срок бана:</b> {ban_text}")
|