forked from Muzifs/LGBot
Исправления критикал проблем
This commit is contained in:
@@ -1,16 +1,15 @@
|
||||
from telebot.async_telebot import AsyncTeleBot
|
||||
from telebot.types import Message
|
||||
import asyncio
|
||||
import logging
|
||||
|
||||
from database import db
|
||||
from thank_words import contains_thank_word
|
||||
from bad_words import contains_bad_word
|
||||
from config import THANK_COOLDOWN
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Кулдаун для благодарностей (в секундах) - 1 час
|
||||
THANK_COOLDOWN = 3600
|
||||
|
||||
def register_handlers(bot: AsyncTeleBot):
|
||||
"""Регистрирует обработчики для отслеживания благодарностей"""
|
||||
logger.info("Регистрация обработчика благодарностей (karma_tracker)")
|
||||
@@ -55,15 +54,15 @@ def register_handlers(bot: AsyncTeleBot):
|
||||
logger.info(f"Пользователь {from_user.id} попытался поблагодарить бота")
|
||||
return
|
||||
|
||||
# Проверяем кулдаун (можно благодарить одного пользователя раз в час)
|
||||
if not db.can_thank(from_user.id, to_user.id, chat_id, THANK_COOLDOWN):
|
||||
# Атомарно проверяем кулдаун и записываем благодарность
|
||||
# Это предотвращает race condition при параллельных запросах
|
||||
if not db.try_add_karma_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)
|
||||
@@ -83,7 +82,6 @@ def register_handlers(bot: AsyncTeleBot):
|
||||
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)
|
||||
|
Reference in New Issue
Block a user