|
|
|
|
|
from functools import cached_property |
|
import uuid |
|
import hmac, base64 |
|
import aiohttp |
|
import time |
|
import requests |
|
import re |
|
|
|
from .common import CommonTranslator, InvalidServerResponse |
|
|
|
class PapagoTranslator(CommonTranslator): |
|
_LANGUAGE_CODE_MAP = { |
|
'CHS': 'zh-CN', |
|
'CHT': 'zh-TW', |
|
'JPN': 'ja', |
|
'ENG': 'en', |
|
'KOR': 'ko', |
|
'VIN': 'vi', |
|
'FRA': 'fr', |
|
'DEU': 'de', |
|
'ITA': 'it', |
|
'PTB': 'pt', |
|
'RUS': 'ru', |
|
'ESP': 'es', |
|
'THA': 'th', |
|
'IND': 'id' |
|
} |
|
_API_URL = 'https://papago.naver.com/apis/n2mt/translate' |
|
|
|
async def _translate(self, from_lang, to_lang, queries): |
|
data = {} |
|
data['honorific'] = "false" |
|
data['source'] = from_lang |
|
data['target'] = to_lang |
|
data['text'] = '\n'.join(queries) |
|
result = await self._do_request(data, self._version_key) |
|
if "translatedText" not in result: |
|
raise InvalidServerResponse(f'Papago returned invalid response: {result}\nAre the API keys set correctly?') |
|
result_list = [str.strip() for str in result["translatedText"].split("\n")] |
|
return result_list |
|
|
|
@cached_property |
|
def _version_key(self): |
|
script = requests.get('https://papago.naver.com') |
|
mainJs = re.search(r'\/(main.*\.js)', script.text).group(1) |
|
papagoVerData = requests.get('https://papago.naver.com/' + mainJs) |
|
papagoVer = re.search(r'"PPG .*,"(v[^"]*)', papagoVerData.text).group(1) |
|
return papagoVer |
|
|
|
async def _do_request(self, data, version_key): |
|
guid = uuid.uuid4() |
|
timestamp = int(time.time() * 1000) |
|
key = version_key.encode("utf-8") |
|
code = f"{guid}\n{self._API_URL}\n{timestamp}".encode("utf-8") |
|
token = base64.b64encode(hmac.new(key, code, "MD5").digest()).decode("utf-8") |
|
headers = { |
|
"Authorization": f"PPG {guid}:{token}", |
|
"Content-Type": "application/x-www-form-urlencoded; charset=UTF-8", |
|
"Timestamp": str(timestamp), |
|
} |
|
async with aiohttp.ClientSession() as session: |
|
async with session.post(self._API_URL, data=data, headers=headers) as resp: |
|
return await resp.json() |
|
|
