mirror of
https://github.com/MCV-Software/TWBlue.git
synced 2025-04-11 13:42:29 -04:00
62 lines
2.9 KiB
Python
62 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
import logging
|
|
import threading
|
|
import wx
|
|
import config
|
|
from pubsub import pub
|
|
from . engines import libre_translate, deep_l
|
|
from .wx_ui import translateDialog
|
|
|
|
log = logging.getLogger("extras.translator")
|
|
|
|
class TranslatorController(object):
|
|
def __init__(self, text):
|
|
super(TranslatorController, self).__init__()
|
|
self.text = text
|
|
self.languages = []
|
|
self.response = False
|
|
self.dialog = translateDialog()
|
|
pub.subscribe(self.on_engine_changed, "translator.engine_changed")
|
|
if config.app["translator"]["engine"] == "libretranslate":
|
|
self.dialog.engine_select.SetSelection(0)
|
|
elif config.app["translator"]["engine"] == "deepl":
|
|
self.dialog.engine_select.SetSelection(1)
|
|
threading.Thread(target=self.load_languages).start()
|
|
if self.dialog.ShowModal() == wx.ID_OK:
|
|
self.response = True
|
|
for k in self.language_dict:
|
|
if self.language_dict[k] == self.dialog.dest_lang.GetStringSelection():
|
|
self.target_language= k
|
|
pub.unsubscribe(self.on_engine_changed, "translator.engine_changed")
|
|
|
|
def load_languages(self):
|
|
self.language_dict = self.get_languages()
|
|
self.languages = [self.language_dict[k] for k in self.language_dict]
|
|
self.dialog.set_languages(self.languages)
|
|
|
|
def on_engine_changed(self, engine):
|
|
if engine == "LibreTranslate":
|
|
config.app["translator"]["engine"] = engine.lower()
|
|
elif engine == "DeepL":
|
|
config.app["translator"]["engine"] = engine.lower()
|
|
config.app.write()
|
|
threading.Thread(target=self.load_languages).start()
|
|
|
|
def translate(self):
|
|
log.debug("Received translation request for language %s, text=%s" % (self.target_language, self.text))
|
|
if config.app["translator"].get("engine") == "libretranslate":
|
|
translator = libre_translate.CustomLibreTranslateAPI(config.app["translator"]["translator_api_url"])
|
|
vars = dict(q=self.text, target=self.target_language)
|
|
return translator.translate(**vars)
|
|
elif config.app["translator"]["engine"] == "deepl" and config.app["translator"]["translator_api_key"] != "":
|
|
return deep_l.translate(text=self.text, target_language=self.target_language)
|
|
|
|
def get_languages(self):
|
|
languages = {}
|
|
if config.app["translator"].get("engine") == "libretranslate":
|
|
translator = libre_translate.CustomLibreTranslateAPI(config.app["translator"]["translator_api_url"])
|
|
languages = {l.get("code"): l.get("name") for l in translator.languages()}
|
|
elif config.app["translator"]["engine"] == "deepl" and config.app["translator"]["translator_api_key"] != "":
|
|
languages = {language.code: language.name for language in deep_l.languages()}
|
|
return dict(sorted(languages.items(), key=lambda x: x[1]))
|