nickpons666/traduccion_bots
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(discord): solución definitiva de persistencia con registro global y custom_ids estáticos

Browse Source
This commit is contained in:
nickpons666
2026-03-07 00:59:20 -06:00
parent d82ed4b29e
commit e3b1d06442
2 changed files with 68 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
botdiscord/ui.py
View File

@@ -1,49 +1,37 @@
import discord
import html
import re
from botdiscord.translate import get_lang_mapping, get_flag_mapping, get_name_to_code, translate_text
from botdiscord.database import get_message, save_translation, get_cached_translation
class TranslationView(discord.ui.View):
def __init__(self, languages: list = None):
super().__init__(timeout=None)
if languages:
flag_mapping = get_flag_mapping()
name_to_code = get_name_to_code()
for lang in languages:
lang_code = name_to_code.get(lang)
flag = flag_mapping.get(lang_code, "") if lang_code else ""
# El custom_id es vital para la persistencia
custom_id = f"btn_trans_{lang_code}"
self.add_item(TranslationButton(lang, lang_code, flag, custom_id))
class TranslationButton(discord.ui.Button):
def __init__(self, lang_name: str, lang_code: str, flag: str, custom_id: str):
def __init__(self, lang_name: str, lang_code: str, flag: str):
label = flag if flag else lang_name
# custom_id estático: esto es lo que Discord usa para vincular el clic
custom_id = f"btn_trans_{lang_code}"
super().__init__(label=label, style=discord.ButtonStyle.primary, custom_id=custom_id)
self.lang_code = lang_code
async def callback(self, interaction: discord.Interaction):
# 1. Avisamos a Discord que estamos procesando (evita el "Interacción fallida" por timeout)
# Evita el "Interacción fallida" por timeout de 3s
await interaction.response.defer()
try:
if not interaction.message.reference:
await interaction.followup.send("⚠️ No se pudo encontrar la referencia al mensaje original.", ephemeral=True)
await interaction.followup.send("⚠️ No se pudo encontrar el mensaje original.", ephemeral=True)
return
original_msg_id = interaction.message.reference.message_id
db_msg = get_message(original_msg_id)
if not db_msg:
# Si no está en MySQL, intentamos ver si el mensaje actual tiene el texto (fallback)
await interaction.followup.send("⚠️ Mensaje no encontrado en la base de datos.", ephemeral=True)
return
text = db_msg['content']
mentions_map = db_msg['mentions_map']
# Verificamos caché
# Recuperar o traducir
cached = get_cached_translation(original_msg_id, self.lang_code)
if cached:
translated = cached
@@ -51,27 +39,30 @@ class TranslationButton(discord.ui.Button):
translated = await translate_text(text, self.lang_code)
save_translation(original_msg_id, self.lang_code, translated)
# Procesar traducción
import html
import re
# Limpiar texto
translated = html.unescape(translated)
if mentions_map:
for placeholder, mention in mentions_map.items():
match = re.search(r'm\d+', placeholder)
if not match: continue
tag_num = match.group()
open_pattern = re.compile(rf'<\s*{tag_num}\s*/?\s*>')
translated = open_pattern.sub(mention, translated)
close_pattern = re.compile(rf'<\s*/\s*{tag_num}\s*>')
translated = close_pattern.sub('', translated)
# El mapa ya viene con el formato <m0 />: "<@123>"
translated = translated.replace(placeholder, mention)
# Por si el traductor quitó el espacio: <m0/>
translated = translated.replace(placeholder.replace(" ", ""), mention)
# 2. Usamos edit_original_response porque ya hicimos defer()
# Editar el mensaje que contiene los botones
await interaction.edit_original_response(content=translated, view=self.view)
except Exception as e:
print(f"[ERROR UI] Error en callback de traducción: {e}")
await interaction.followup.send(f"❌ Error al traducir: {str(e)}", ephemeral=True)
print(f"[ERROR UI] {e}")
await interaction.followup.send(f"❌ Error: {str(e)}", ephemeral=True)
class PersistentTranslationView(discord.ui.View):
def __init__(self):
super().__init__(timeout=None)
# Cargamos los botones para todos los idiomas habilitados en la config
# Esto asegura que cualquier botón de cualquier idioma sea reconocido
from botdiscord.config import get_languages
for lang in get_languages():
self.add_item(TranslationButton(lang['name'], lang['code'], lang.get('flag', '')))
class ConfigSelect(discord.ui.Select):
def __init__(self, guild_id: int, bot_type: str = "discord"):
@@ -82,38 +73,15 @@ class ConfigSelect(discord.ui.Select):
options = []
for name, code in lang_mapping.items():
flag = flag_mapping.get(code, "")
if flag:
options.append(discord.SelectOption(label=flag, value=name, default=(code in active)))
else:
options.append(discord.SelectOption(label=name, value=name, default=(code in active)))
super().__init__(
placeholder="Selecciona los idiomas activos...",
min_values=0,
max_values=len(options),
options=options
)
options.append(discord.SelectOption(label=f"{flag} {name}" if flag else name, value=name, default=(code in active)))
super().__init__(placeholder="Selecciona idiomas...", min_values=0, max_values=len(options), options=options)
async def callback(self, interaction: discord.Interaction):
from botdiscord.database import set_active_languages
from botdiscord.translate import get_lang_mapping, get_flag_mapping
guild_id = interaction.guild_id
lang_mapping = get_lang_mapping("discord")
flag_mapping = get_flag_mapping("discord")
selected_codes = [lang_mapping[val] for val in self.values]
set_active_languages(guild_id, selected_codes)
selected_flags = []
for val in self.values:
code = lang_mapping.get(val)
flag = flag_mapping.get(code, "") if code else ""
selected_flags.append(flag if flag else val)
await interaction.response.send_message(
f"Configuración actualizada: {', '.join(selected_flags)}",
ephemeral=True
)
set_active_languages(interaction.guild_id, selected_codes)
await interaction.response.send_message("Configuración actualizada.", ephemeral=True)
class ConfigView(discord.ui.View):
def __init__(self, guild_id: int, bot_type: str = "discord"):