Добавление автомута

src/modules/auto_mute.py

@@ -0,0 +1,199 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message, ChatPermissions
+import asyncio
+import logging
+import time
+
+from database import db
+from bad_words import contains_bad_word, get_bad_words_from_text
+from action_reporter import action_reporter
+from utils import delete_messages, format_mute_time
+
+# Получаем логгер для текущего модуля
+logger = logging.getLogger(__name__)
+
+# Система прогрессирующих мутов (в секундах)
+# Более плавная прогрессия для накопительного эффекта
+MUTE_LEVELS = [
+    300,        # 1.  5 минут (первое нарушение - символический мут)
+    900,        # 2.  15 минут
+    1800,       # 3.  30 минут
+    3600,       # 4.  1 час
+    7200,       # 5.  2 часа
+    14400,      # 6.  4 часа
+    28800,      # 7.  8 часов
+    43200,      # 8.  12 часов
+    86400,      # 9.  1 день
+    172800,     # 10. 2 дня
+    259200,     # 11. 3 дня
+    432000,     # 12. 5 дней
+    604800,     # 13. 7 дней
+    None,       # 16. Перманентный мут (режим только чтения навсегда)
+]
+
+# Период для подсчета нарушений (30 дней в секундах)
+VIOLATIONS_PERIOD = 2592000
+
+def get_mute_duration(violations_count: int) -> int:
+    """
+    Определяет длительность мута на основе количества нарушений.
+
+    Args:
+        violations_count: Количество нарушений пользователя
+
+    Returns:
+        Длительность мута в секундах (или None для перманентного мута)
+    """
+    if violations_count < 1:
+        return MUTE_LEVELS[0]
+
+    # Индекс уровня мута (количество нарушений - 1, т.к. начинаем с 0)
+    level_index = violations_count - 1
+
+    # Если превысили количество уровней, возвращаем перманентный мут
+    if level_index >= len(MUTE_LEVELS):
+        return None
+
+    return MUTE_LEVELS[level_index]
+
+async def apply_mute(bot: AsyncTeleBot, message: Message, user_id: int, duration: int, violations_count: int):
+    """
+    Применяет мут к пользователю.
+
+    Args:
+        bot: Экземпляр бота
+        message: Сообщение, которое вызвало мут
+        user_id: ID пользователя
+        duration: Длительность мута в секундах (None для перманентного)
+        violations_count: Количество нарушений
+    """
+    try:
+        # Устанавливаем ограничения (только чтение)
+        permissions = ChatPermissions(
+            can_send_messages=False,
+            can_send_media_messages=False,
+            can_send_polls=False,
+            can_send_other_messages=False,
+            can_add_web_page_previews=False,
+            can_change_info=False,
+            can_invite_users=False,
+            can_pin_messages=False,
+        )
+
+        # Вычисляем время окончания мута
+        until_date = None if duration is None else int(time.time()) + duration
+
+        # Выполняем мут
+        await bot.restrict_chat_member(
+            chat_id=message.chat.id,
+            user_id=user_id,
+            permissions=permissions,
+            until_date=until_date
+        )
+
+        # Удаляем сообщение с матом
+        try:
+            await bot.delete_message(chat_id=message.chat.id, message_id=message.message_id)
+        except Exception as e:
+            logger.warning(f"Не удалось удалить сообщение: {e}")
+
+        # Формируем сообщение о муте
+        if duration is None:
+            time_display = "навсегда"
+            warning_msg = (
+                f"⛔️ Пользователь <b>{message.from_user.first_name}</b> получил перманентный мут "
+                f"за злостное нарушение правил чата (использование нецензурной лексики).\n\n"
+                f"📊 Количество нарушений: <b>{violations_count}</b>\n"
+                f"🔒 Режим: только чтение (навсегда)"
+            )
+        else:
+            time_display = format_mute_time(duration)
+            warning_msg = (
+                f"⚠️ Пользователь <b>{message.from_user.first_name}</b> получил мут на <b>{time_display}</b> "
+                f"за использование нецензурной лексики.\n\n"
+                f"📊 Нарушение #{violations_count}\n"
+                f"💡 При повторных нарушениях время мута будет увеличиваться."
+            )
+
+        # Отправляем сообщение в чат
+        await bot.send_message(
+            chat_id=message.chat.id,
+            text=warning_msg,
+            message_thread_id=message.message_thread_id,
+        )
+
+        # Отправляем сообщение-лог в админ-чат
+        await action_reporter.log_action(
+            action="АВТОМУТ",
+            user_id=user_id,
+            admin_id=None,  # Автоматическое действие
+            reason=f"Использование нецензурной лексики (нарушение #{violations_count})",
+            duration=time_display,
+        )
+
+        # Записываем действие в логи
+        logger.info(
+            f"Пользователь {user_id} получил автоматический мут на {time_display} "
+            f"за нецензурную лексику (нарушение #{violations_count})"
+        )
+
+    except Exception as e:
+        logger.error(f"Ошибка при применении мута: {e}")
+
+async def check_message_for_profanity(bot: AsyncTeleBot, message: Message):
+    """
+    Проверяет сообщение на наличие бранных слов и применяет мут при необходимости.
+
+    Args:
+        bot: Экземпляр бота
+        message: Сообщение для проверки
+    """
+    # Проверяем только текстовые сообщения
+    if not message.text:
+        return
+
+    # Не проверяем команды
+    if message.text.startswith('/'):
+        return
+
+    # Проверяем, содержит ли сообщение бранные слова
+    if not contains_bad_word(message.text):
+        return
+
+    # Получаем ID пользователя и чата
+    user_id = message.from_user.id
+    chat_id = message.chat.id
+
+    # Проверяем, является ли отправитель администратором
+    try:
+        chat_member = await bot.get_chat_member(chat_id, user_id)
+        if chat_member.status in ['administrator', 'creator']:
+            logger.info(f"Администратор {user_id} использовал нецензурную лексику, мут не применен")
+            return
+    except Exception as e:
+        logger.error(f"Ошибка проверки статуса пользователя: {e}")
+        return
+
+    # Добавляем нарушение в базу данных
+    db.add_violation(user_id, chat_id, violation_type='bad_language')
+
+    # Получаем количество нарушений за последний месяц
+    violations_count = db.get_violations_count(user_id, chat_id, VIOLATIONS_PERIOD)
+
+    # Определяем длительность мута
+    mute_duration = get_mute_duration(violations_count)
+
+    # Применяем мут
+    await apply_mute(bot, message, user_id, mute_duration, violations_count)
+
+def register_handlers(bot: AsyncTeleBot):
+    """
+    Регистрирует обработчики для автоматического мута.
+    """
+
+    # Обработчик всех текстовых сообщений
+    @bot.message_handler(func=lambda message: message.text and message.chat.type in ['group', 'supergroup'])
+    async def handle_text_message(message: Message):
+        await check_message_for_profanity(bot, message)
+
+    logger.info("Модуль автоматического мута успешно загружен")