from telebot.async_telebot import AsyncTeleBot from telebot.types import Message, User import asyncio import logging from database import db from action_reporter import action_reporter from utils import ( delete_messages, check_admin_status, check_target_status, ) from config import COMMAND_MESSAGES logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля # Регистрирует все обработчики команд def register_handlers(bot: AsyncTeleBot): # Обработчик команды /ban @bot.message_handler(commands=['ban']) async def _ban_command_wrapper(message: Message): await ban_command(bot, message) # Основная функция команды /ban async def ban_command(bot: AsyncTeleBot, message: Message, photo_path: str = None): # Определяем целевого пользователя target_user = None # Определяем причину reason = None # Разбиваем текст сообщения на части parts_msg = message.text.split() # Команда /ban help if len(parts_msg) == 2 and parts_msg[1].strip() in ('help', 'помощь'): # Отправляем инструкцию await bot.send_message( chat_id=message.chat.id, text=COMMAND_MESSAGES['manual_ban'], message_thread_id=message.message_thread_id, ) # Удаляем сообщения через 30 секунд await delete_messages(bot, message, time_sleep=30, number_message=2) return try: # Проверяем, является ли отправитель администратором if await check_admin_status(bot, message) == 1: return # Команда через ответ на сообщение, если одно слово (/ban) if len(parts_msg) == 1: # Если это топик if message.is_topic_message: # Если без ответа на сообщение if not message.reply_to_message or message.message_thread_id == message.reply_to_message.message_id: # Удаляем сообщение через 3 секунды await delete_messages(bot, message, time_sleep=3, number_message=1) return # Если с ответом на сообщение else: # Собираем данные target_user = message.reply_to_message.from_user reason = 'отсутствует' # Если это General (обычный чат) elif message.reply_to_message and not message.is_topic_message: # Собираем данные target_user = message.reply_to_message.from_user reason = 'отсутствует' message.message_thread_id = None # Если команда неправильная else: # Удаляем сообщение через 3 секунды await delete_messages(bot, message, time_sleep=3, number_message=1) return # В сообщении больше одного слова else: # Если второе слово это тег или ID if parts_msg[1].strip().isdigit() or parts_msg[1].startswith('@'): # Собираем данные identifier = parts_msg[1].strip() reason = ' '.join(parts_msg[2:]) if parts_msg[2:] != [] else 'отсутствует' # Делаем поиск по ID if identifier.isdigit(): # Ищем пользователя в базе данных user_info = db.get_user(int(identifier)) # Если нашли пользователя if user_info: # Создаем объект пользователя target_user = User( id=user_info[0], first_name=user_info[1], username=user_info[2], is_bot=False ) # Делаем поиск по тегу elif identifier.startswith('@'): # Ищем пользователя в базе данных (убрали @) user_info = db.get_user_by_username(identifier[1:]) # Если нашли пользователя if user_info: # Создаем объект пользователя target_user = User( id=user_info[0], first_name=user_info[1], username=user_info[2], is_bot=False ) # Команда через ответ на сообщение с указанием причины else: # Если это топик if message.is_topic_message: # Если без ответа на сообщение if not message.reply_to_message or message.message_thread_id == message.reply_to_message.message_id: # Удаляем сообщение через 3 секунды await delete_messages(bot, message, time_sleep=3, number_message=1) return # Если с ответом на сообщение else: # Собираем данные target_user = message.reply_to_message.from_user reason = ' '.join(parts_msg[1:]) # Если это General (обычный чат) elif message.reply_to_message and not message.is_topic_message: # Собираем данные target_user = message.reply_to_message.from_user reason = ' '.join(parts_msg[1:]) message.message_thread_id = None # Если команда неправильная else: # Удаляем сообщение через 3 секунды await delete_messages(bot, message, time_sleep=3, number_message=1) return # Если пользователь не найден в базе данных if not target_user: # Отправляем предупреждение await bot.send_message( chat_id=message.chat.id, text=COMMAND_MESSAGES['user_not_found'], message_thread_id=message.message_thread_id, ) # Удаляем сообщения через 5 секунд await delete_messages(bot, message, time_sleep=5, number_message=2) return # Проверяем статус целевого пользователя if await check_target_status(bot, message, target_user) == 1: return try: # Выполняем бан await bot.ban_chat_member( chat_id=message.chat.id, user_id=target_user.id, ) # Отправляем лог в админ-чат await action_reporter.log_action( action="БАН", user_id=target_user.id, admin_id=message.from_user.id, reason=reason, duration=None, photo_path=photo_path, ) # Отправляем сообщения, что пользователь получил бан await bot.send_message( chat_id=message.chat.id, text=COMMAND_MESSAGES['banned'], message_thread_id=message.message_thread_id, ) # Записываем действие в логи logger.info(f"Пользователь {target_user.id} забанен администратором {message.from_user.id}.") # Удаляем сообщения через 5 секунд await delete_messages(bot, message, time_sleep=5, number_message=2) except Exception as e: # Отправляем ошибку await bot.send_message( chat_id=message.chat.id, text=COMMAND_MESSAGES['error'].format(e=str(e)), message_thread_id=message.message_thread_id, ) # Записываем ошибку в логи logger.error(f"Ошибка бана: {str(e)}") # Удаляем сообщения через 5 секунд await delete_messages(bot, message, time_sleep=5, number_message=2) except Exception as e: # Отправляем ошибку await bot.send_message( chat_id=message.chat.id, text=COMMAND_MESSAGES['general_error'], message_thread_id=message.message_thread_id, ) # Записываем ошибку в логи logger.error(f"Общая ошибка в ban_command: {str(e)}") # Удаляем сообщения через 5 секунд await delete_messages(bot, message, time_sleep=5, number_message=2)