Añadiendo todos los archivos del proyecto (incluidos secretos y venv)
This commit is contained in:
93
botdiscord/translate.py
Normal file
93
botdiscord/translate.py
Normal file
@@ -0,0 +1,93 @@
|
||||
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, "")
|
||||
Reference in New Issue
Block a user