forked from Muzifs/LGBot
199 lines
8.5 KiB
Python
199 lines
8.5 KiB
Python
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 not message.text.startswith('/') and message.chat.type in ['group', 'supergroup'])
|
||
async def handle_text_message(message: Message):
|
||
await check_message_for_profanity(bot, message)
|
||
|
||
logger.info("Модуль автоматического мута успешно загружен") |