mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-03-06 08:17:36 +08:00
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import random
|
|
from hashlib import md5
|
|
|
|
import requests
|
|
|
|
from config import conf
|
|
from translate.translator import Translator
|
|
|
|
|
|
class BaiduTranslator(Translator):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
endpoint = "http://api.fanyi.baidu.com"
|
|
path = "/api/trans/vip/translate"
|
|
self.url = endpoint + path
|
|
self.appid = conf().get("baidu_translate_app_id")
|
|
self.appkey = conf().get("baidu_translate_app_key")
|
|
if not self.appid or not self.appkey:
|
|
raise Exception("baidu translate appid or appkey not set")
|
|
|
|
# For list of language codes, please refer to `https://api.fanyi.baidu.com/doc/21`, need to convert to ISO 639-1 codes
|
|
def translate(self, query: str, from_lang: str = "", to_lang: str = "en") -> str:
|
|
if not from_lang:
|
|
from_lang = "auto" # baidu suppport auto detect
|
|
salt = random.randint(32768, 65536)
|
|
sign = self.make_md5("{}{}{}{}".format(self.appid, query, salt, self.appkey))
|
|
headers = {"Content-Type": "application/x-www-form-urlencoded"}
|
|
payload = {"appid": self.appid, "q": query, "from": from_lang, "to": to_lang, "salt": salt, "sign": sign}
|
|
|
|
retry_cnt = 3
|
|
while retry_cnt:
|
|
r = requests.post(self.url, params=payload, headers=headers)
|
|
result = r.json()
|
|
errcode = result.get("error_code", "52000")
|
|
if errcode != "52000":
|
|
if errcode == "52001" or errcode == "52002":
|
|
retry_cnt -= 1
|
|
continue
|
|
else:
|
|
raise Exception(result["error_msg"])
|
|
else:
|
|
break
|
|
text = "\n".join([item["dst"] for item in result["trans_result"]])
|
|
return text
|
|
|
|
def make_md5(self, s, encoding="utf-8"):
|
|
return md5(s.encode(encoding)).hexdigest()
|