forked from Muzifs/LGBot
Добавление исправление ошибки работы кармы
This commit is contained in:
87
src/modules/0_karma_tracker.py
Normal file
87
src/modules/0_karma_tracker.py
Normal file
@@ -0,0 +1,87 @@
|
||||
from telebot.async_telebot import AsyncTeleBot
|
||||
from telebot.types import Message
|
||||
import logging
|
||||
|
||||
from database import db
|
||||
from thank_words import contains_thank_word
|
||||
from bad_words import contains_bad_word
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Кулдаун для благодарностей (в секундах) - 1 час
|
||||
THANK_COOLDOWN = 3600
|
||||
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
"""Регистрирует обработчики для отслеживания благодарностей"""
|
||||
|
||||
@bot.message_handler(func=lambda message: message.reply_to_message is not None and message.text and not message.text.startswith('/'))
|
||||
async def handle_thank_message(message: Message):
|
||||
"""
|
||||
Обрабатывает сообщения, которые являются ответами на другие сообщения.
|
||||
Если сообщение содержит благодарность, начисляет карму автору оригинального сообщения.
|
||||
"""
|
||||
try:
|
||||
# Проверяем, что это групповой чат
|
||||
if message.chat.type not in ['group', 'supergroup']:
|
||||
return
|
||||
|
||||
# Проверяем наличие благодарственных слов
|
||||
if not contains_thank_word(message.text):
|
||||
return
|
||||
|
||||
# Проверяем, что в сообщении нет мата (не начисляем карму за мат)
|
||||
if contains_bad_word(message.text):
|
||||
logger.info(f"Пользователь {message.from_user.id} написал благодарность с матом - карма не начислена")
|
||||
return
|
||||
|
||||
from_user = message.from_user
|
||||
to_user = message.reply_to_message.from_user
|
||||
chat_id = message.chat.id
|
||||
|
||||
# Защита от самоблагодарности
|
||||
if from_user.id == to_user.id:
|
||||
logger.info(f"Пользователь {from_user.id} попытался поблагодарить сам себя")
|
||||
return
|
||||
|
||||
# Проверяем, не является ли благодарность ботам
|
||||
if to_user.is_bot:
|
||||
logger.info(f"Пользователь {from_user.id} попытался поблагодарить бота")
|
||||
return
|
||||
|
||||
# Проверяем кулдаун (можно благодарить одного пользователя раз в час)
|
||||
if not db.can_thank(from_user.id, to_user.id, chat_id, THANK_COOLDOWN):
|
||||
logger.info(f"Пользователь {from_user.id} уже благодарил {to_user.id} недавно")
|
||||
# Молча игнорируем, чтобы не спамить
|
||||
return
|
||||
|
||||
# Начисляем карму
|
||||
db.add_karma(to_user.id, chat_id, 1)
|
||||
db.add_karma_history(from_user.id, to_user.id, chat_id)
|
||||
|
||||
# Получаем новую карму пользователя
|
||||
new_karma = db.get_karma(to_user.id, chat_id)
|
||||
|
||||
# Формируем имя пользователя для отображения
|
||||
to_user_name = to_user.first_name
|
||||
if to_user.username:
|
||||
to_user_display = f"@{to_user.username}"
|
||||
else:
|
||||
to_user_display = to_user_name
|
||||
|
||||
# Отправляем уведомление
|
||||
response = f"👍 Карма пользователя {to_user_display} увеличена! Текущая карма: {new_karma}"
|
||||
|
||||
sent_message = await bot.reply_to(message, response)
|
||||
|
||||
logger.info(f"Пользователь {from_user.id} поблагодарил {to_user.id}, карма: {new_karma}")
|
||||
|
||||
# Удаляем уведомление через 5 секунд
|
||||
import asyncio
|
||||
await asyncio.sleep(5)
|
||||
try:
|
||||
await bot.delete_message(chat_id, sent_message.message_id)
|
||||
except Exception as e:
|
||||
logger.error(f"Не удалось удалить уведомление о карме: {e}")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при обработке благодарности: {e}", exc_info=True)
|
Reference in New Issue
Block a user