mirror of
https://github.com/yihong0618/bilingual_book_maker.git
synced 2025-06-05 19:15:34 +00:00
90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
import json
|
|
import time
|
|
|
|
import requests
|
|
import re
|
|
|
|
from book_maker.utils import LANGUAGES, TO_LANGUAGE_CODE
|
|
|
|
from .base_translator import Base
|
|
from rich import print
|
|
|
|
|
|
class DeepL(Base):
|
|
"""
|
|
DeepL translator
|
|
"""
|
|
|
|
def __init__(self, key, language, **kwargs) -> None:
|
|
super().__init__(key, language)
|
|
self.api_url = "https://deepl-translator.p.rapidapi.com/translate"
|
|
self.headers = {
|
|
"content-type": "application/json",
|
|
"X-RapidAPI-Key": "",
|
|
"X-RapidAPI-Host": "deepl-translator.p.rapidapi.com",
|
|
}
|
|
l = None
|
|
l = language if language in LANGUAGES else TO_LANGUAGE_CODE.get(language)
|
|
if l not in [
|
|
"bg",
|
|
"zh",
|
|
"cs",
|
|
"da",
|
|
"nl",
|
|
"en-US",
|
|
"en-GB",
|
|
"et",
|
|
"fi",
|
|
"fr",
|
|
"de",
|
|
"el",
|
|
"hu",
|
|
"id",
|
|
"it",
|
|
"ja",
|
|
"lv",
|
|
"lt",
|
|
"pl",
|
|
"pt-PT",
|
|
"pt-BR",
|
|
"ro",
|
|
"ru",
|
|
"sk",
|
|
"sl",
|
|
"es",
|
|
"sv",
|
|
"tr",
|
|
"uk",
|
|
"ko",
|
|
"nb",
|
|
]:
|
|
raise Exception(f"DeepL do not support {l}")
|
|
self.language = l
|
|
|
|
def rotate_key(self):
|
|
self.headers["X-RapidAPI-Key"] = f"{next(self.keys)}"
|
|
|
|
def translate(self, text):
|
|
self.rotate_key()
|
|
print(text)
|
|
payload = {"text": text, "source": "EN", "target": self.language}
|
|
try:
|
|
response = requests.request(
|
|
"POST",
|
|
self.api_url,
|
|
data=json.dumps(payload),
|
|
headers=self.headers,
|
|
)
|
|
except Exception as e:
|
|
print(e)
|
|
time.sleep(30)
|
|
response = requests.request(
|
|
"POST",
|
|
self.api_url,
|
|
data=json.dumps(payload),
|
|
headers=self.headers,
|
|
)
|
|
t_text = response.json().get("text", "")
|
|
print("[bold green]" + re.sub("\n{3,}", "\n\n", t_text) + "[/bold green]")
|
|
return t_text
|