исправление работы списков слов

добавление сброса счётчиков нарушений
This commit is contained in:
2025-10-13 14:20:18 +03:00
parent ab10879c06
commit 69d55a68b9
5 changed files with 161 additions and 3 deletions

View File

@@ -123,6 +123,18 @@ COMMAND_MESSAGES = {
"<u>Прочее:</u>\n" "<u>Прочее:</u>\n"
"• <code>/badwords reload</code> - Перезагрузить из файла\n\n" "• <code>/badwords reload</code> - Перезагрузить из файла\n\n"
"<i>💡 Все изменения применяются немедленно</i>" "<i>💡 Все изменения применяются немедленно</i>"
),
'reset_violations_help': (
"<b>🔄 Команда /reset_violations</b>\n\n"
"<i>Сбрасывает счётчик нарушений пользователя</i>\n\n"
"<u>🎯 Способы использования:</u>\n"
"1. Ответ на сообщение:\n"
" <code>/reset_violations</code>\n"
"2. По тегу пользователя:\n"
" <code>/reset_violations @username</code>\n"
"3. По ID пользователя:\n"
" <code>/reset_violations 123456789</code>\n\n"
"<i> Сбрасывает все записи об автомутах пользователя</i>"
) )
} }

View File

@@ -143,5 +143,18 @@ class Database: # Инициализация класса
logger.info(f"Удалено старых нарушений: {deleted_count}") logger.info(f"Удалено старых нарушений: {deleted_count}")
return deleted_count return deleted_count
# Сбрасывает все нарушения пользователя в чате
def reset_user_violations(self, user_id: int, chat_id: int):
with self._get_connection() as connect:
cursor = connect.cursor()
cursor.execute('''
DELETE FROM violations
WHERE user_id = ? AND chat_id = ?
''', (user_id, chat_id))
deleted_count = cursor.rowcount
connect.commit()
logger.info(f"Сброшено {deleted_count} нарушений пользователя {user_id} в чате {chat_id}")
return deleted_count
# Создаем экземпляр базы данных для импорта в других модулях # Создаем экземпляр базы данных для импорта в других модулях
db = Database() db = Database()

View File

@@ -124,6 +124,7 @@ async def setup_bot_commands():
BotCommand("mute", "Замутить пользователя. Использование: /mute help"), BotCommand("mute", "Замутить пользователя. Использование: /mute help"),
BotCommand("unmute", "Размутить пользователя. Использование: /unmute help"), BotCommand("unmute", "Размутить пользователя. Использование: /unmute help"),
BotCommand("badwords", "Управление списком бранных слов. /badwords help"), BotCommand("badwords", "Управление списком бранных слов. /badwords help"),
BotCommand("reset_violations", "Сбросить счётчик нарушений пользователя"),
BotCommand("botdata", "Получить данные бота (только для админов)"), BotCommand("botdata", "Получить данные бота (только для админов)"),
] ]

View File

@@ -56,7 +56,7 @@ def get_mute_duration(violations_count: int) -> int:
return MUTE_LEVELS[level_index] return MUTE_LEVELS[level_index]
async def apply_mute(bot: AsyncTeleBot, message: Message, user_id: int, duration: int, violations_count: int): async def apply_mute(bot: AsyncTeleBot, message: Message, user_id: int, duration: int, violations_count: int, bad_words_found: list = None):
""" """
Применяет мут к пользователю. Применяет мут к пользователю.
@@ -66,6 +66,7 @@ async def apply_mute(bot: AsyncTeleBot, message: Message, user_id: int, duration
user_id: ID пользователя user_id: ID пользователя
duration: Длительность мута в секундах (None для перманентного) duration: Длительность мута в секундах (None для перманентного)
violations_count: Количество нарушений violations_count: Количество нарушений
bad_words_found: Список найденных плохих слов
""" """
try: try:
# Устанавливаем ограничения (только чтение) # Устанавливаем ограничения (только чтение)
@@ -97,6 +98,14 @@ async def apply_mute(bot: AsyncTeleBot, message: Message, user_id: int, duration
except Exception as e: except Exception as e:
logger.warning(f"Не удалось удалить сообщение: {e}") logger.warning(f"Не удалось удалить сообщение: {e}")
# Формируем информацию о найденных словах
words_info = ""
if bad_words_found:
words_list = ", ".join([f"«{word}»" for word in bad_words_found])
words_info = f"Найдено слов: {words_list}"
else:
words_info = "Использование нецензурной лексики"
# Формируем сообщение о муте # Формируем сообщение о муте
if duration is None: if duration is None:
time_display = "навсегда" time_display = "навсегда"
@@ -127,7 +136,7 @@ async def apply_mute(bot: AsyncTeleBot, message: Message, user_id: int, duration
action="АВТОМУТ", action="АВТОМУТ",
user_id=user_id, user_id=user_id,
admin_id=None, # Автоматическое действие admin_id=None, # Автоматическое действие
reason=f"Использование нецензурной лексики (нарушение #{violations_count})", reason=f"{words_info} (нарушение #{violations_count})",
duration=time_display, duration=time_display,
) )
@@ -160,6 +169,9 @@ async def check_message_for_profanity(bot: AsyncTeleBot, message: Message):
if not contains_bad_word(message.text): if not contains_bad_word(message.text):
return return
# Получаем список найденных плохих слов
bad_words_found = get_bad_words_from_text(message.text)
# Получаем ID пользователя и чата # Получаем ID пользователя и чата
user_id = message.from_user.id user_id = message.from_user.id
chat_id = message.chat.id chat_id = message.chat.id
@@ -184,7 +196,7 @@ async def check_message_for_profanity(bot: AsyncTeleBot, message: Message):
mute_duration = get_mute_duration(violations_count) mute_duration = get_mute_duration(violations_count)
# Применяем мут # Применяем мут
await apply_mute(bot, message, user_id, mute_duration, violations_count) await apply_mute(bot, message, user_id, mute_duration, violations_count, bad_words_found)
def register_handlers(bot: AsyncTeleBot): def register_handlers(bot: AsyncTeleBot):
""" """

View File

@@ -0,0 +1,120 @@
from telebot.async_telebot import AsyncTeleBot
from telebot.types import Message
import logging
from database import db
from utils import check_admin_status, delete_messages
from config import COMMAND_MESSAGES
logger = logging.getLogger(__name__)
def register_handlers(bot: AsyncTeleBot):
"""Регистрирует обработчик команды сброса нарушений"""
@bot.message_handler(commands=['reset_violations'])
async def reset_violations_command(message: Message):
"""Команда для сброса счётчика нарушений пользователя"""
logger.info(f"Команда /reset_violations получена от пользователя {message.from_user.id}")
# Проверяем права администратора
admin_check = await check_admin_status(bot, message)
if admin_check == 1:
logger.info(f"Пользователь {message.from_user.id} не является администратором")
return
logger.info(f"Пользователь {message.from_user.id} прошел проверку прав администратора")
# Определяем целевого пользователя
target_user = None
target_user_id = None
# Обработка ответа на сообщение
if message.reply_to_message:
target_user = message.reply_to_message.from_user
target_user_id = target_user.id
# Обработка по username или ID
else:
parts = message.text.split(maxsplit=1)
if len(parts) < 2:
await send_temp_message(
bot,
message,
COMMAND_MESSAGES['reset_violations_help']
)
return
identifier = parts[1].strip()
# Попытка получить по username
if identifier.startswith('@'):
username = identifier[1:]
user_data = db.get_user_by_username(username)
if user_data:
target_user_id = user_data[0]
else:
await send_temp_message(
bot,
message,
COMMAND_MESSAGES['user_not_found']
)
return
# Попытка получить по ID
else:
try:
target_user_id = int(identifier)
except ValueError:
await send_temp_message(
bot,
message,
COMMAND_MESSAGES['user_not_found']
)
return
# Проверяем, что нашли пользователя
if not target_user_id:
await send_temp_message(
bot,
message,
COMMAND_MESSAGES['user_not_found']
)
return
# Получаем информацию о пользователе из базы
user_info = db.get_user(target_user_id)
# Получаем текущее количество нарушений
violations_count = db.get_violations_count(target_user_id, message.chat.id)
# Сбрасываем нарушения
deleted_count = db.reset_user_violations(target_user_id, message.chat.id)
# Формируем сообщение
if user_info:
_, nickname, tag = user_info
user_display = f"<b>{nickname}</b>"
if tag:
user_display += f" (@{tag})"
else:
user_display = f"<code>{target_user_id}</code>"
response = (
f"✅ Счётчик нарушений сброшен\n\n"
f"👤 Пользователь: {user_display}\n"
f"📊 Удалено нарушений: <b>{deleted_count}</b>"
)
await send_temp_message(bot, message, response, time_sleep=30)
logger.info(f"Администратор {message.from_user.id} сбросил счётчик нарушений пользователя {target_user_id}")
async def send_temp_message(bot: AsyncTeleBot, message: Message, text: str, time_sleep: int = 10):
"""Отправляет временное сообщение, которое удаляется через указанное время"""
await bot.send_message(
chat_id=message.chat.id,
text=text,
message_thread_id=message.message_thread_id,
)
await delete_messages(bot, message, time_sleep=time_sleep, number_message=2)