114 lines
3.8 KiB
Python
114 lines
3.8 KiB
Python
import aiohttp
|
||
from ..config import config
|
||
from nonebot.log import logger
|
||
|
||
|
||
async def translate(text: str, to: str):
|
||
# en,jp,zh
|
||
result = await translate_deepl(text, to) or await translate_bing(text, to) or await translate_google_proxy(text, to) or await translate_youdao(text, to)
|
||
if result:
|
||
return result
|
||
else:
|
||
logger.error(f"未找到可用的翻译引擎！")
|
||
return text
|
||
|
||
|
||
async def translate_bing(text: str, to: str):
|
||
"""
|
||
en,jp,zh_Hans
|
||
"""
|
||
if to == "zh":
|
||
to = "zh-Hans"
|
||
key = config.bing_key
|
||
if not key:
|
||
return None
|
||
header = {
|
||
"Ocp-Apim-Subscription-Key": key,
|
||
"Content-Type": "application/json",
|
||
}
|
||
async with aiohttp.ClientSession() as session:
|
||
body = [{'text': text}]
|
||
params = {
|
||
"api-version": "3.0",
|
||
"to": to,
|
||
"profanityAction": "Deleted",
|
||
}
|
||
async with session.post('https://api.cognitive.microsofttranslator.com/translate', json=body, params=params, headers=header) as resp:
|
||
if resp.status != 200:
|
||
logger.error(f"Bing翻译接口调用失败,错误代码{resp.status},{await resp.text()}")
|
||
return None
|
||
jsonresult = await resp.json()
|
||
result=jsonresult[0]["translations"][0]["text"]
|
||
logger.debug(f"Bing翻译启动，获取到{text},翻译后{result}")
|
||
return result
|
||
|
||
|
||
async def translate_deepl(text: str, to: str):
|
||
"""
|
||
EN,JA,ZH
|
||
"""
|
||
to = to.upper()
|
||
key = config.deepl_key
|
||
if not key:
|
||
return None
|
||
async with aiohttp.ClientSession() as session:
|
||
params = {
|
||
"auth_key":key,
|
||
"text": text,
|
||
"target_lang": to,
|
||
}
|
||
async with session.get('https://api-free.deepl.com/v2/translate', params=params) as resp:
|
||
if resp.status != 200:
|
||
logger.error(f"DeepL翻译接口调用失败,错误代码{resp.status},{await resp.text()}")
|
||
return None
|
||
jsonresult = await resp.json()
|
||
result=jsonresult["translations"][0]["text"]
|
||
logger.debug(f"DeepL翻译启动，获取到{text},翻译后{result}")
|
||
return result
|
||
|
||
|
||
async def translate_youdao(input: str, type: str):
|
||
"""
|
||
默认auto
|
||
ZH_CH2EN 中译英
|
||
EN2ZH_CN 英译汉
|
||
"""
|
||
if type == "zh":
|
||
type = "EN2ZH_CN"
|
||
elif type == "en":
|
||
type = "ZH_CH2EN"
|
||
async with aiohttp.ClientSession() as session:
|
||
data = {
|
||
'doctype': 'json',
|
||
'type': type,
|
||
'i': input
|
||
}
|
||
async with session.post("http://fanyi.youdao.com/translate", data=data) as resp:
|
||
if resp.status != 200:
|
||
logger.error(f"有道翻译接口调用失败,错误代码{resp.status},{await resp.text()}")
|
||
return None
|
||
result = await resp.json()
|
||
result=result["translateResult"][0][0]["tgt"]
|
||
logger.debug(f"有道翻译启动，获取到{input},翻译后{result}")
|
||
return result
|
||
|
||
|
||
async def translate_google_proxy(input: str, to: str):
|
||
"""
|
||
en,jp,zh 需要来源语言
|
||
"""
|
||
if to == "zh":
|
||
from_ = "en"
|
||
else:
|
||
from_="zh"
|
||
async with aiohttp.ClientSession()as session:
|
||
data = {"data": [input, from_, to]}
|
||
async with session.post("https://hf.space/embed/mikeee/gradio-gtr/+/api/predict", json=data)as resp:
|
||
if resp.status != 200:
|
||
logger.error(f"谷歌代理翻译接口调用失败,错误代码{resp.status},{await resp.text()}")
|
||
return None
|
||
result = await resp.json()
|
||
result=result["data"][0]
|
||
logger.debug(f"谷歌代理翻译启动，获取到{input},翻译后{result}")
|
||
return result
|