mirror of
https://github.com/MCV-Software/TWBlue.git
synced 2025-08-10 14:56:08 -04:00
Feat: Replaced old translator module. The new translator can translate by using LibreTranslate or DeepL with an user provided API key
This commit is contained in:
1
src/extra/translator/engines/__init__.py
Normal file
1
src/extra/translator/engines/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
# -*- coding: utf-8 -*-
|
14
src/extra/translator/engines/deep_l.py
Normal file
14
src/extra/translator/engines/deep_l.py
Normal file
@@ -0,0 +1,14 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import config
|
||||
from deepl import Translator
|
||||
|
||||
def translate(text: str, target_language: str) -> str:
|
||||
key = config.app["translator"]["translator_api_key"]
|
||||
t = Translator(key)
|
||||
return t.translate_text(text, target_lang=target_language).text
|
||||
|
||||
def languages():
|
||||
key = config.app["translator"]["translator_api_key"]
|
||||
t = Translator(key)
|
||||
langs = t.get_target_languages()
|
||||
return langs
|
45
src/extra/translator/engines/libre_translate.py
Normal file
45
src/extra/translator/engines/libre_translate.py
Normal file
@@ -0,0 +1,45 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
""" Modified Libretranslatepy module which adds an user agent for making requests against more instances. """
|
||||
import json
|
||||
from typing import Any, Dict
|
||||
from urllib import request, parse
|
||||
from libretranslatepy import LibreTranslateAPI
|
||||
|
||||
class CustomLibreTranslateAPI(LibreTranslateAPI):
|
||||
USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
|
||||
|
||||
def _create_request(self, url: str, method: str, data: Dict[str, str]) -> request.Request:
|
||||
url_params = parse.urlencode(data)
|
||||
req = request.Request(url, method=method, data=url_params.encode())
|
||||
req.add_header("User-Agent", self.USER_AGENT)
|
||||
return req
|
||||
|
||||
def translate(self, q: str, source: str = "en", target: str = "es", timeout: int | None = None) -> Any:
|
||||
url = self.url + "translate"
|
||||
params: Dict[str, str] = {"q": q, "source": source, "target": target}
|
||||
if self.api_key is not None:
|
||||
params["api_key"] = self.api_key
|
||||
req = self._create_request(url=url, method="POST", data=params)
|
||||
response = request.urlopen(req, timeout=timeout)
|
||||
response_str = response.read().decode()
|
||||
return json.loads(response_str)["translatedText"]
|
||||
|
||||
def detect(self, q: str, timeout: int | None = None) -> Any:
|
||||
url = self.url + "detect"
|
||||
params: Dict[str, str] = {"q": q}
|
||||
if self.api_key is not None:
|
||||
params["api_key"] = self.api_key
|
||||
req = self._create_request(url=url, method="POST", data=params)
|
||||
response = request.urlopen(req, timeout=timeout)
|
||||
response_str = response.read().decode()
|
||||
return json.loads(response_str)
|
||||
|
||||
def languages(self, timeout: int | None = None) -> Any:
|
||||
url = self.url + "languages"
|
||||
params: Dict[str, str] = dict()
|
||||
if self.api_key is not None:
|
||||
params["api_key"] = self.api_key
|
||||
req = self._create_request(url=url, method="GET", data=params)
|
||||
response = request.urlopen(req, timeout=timeout)
|
||||
response_str = response.read().decode()
|
||||
return json.loads(response_str)
|
Reference in New Issue
Block a user