94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
import aiohttp
|
|
from botdiscord.config import get_libretranslate_url, get_languages
|
|
from botdiscord.database import get_available_languages, get_bot_languages
|
|
|
|
def load_lang_mappings(bot_type: str = None):
|
|
global LANG_MAPPING, REVERSE_MAPPING, FLAG_MAPPING, _cached_bot_type, NAME_TO_CODE
|
|
|
|
if bot_type:
|
|
_cached_bot_type = bot_type
|
|
|
|
available = get_available_languages()
|
|
|
|
if not available:
|
|
from botdiscord.config import get_languages
|
|
available = get_languages()
|
|
print(f"[DEBUG] Idiomas desde config: {available}")
|
|
|
|
all_codes = [lang["code"] for lang in available]
|
|
print(f"[DEBUG] Códigos disponibles: {all_codes}")
|
|
|
|
if _cached_bot_type:
|
|
active_codes = get_bot_languages(_cached_bot_type)
|
|
print(f"[DEBUG] Códigos activos para {_cached_bot_type}: {active_codes}")
|
|
if not active_codes:
|
|
active_codes = all_codes
|
|
else:
|
|
active_codes = all_codes
|
|
|
|
if not active_codes:
|
|
active_codes = all_codes
|
|
|
|
name_to_code = {lang["name"]: lang["code"] for lang in available if lang["code"] in active_codes}
|
|
code_to_name = {lang["code"]: lang["name"] for lang in available if lang["code"] in active_codes}
|
|
flag_dict = {lang["code"]: lang.get("flag", "") for lang in available}
|
|
|
|
print(f"[DEBUG] FLAG_MAPPING: {flag_dict}")
|
|
print(f"[DEBUG] NAME_TO_CODE: {name_to_code}")
|
|
|
|
LANG_MAPPING = code_to_name
|
|
NAME_TO_CODE = name_to_code
|
|
FLAG_MAPPING = flag_dict
|
|
REVERSE_MAPPING = code_to_name
|
|
|
|
_cached_bot_type = None
|
|
LANG_MAPPING = {}
|
|
REVERSE_MAPPING = {}
|
|
FLAG_MAPPING = {}
|
|
NAME_TO_CODE = {}
|
|
|
|
async def translate_text(text: str, target_lang: str) -> str:
|
|
url = get_libretranslate_url()
|
|
# Nos aseguramos de enviar el CÓDIGO del idioma (ej. 'en') a la API.
|
|
# Si recibimos un nombre (ej. 'English'), NAME_TO_CODE lo convierte a 'en'.
|
|
# Si ya recibimos un código (ej. 'en'), lo usamos directamente.
|
|
target_code = NAME_TO_CODE.get(target_lang, target_lang)
|
|
|
|
payload = {
|
|
"q": text,
|
|
"source": "auto",
|
|
"target": target_code,
|
|
"format": "text"
|
|
}
|
|
|
|
async with aiohttp.ClientSession() as session:
|
|
try:
|
|
async with session.post(url, json=payload, timeout=10) as resp:
|
|
if resp.status == 200:
|
|
data = await resp.json()
|
|
return data.get("translatedText", "Error en la traducción.")
|
|
else:
|
|
return f"Error de API: {resp.status}"
|
|
except Exception as e:
|
|
return f"Error de conexión: {str(e)}"
|
|
|
|
def get_lang_mapping(bot_type: str = None) -> dict:
|
|
load_lang_mappings(bot_type)
|
|
return LANG_MAPPING.copy()
|
|
|
|
def get_reverse_mapping(bot_type: str = None) -> dict:
|
|
load_lang_mappings(bot_type)
|
|
return REVERSE_MAPPING.copy()
|
|
|
|
def get_flag_mapping(bot_type: str = None) -> dict:
|
|
load_lang_mappings(bot_type)
|
|
return FLAG_MAPPING.copy()
|
|
|
|
def get_name_to_code(bot_type: str = None) -> dict:
|
|
load_lang_mappings(bot_type)
|
|
print(f"[DEBUG] get_name_to_code returning: {NAME_TO_CODE}")
|
|
return NAME_TO_CODE.copy()
|
|
|
|
def get_lang_flag(lang_code: str) -> str:
|
|
return FLAG_MAPPING.get(lang_code, "")
|