feat: implementar caché de traducciones y persistencia de mensajes en MySQL para Discord y Telegram
This commit is contained in:
@@ -10,7 +10,10 @@ from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
||||
from telegram.ext import Application, CommandHandler, MessageHandler, CallbackQueryHandler, ContextTypes, filters
|
||||
|
||||
from botdiscord.config import load_config, get_telegram_token, get_libretranslate_url
|
||||
from botdiscord.database import init_db as init_shared_db
|
||||
from botdiscord.database import (
|
||||
init_db as init_shared_db,
|
||||
save_message, get_message, save_translation, get_cached_translation
|
||||
)
|
||||
from botdiscord.translate import load_lang_mappings, get_lang_mapping, get_flag_mapping, get_name_to_code
|
||||
|
||||
load_config()
|
||||
@@ -132,9 +135,8 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
reply_markup=reply_markup
|
||||
)
|
||||
|
||||
with lock:
|
||||
cleanup_pending_translations()
|
||||
pending_translations[str(sent_message.message_id)] = text
|
||||
# Guardamos en MySQL en lugar del JSON local
|
||||
save_message(sent_message.message_id, update.effective_chat.id, update.effective_user.id, text, {}, 'telegram')
|
||||
|
||||
async def handle_photo(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
caption = update.message.caption
|
||||
@@ -147,23 +149,40 @@ async def handle_photo(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
parse_mode="Markdown",
|
||||
reply_markup=reply_markup
|
||||
)
|
||||
with lock:
|
||||
cleanup_pending_translations()
|
||||
pending_translations[str(sent_message.message_id)] = caption
|
||||
# Guardamos en MySQL
|
||||
save_message(sent_message.message_id, update.effective_chat.id, update.effective_user.id, caption, {}, 'telegram')
|
||||
|
||||
async def translation_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||
query = update.callback_query
|
||||
await query.answer()
|
||||
|
||||
lang_code = query.data.replace("trans_", "")
|
||||
message_id = str(query.message.message_id)
|
||||
message_id = query.message.message_id
|
||||
|
||||
text = pending_translations.get(message_id)
|
||||
if not text:
|
||||
await query.message.reply_text("⚠️ No se encontró el texto original. Es posible que el bot se haya reiniciado recientemente.")
|
||||
# Intentamos obtener del caché de traducciones primero
|
||||
cached = get_cached_translation(message_id, lang_code)
|
||||
|
||||
# Recuperamos el mensaje original de MySQL
|
||||
db_msg = get_message(message_id)
|
||||
|
||||
if not db_msg:
|
||||
await query.message.reply_text("⚠️ No se encontró el texto original en la base de datos.")
|
||||
return
|
||||
|
||||
translated = await translate_text_telegram(text, lang_code)
|
||||
text = db_msg['content']
|
||||
|
||||
if cached:
|
||||
translated = cached
|
||||
else:
|
||||
# Traducimos usando la lógica compartida
|
||||
from botdiscord.translate import translate_text
|
||||
translated = await translate_text(text, lang_code)
|
||||
# Guardamos en caché MySQL
|
||||
save_translation(message_id, lang_code, translated)
|
||||
|
||||
# Desescapamos el HTML para Telegram
|
||||
import html
|
||||
translated = html.unescape(translated)
|
||||
|
||||
lang_mapping = get_lang_mapping("telegram")
|
||||
flag_mapping = get_flag_mapping("telegram")
|
||||
|
||||
Reference in New Issue
Block a user