|
|
|
|
|
import uuid |
|
import hashlib |
|
import time |
|
import aiohttp |
|
import time |
|
|
|
from .common import CommonTranslator, InvalidServerResponse, MissingAPIKeyException |
|
from .keys import YOUDAO_APP_KEY, YOUDAO_SECRET_KEY |
|
|
|
def sha256_encode(signStr): |
|
hash_algorithm = hashlib.sha256() |
|
hash_algorithm.update(signStr.encode('utf-8')) |
|
return hash_algorithm.hexdigest() |
|
|
|
class YoudaoTranslator(CommonTranslator): |
|
_LANGUAGE_CODE_MAP = { |
|
'CHS': 'zh-CHS', |
|
'JPN': "ja", |
|
'ENG': 'en', |
|
'KOR': 'ko', |
|
'VIN': 'vi', |
|
'CSY': 'cs', |
|
'NLD': 'nl', |
|
'FRA': 'fr', |
|
'DEU': 'de', |
|
'HUN': 'hu', |
|
'ITA': 'it', |
|
'PLK': 'pl', |
|
'PTB': 'pt', |
|
'ROM': 'ro', |
|
'RUS': 'ru', |
|
'ESP': 'es', |
|
'TRK': 'tr', |
|
'THA': 'th', |
|
'IND': 'id' |
|
} |
|
_API_URL = 'https://openapi.youdao.com/api' |
|
|
|
def __init__(self): |
|
super().__init__() |
|
if not YOUDAO_APP_KEY or not YOUDAO_SECRET_KEY: |
|
raise MissingAPIKeyException('Please set the YOUDAO_APP_KEY and YOUDAO_SECRET_KEY environment variables before using the youdao translator.') |
|
|
|
async def _translate(self, from_lang, to_lang, queries): |
|
data = {} |
|
query_text = '\n'.join(queries) |
|
data['from'] = from_lang |
|
data['to'] = to_lang |
|
data['signType'] = 'v3' |
|
curtime = str(int(time.time())) |
|
data['curtime'] = curtime |
|
salt = str(uuid.uuid1()) |
|
signStr = YOUDAO_APP_KEY + self._truncate(query_text) + salt + curtime + YOUDAO_SECRET_KEY |
|
sign = sha256_encode(signStr) |
|
data['appKey'] = YOUDAO_APP_KEY |
|
data['q'] = query_text |
|
data['salt'] = salt |
|
data['sign'] = sign |
|
|
|
|
|
result = await self._do_request(data) |
|
result_list = [] |
|
if "translation" not in result: |
|
raise InvalidServerResponse(f'Youdao returned invalid response: {result}\nAre the API keys set correctly?') |
|
for ret in result["translation"]: |
|
result_list.extend(ret.split('\n')) |
|
return result_list |
|
|
|
def _truncate(self, q): |
|
if q is None: |
|
return None |
|
size = len(q) |
|
return q if size <= 20 else q[0:10] + str(size) + q[size - 10:size] |
|
|
|
async def _do_request(self, data): |
|
headers = {'Content-Type': 'application/x-www-form-urlencoded'} |
|
async with aiohttp.ClientSession() as session: |
|
async with session.post(self._API_URL, data=data, headers=headers) as resp: |
|
return await resp.json() |
|
|
