feat(discord): proteger menciones y mejorar traducción HTML para mensajes multilínea

This commit is contained in:
2026-03-06 19:28:48 -06:00
parent e4d50b6eb5
commit 7601979d3e
3 changed files with 29 additions and 7 deletions

View File

@@ -77,17 +77,21 @@ async def on_message(message):
if not active_langs:
return
text_to_translate = message.content
import html
text_escaped = html.escape(message.content)
mention_pattern = re.compile(r'<@!?(\d+)>|<@&(\d+)>|<#(\d+)>')
# Buscamos menciones que ya están escapadas (&lt;@...&gt;)
mention_pattern = re.compile(r'&lt;@!?(\d+)&gt;|&lt;@&(\d+)&gt;|&lt;#(\d+)&gt;')
mentions_map = {}
def replace_mention(match):
placeholder = f"__MENTION_{len(mentions_map)}__"
mentions_map[placeholder] = match.group(0)
# Usamos una etiqueta autocerrada para que el traductor no intente cerrarla él mismo
placeholder = f"<m{len(mentions_map)} />"
# Guardamos la mención original (sin escapar) para restaurarla luego
mentions_map[placeholder] = html.unescape(match.group(0))
return placeholder
text_to_translate = mention_pattern.sub(replace_mention, text_to_translate)
text_to_translate = mention_pattern.sub(replace_mention, text_escaped)
langs_to_show = active_langs

View File

@@ -58,7 +58,7 @@ async def translate_text(text: str, target_lang: str) -> str:
"q": text,
"source": "auto",
"target": target_code,
"format": "text"
"format": "html"
}
async with aiohttp.ClientSession() as session:

View File

@@ -34,9 +34,27 @@ class TranslationButton(discord.ui.Button):
# Traducimos el texto
translated = await translate_text(self.text, self.lang_code)
# Desescapamos el HTML para recuperar caracteres especiales
import html
import re
translated = html.unescape(translated)
# Reemplazamos menciones si existen
for placeholder, mention in self.mentions_map.items():
translated = translated.replace(placeholder, mention)
# Extraer el número del tag, ej: m0 de <m0 />
match = re.search(r'm\d+', placeholder)
if not match: continue
tag_num = match.group()
# 1. Reemplazamos la etiqueta de apertura (o autocontenida) por la mención
# Patrón: < \s* mX \s* /? \s* >
open_pattern = re.compile(rf'<\s*{tag_num}\s*/?\s*>')
translated = open_pattern.sub(mention, translated)
# 2. Eliminamos cualquier etiqueta de cierre que el traductor haya "inventado"
# Patrón: < \s* / \s* mX \s* >
close_pattern = re.compile(rf'<\s*/\s*{tag_num}\s*>')
translated = close_pattern.sub('', translated)
# En lugar de enviar un mensaje nuevo, EDITAMOS el mensaje actual (el de los botones)
if self.attachments: