added utils.py

1. Добавил message_thread_id для всех команд, убрав костыль, который отвечал за корректную отправку
   сообщений в топик.
2. Функции (определение администратора, удаление сообщений) вынес в utils.py, от куда они будут вызываться в командах. Модули стали более читаемы из-за уменьшения количества строк кода в них.
3. Дописал manual_unban и добавил error в config.py
4. Оптимизация

src/modules/unmute.py

@@ -5,112 +5,99 @@ 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__) # Получаем логгер для текущего модуля
+# Получаем логгер для текущего модуля
+logger = logging.getLogger(__name__)
 
-# Удаляет два последних сообщения
-async def delete_messages(bot: AsyncTeleBot, message: Message, time_sleep: int):
-    await asyncio.sleep(time_sleep)
-    await bot.delete_message(message.chat.id, message.message_id)
-    await bot.delete_message(message.chat.id, message.message_id+1)
+# Регистрирует все обработчики команд
+def register_handlers(bot: AsyncTeleBot):
 
-def register_handlers(bot: AsyncTeleBot): # Регистрирует все обработчики команд
-
-    @bot.message_handler(commands=['unmute']) # Обработчик команды /unmute
+    # Обработчик команды /unmute
+    @bot.message_handler(commands=['unmute'])
     async def unmute_command(message: Message):
 
-        # Отправка сообщения в тему или обычный чат
-        send_message = bot.reply_to if message.is_topic_message else bot.send_message
-        chat_id = message if message.is_topic_message else message.chat.id
-
         # Определяем целевого пользователя
         target_user = None
 
         # Разбиваем текст сообщения на части
         parts_msg = message.text.split()
 
-        # Выводим помощь (/unmute help)
-        if len(parts_msg) == 2 and parts_msg[1].strip() in ['help', 'помощь']:
-            await send_message(chat_id, COMMAND_MESSAGES['manual_unmute'])
+        # Команда /unmute 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_unmute'], 
+                message_thread_id=message.message_thread_id,
+            )
 
             # Удаляем сообщения через 30 секунд
-            await delete_messages(bot, message, 30)
-
+            await delete_messages(bot, message, time_sleep=30, number_message=2)
             return
 
         try:
+
             # Проверяем, является ли отправитель администратором
-            try:
-                admin_status = await bot.get_chat_member(message.chat.id, message.from_user.id)
-                
-                # Проверяем статус администратора (создателя)
-                if admin_status.status not in ['administrator', 'creator']:
-                    await send_message(chat_id, COMMAND_MESSAGES['no_admin_rights'])
-
-                    # Удаляем сообщения через 5 секунд
-                    await delete_messages(bot, message, 5)
-
-                    return
-                    
-                # Проверяем право администратора на разбан
-                if admin_status.status == 'administrator' and not admin_status.can_restrict_members:
-                    await send_message(chat_id, COMMAND_MESSAGES['no_restrict_rights'])
-
-                    # Удаляем сообщения через 5 секунд
-                    await delete_messages(bot, message, 5)
-
-                    return
-            
-            except Exception as e:
-                await send_message(chat_id, f"⚠️ Ошибка: {str(e)}")
-                logger.error(f"Ошибка при получении статуса администратора: {str(e)}")
-
-                # Удаляем сообщения через 5 секунд
-                await delete_messages(bot, message, 5)
-
+            if await check_admin_status(bot, message) == 1:
                 return
 
-            # Если одно слово, то ответом на сообщение
+            # Команда через ответ на сообщение, если одно слово (/unmute)
             if len(parts_msg) == 1:
 
-                # Если это тема
+                # Если это топик
                 if message.is_topic_message:
 
-                    # Если без ответа на сообщение, ошибка
+                    # Если без ответа на сообщение
                     if message.message_thread_id == message.reply_to_message.message_id:
-                        await asyncio.sleep(3)
-                        await bot.delete_message(message.chat.id, 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
 
-                # Если это обычный чат
+                # Если это General (обычный чат)
                 elif message.reply_to_message and message.is_topic_message is None:
+
+                    # Собираем данные
                     target_user = message.reply_to_message.from_user
+                    message.message_thread_id = None
                 
                 # Удаляем сообщение, если команда неправильная
                 else:
-                    await asyncio.sleep(3)
-                    await bot.delete_message(message.chat.id, message.message_id)
+                    
+                    # Удаляем сообщение через 3 секунды
+                    await delete_messages(bot, message, time_sleep=3, number_message=1)
                     return
             
-            # Если два, то через тег или ID
+            # Команда через тег или ID, если два слова
             elif len(parts_msg) == 2:
                 
+                # Собираем данные
                 identifier = parts_msg[1].strip()
                 
-                # Поиск по ID
+                # Делаем поиск по ID
                 if identifier.isdigit():
 
-                    # Делаем в int и ищем
+                    # Ищем пользователя в базе данных
                     user_info = db.get_user(int(identifier))
 
+                    # Если нашли пользователя
                     if user_info:
-                        # Создаем объект пользователя из данных базы
+
+                        # Создаем объект пользователя
                         target_user = User(
                             id=user_info[0],
                             first_name=user_info[1],
@@ -118,14 +105,16 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
                             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],
@@ -133,17 +122,26 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
                             is_bot=False
                         )
 
-            # Если пользователь не найден
+            # Если пользователь не найден в базе данных
             if not target_user:
-                await send_message(chat_id, COMMAND_MESSAGES['user_not_found'])
+
+                # Отправляем предупреждение
+                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, 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:
+
                 # Убираем ограничения (можно писать в чат)
                 permissions = ChatPermissions(
                     can_send_messages=True,
@@ -153,13 +151,14 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
                     can_add_web_page_previews=True,
                     can_change_info=False,
                     can_invite_users=True,
-                    can_pin_messages=False,
+                    can_pin_messages=False
                 )
 
+                # Выполняем размут
                 await bot.restrict_chat_member(
                     chat_id=message.chat.id,
                     user_id=target_user.id,
-                    permissions=permissions
+                    permissions=permissions,
                 )
 
                 # Отправляем лог в админ-чат
@@ -169,26 +168,48 @@ def register_handlers(bot: AsyncTeleBot): # Регистрирует все об
                     admin_id=message.from_user.id,
                     reason=None,
                     duration=None,
-                    photo_path=None
+                    photo_path=None,
                 )
                 
                 # Отправляем сообщения, что пользователь получил размут
-                await send_message(chat_id, COMMAND_MESSAGES['unmuted'])
+                await bot.send_message(
+                    chat_id=message.chat.id, 
+                    text=COMMAND_MESSAGES['unmuted'], 
+                    message_thread_id=message.message_thread_id,
+                )
+
+                # Записываем действие в логи
                 logger.info(f"Пользователь {target_user.id} получил размут от администратора {message.from_user.id}.")
                 
                 # Удаляем сообщения через 5 секунд
-                await delete_messages(bot, message, 5)
+                await delete_messages(bot, message, time_sleep=5, number_message=2)
                 
             except Exception as e:
-                await send_message(chat_id, f"⚠️ Ошибка: {str(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, 5)
+                await delete_messages(bot, message, time_sleep=5, number_message=2)
                 
         except Exception as e:
-            await send_message(chat_id, COMMAND_MESSAGES['general_error'])
+
+            # Отправляем ошибку
+            await bot.send_message(
+                chat_id=message.chat.id, 
+                text=COMMAND_MESSAGES['general_error'], 
+                message_thread_id=message.message_thread_id,
+            )
+
+            # Записываем ошибку в логи
             logger.error(f"Общая ошибка в unmute_command: {str(e)}")
 
             # Удаляем сообщения через 5 секунд
-            await delete_messages(bot, message, 5)
+            await delete_messages(bot, message, time_sleep=5, number_message=2)