feat: Añadir sistema de mensajes de bienvenida con traducción para Discord
- Nueva tabla 'welcome_messages' en la base de datos - Panel web con página de configuración de bienvenida (/welcome) - Listar, crear, editar y eliminar mensajes por servidor - Vista previa del mensaje - Plantillas predefinidas - Bot Discord: - Nuevo intent 'members' para detectar nuevos usuarios - Evento on_member_join que envía mensaje de bienvenida - Botones de traducción en mensajes de bienvenida - Actualizada configuración de MySQL en docker-compose.yml - Añadido logging de debug para traducciones
This commit is contained in:
@@ -9,14 +9,15 @@ import re
|
||||
import html
|
||||
|
||||
from botdiscord.config import get_discord_token, load_config, get_languages
|
||||
from botdiscord.database import init_db, get_active_languages, get_bot_languages, save_message
|
||||
from botdiscord.ui import PersistentTranslationView, ConfigView
|
||||
from botdiscord.database import init_db, get_active_languages, get_bot_languages, save_message, get_welcome_message
|
||||
from botdiscord.ui import PersistentTranslationView, ConfigView, WelcomeTranslationView
|
||||
from botdiscord.translate import get_reverse_mapping, load_lang_mappings
|
||||
|
||||
load_config()
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
intents.members = True
|
||||
bot = commands.Bot(command_prefix="!", intents=intents)
|
||||
|
||||
@bot.event
|
||||
@@ -24,8 +25,6 @@ async def on_ready():
|
||||
init_db()
|
||||
load_lang_mappings("discord")
|
||||
|
||||
# Registramos la vista persistente GLOBALMENTE
|
||||
# Esto asegura que cualquier botón con el custom_id adecuado sea escuchado
|
||||
bot.add_view(PersistentTranslationView())
|
||||
|
||||
print(f"Bot Discord conectado como {bot.user}")
|
||||
@@ -35,6 +34,70 @@ async def on_ready():
|
||||
except Exception as e:
|
||||
print(f"Error sync: {e}")
|
||||
|
||||
@bot.event
|
||||
async def on_member_join(member):
|
||||
print(f"[WELCOME] Nuevo miembro: {member.name} ({member.id}) en guild: {member.guild.id}")
|
||||
|
||||
if member.bot:
|
||||
print(f"[WELCOME] Ignorado: es bot")
|
||||
return
|
||||
|
||||
try:
|
||||
welcome_config = get_welcome_message(member.guild.id)
|
||||
print(f"[WELCOME] Config obtained: {welcome_config}")
|
||||
|
||||
if not welcome_config:
|
||||
print(f"[WELCOME] No hay configuración para guild {member.guild.id}")
|
||||
return
|
||||
|
||||
enabled = welcome_config.get('enabled', False) if hasattr(welcome_config, 'get') else welcome_config[2]
|
||||
if not enabled:
|
||||
print(f"[WELCOME] Bienvenida deshabilitada para guild {member.guild.id}")
|
||||
return
|
||||
|
||||
channel_id = welcome_config.get('channel_id') if hasattr(welcome_config, 'get') else welcome_config[0]
|
||||
message_template = welcome_config.get('message_content') if hasattr(welcome_config, 'get') else welcome_config[1]
|
||||
|
||||
print(f"[WELCOME] Channel ID: {channel_id}, Template: {message_template}")
|
||||
|
||||
if not channel_id or not message_template:
|
||||
return
|
||||
|
||||
channel = member.guild.get_channel(channel_id)
|
||||
if not channel:
|
||||
print(f"[WELCOME] Canal {channel_id} no encontrado")
|
||||
return
|
||||
|
||||
formatted_message = message_template.format(
|
||||
user_mention=member.mention,
|
||||
username=member.name,
|
||||
server_name=member.guild.name,
|
||||
member_count=member.guild.member_count
|
||||
)
|
||||
|
||||
text_escaped = html.escape(formatted_message)
|
||||
|
||||
welcome_msg = await channel.send(formatted_message)
|
||||
print(f"[WELCOME] Mensaje enviado: {welcome_msg.id}")
|
||||
|
||||
save_message(
|
||||
welcome_msg.id,
|
||||
member.guild.id,
|
||||
member.id,
|
||||
text_escaped,
|
||||
{},
|
||||
'discord'
|
||||
)
|
||||
|
||||
view = WelcomeTranslationView(member.guild.id)
|
||||
await welcome_msg.edit(content=formatted_message + "\n\n¿Traducir este mensaje?", view=view)
|
||||
print(f"[WELCOME] Vista de traducción añadida")
|
||||
|
||||
except Exception as e:
|
||||
import traceback
|
||||
print(f"[WELCOME] Error: {e}")
|
||||
traceback.print_exc()
|
||||
|
||||
def get_active_langs_for_guild(guild_id):
|
||||
from botdiscord.translate import get_reverse_mapping
|
||||
active = get_active_languages(guild_id)
|
||||
|
||||
@@ -165,6 +165,14 @@ def init_db():
|
||||
password_hash VARCHAR(255) NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4''')
|
||||
|
||||
cursor.execute('''CREATE TABLE IF NOT EXISTS welcome_messages
|
||||
(guild_id BIGINT PRIMARY KEY,
|
||||
channel_id BIGINT NOT NULL,
|
||||
message_content TEXT NOT NULL,
|
||||
enabled BOOLEAN DEFAULT TRUE,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4''')
|
||||
|
||||
conn.commit()
|
||||
cursor.close()
|
||||
else:
|
||||
@@ -212,6 +220,12 @@ def init_db():
|
||||
username TEXT NOT NULL UNIQUE,
|
||||
password_hash TEXT NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP)''')
|
||||
|
||||
c.execute('''CREATE TABLE IF NOT EXISTS welcome_messages
|
||||
(guild_id INTEGER PRIMARY KEY,
|
||||
channel_id INTEGER NOT NULL,
|
||||
message_content TEXT NOT NULL,
|
||||
enabled INTEGER DEFAULT 1)''')
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
@@ -593,3 +607,95 @@ def update_admin_password(admin_id: int, password_hash: str):
|
||||
c.execute("UPDATE admins SET password_hash = ? WHERE id = ?", (password_hash, admin_id))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
def save_welcome_message(guild_id: int, channel_id: int, message_content: str, enabled: bool = True):
|
||||
db_type = get_db_type()
|
||||
|
||||
if db_type == "mysql":
|
||||
conn = get_connection()
|
||||
cursor = conn.cursor()
|
||||
query = """INSERT INTO welcome_messages (guild_id, channel_id, message_content, enabled)
|
||||
VALUES (%s, %s, %s, %s)
|
||||
ON DUPLICATE KEY UPDATE channel_id = %s, message_content = %s, enabled = %s"""
|
||||
cursor.execute(query, (guild_id, channel_id, message_content, enabled, channel_id, message_content, enabled))
|
||||
conn.commit()
|
||||
cursor.close()
|
||||
else:
|
||||
conn = get_connection()
|
||||
c = conn.cursor()
|
||||
c.execute("""INSERT OR REPLACE INTO welcome_messages (guild_id, channel_id, message_content, enabled)
|
||||
VALUES (?, ?, ?, ?)""", (guild_id, channel_id, message_content, 1 if enabled else 0))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
def get_welcome_message(guild_id: int):
|
||||
db_type = get_db_type()
|
||||
|
||||
if db_type == "mysql":
|
||||
conn = get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute("SELECT channel_id, message_content, enabled FROM welcome_messages WHERE guild_id = %s", (guild_id,))
|
||||
row = cursor.fetchone()
|
||||
cursor.close()
|
||||
if row:
|
||||
return {
|
||||
'channel_id': int(row['channel_id']) if row['channel_id'] else 0,
|
||||
'message_content': row['message_content'] or '',
|
||||
'enabled': bool(row['enabled'])
|
||||
}
|
||||
return None
|
||||
else:
|
||||
conn = get_connection()
|
||||
c = conn.cursor()
|
||||
c.execute("SELECT channel_id, message_content, enabled FROM welcome_messages WHERE guild_id = ?", (guild_id,))
|
||||
row = c.fetchone()
|
||||
conn.close()
|
||||
if row:
|
||||
return {
|
||||
"channel_id": int(row[0]) if row[0] else 0,
|
||||
"message_content": row[1] or '',
|
||||
"enabled": bool(row[2])
|
||||
}
|
||||
return None
|
||||
|
||||
def get_all_welcome_configs():
|
||||
db_type = get_db_type()
|
||||
|
||||
if db_type == "mysql":
|
||||
conn = get_connection()
|
||||
cursor = conn.cursor(dictionary=True)
|
||||
cursor.execute("SELECT guild_id, channel_id, message_content, enabled FROM welcome_messages")
|
||||
rows = cursor.fetchall()
|
||||
cursor.close()
|
||||
result = []
|
||||
for row in rows:
|
||||
result.append({
|
||||
'guild_id': int(row['guild_id']) if row['guild_id'] else 0,
|
||||
'channel_id': int(row['channel_id']) if row['channel_id'] else 0,
|
||||
'message_content': row['message_content'] or '',
|
||||
'enabled': bool(row['enabled'])
|
||||
})
|
||||
return result
|
||||
else:
|
||||
conn = get_connection()
|
||||
c = conn.cursor()
|
||||
c.execute("SELECT guild_id, channel_id, message_content, enabled FROM welcome_messages")
|
||||
rows = [{"guild_id": r[0], "channel_id": r[1], "message_content": r[2], "enabled": bool(r[3])} for r in c.fetchall()]
|
||||
conn.close()
|
||||
return rows
|
||||
|
||||
def delete_welcome_message(guild_id: int):
|
||||
db_type = get_db_type()
|
||||
|
||||
if db_type == "mysql":
|
||||
conn = get_connection()
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("DELETE FROM welcome_messages WHERE guild_id = %s", (guild_id,))
|
||||
conn.commit()
|
||||
cursor.close()
|
||||
else:
|
||||
conn = get_connection()
|
||||
c = conn.cursor()
|
||||
c.execute("DELETE FROM welcome_messages WHERE guild_id = ?", (guild_id,))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
@@ -76,9 +76,14 @@ async def _do_translate_request(session, url, text, target_code):
|
||||
async def translate_text(text: str, target_lang: str) -> str:
|
||||
url = get_libretranslate_url()
|
||||
if not url:
|
||||
print(f"[TRANSLATE] URL no configurada")
|
||||
return text
|
||||
|
||||
|
||||
print(f"[TRANSLATE] target_lang recibido: {target_lang}")
|
||||
print(f"[TRANSLATE] NAME_TO_CODE: {NAME_TO_CODE}")
|
||||
|
||||
target_code = NAME_TO_CODE.get(target_lang, target_lang)
|
||||
print(f"[TRANSLATE] target_code resuelto: {target_code}")
|
||||
|
||||
# Segmentación mejorada
|
||||
segments = re.split(r'([.!?]+\s*|\n+)', text)
|
||||
|
||||
@@ -8,7 +8,8 @@ from botdiscord.database import (
|
||||
get_cached_translation,
|
||||
get_active_languages,
|
||||
get_bot_languages,
|
||||
get_available_languages # Importante: esta función trae los datos del panel web
|
||||
get_available_languages,
|
||||
save_message
|
||||
)
|
||||
|
||||
class TranslationButton(discord.ui.Button):
|
||||
@@ -109,3 +110,71 @@ class ConfigView(discord.ui.View):
|
||||
def __init__(self, guild_id: int, bot_type: str = "discord"):
|
||||
super().__init__()
|
||||
self.add_item(ConfigSelect(guild_id, bot_type))
|
||||
|
||||
class WelcomeTranslationButton(discord.ui.Button):
|
||||
def __init__(self, lang_name: str, lang_code: str, flag: str):
|
||||
label = flag if flag else lang_name
|
||||
custom_id = f"btn_welcome_{lang_code}"
|
||||
super().__init__(label=label, style=discord.ButtonStyle.primary, custom_id=custom_id)
|
||||
self.lang_code = lang_code
|
||||
self.lang_name = lang_name
|
||||
self.flag = flag
|
||||
|
||||
async def callback(self, interaction: discord.Interaction):
|
||||
await interaction.response.defer()
|
||||
|
||||
try:
|
||||
original_msg_id = interaction.message.id
|
||||
db_msg = get_message(original_msg_id)
|
||||
|
||||
if not db_msg:
|
||||
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'] or {}
|
||||
|
||||
cached = get_cached_translation(original_msg_id, self.lang_code)
|
||||
if cached:
|
||||
translated = cached
|
||||
else:
|
||||
translated = await translate_text(text, self.lang_code)
|
||||
save_translation(original_msg_id, self.lang_code, translated)
|
||||
|
||||
translated = html.unescape(translated)
|
||||
if mentions_map:
|
||||
for placeholder, mention in mentions_map.items():
|
||||
translated = translated.replace(placeholder, mention)
|
||||
translated = translated.replace(placeholder.replace(" ", ""), mention)
|
||||
|
||||
guild_id = interaction.guild_id
|
||||
active_codes = get_active_languages(guild_id)
|
||||
if not active_codes:
|
||||
active_codes = get_bot_languages("discord")
|
||||
|
||||
db_langs = get_available_languages()
|
||||
|
||||
new_view = discord.ui.View(timeout=None)
|
||||
for lang in db_langs:
|
||||
if lang['code'] in active_codes:
|
||||
new_view.add_item(WelcomeTranslationButton(lang['name'], lang['code'], lang.get('flag', '')))
|
||||
|
||||
await interaction.edit_original_response(content=translated, view=new_view)
|
||||
|
||||
except Exception as e:
|
||||
import traceback
|
||||
print(f"[ERROR Welcome UI] {e}")
|
||||
traceback.print_exc()
|
||||
await interaction.followup.send(f"❌ Error: {str(e)}", ephemeral=True)
|
||||
|
||||
class WelcomeTranslationView(discord.ui.View):
|
||||
def __init__(self, guild_id: int):
|
||||
super().__init__(timeout=None)
|
||||
active_codes = get_active_languages(guild_id)
|
||||
if not active_codes:
|
||||
active_codes = get_bot_languages("discord")
|
||||
|
||||
db_langs = get_available_languages()
|
||||
for lang in db_langs:
|
||||
if lang['code'] in active_codes:
|
||||
self.add_item(WelcomeTranslationButton(lang['name'], lang['code'], lang.get('flag', '')))
|
||||
|
||||
Reference in New Issue
Block a user