|
|
from scripts.physton_prompt.translator.base_tanslator import BaseTranslator |
|
|
import requests |
|
|
import datetime |
|
|
import hashlib |
|
|
import base64 |
|
|
import hmac |
|
|
import json |
|
|
from scripts.physton_prompt.get_lang import get_lang |
|
|
|
|
|
|
|
|
class IflytekV1Translator(BaseTranslator): |
|
|
def __init__(self): |
|
|
super().__init__('iflytekV1') |
|
|
|
|
|
def translate(self, text): |
|
|
if not text: |
|
|
return '' |
|
|
app_id = self.api_config.get('app_id', '') |
|
|
if not app_id: |
|
|
raise Exception(get_lang('is_required', {'0': 'APP ID'})) |
|
|
api_secret = self.api_config.get('api_secret', '') |
|
|
if not api_secret: |
|
|
raise Exception(get_lang('is_required', {'0': 'API Secret'})) |
|
|
api_key = self.api_config.get('api_key', '') |
|
|
if not api_key: |
|
|
raise Exception(get_lang('is_required', {'0': 'API Key'})) |
|
|
|
|
|
response = translate(text, From=self.from_lang, To=self.to_lang, APPID=app_id, Secret=api_secret, APIKey=api_key) |
|
|
if response.status_code != 200: |
|
|
raise Exception(get_lang('request_error', {'0': 'iflytekV1'})) |
|
|
if not response.text: |
|
|
raise Exception(get_lang('response_is_empty', {'0': 'iflytekV1'})) |
|
|
result = response.json() |
|
|
if 'code' not in result: |
|
|
raise Exception(get_lang('no_response_from', {'0': 'iflytekV1'})) |
|
|
if result['code'] != 0: |
|
|
raise Exception(result['message']) |
|
|
return result['data']['result']['trans_result']['dst'] |
|
|
|
|
|
|
|
|
def hashlib_256(res): |
|
|
m = hashlib.sha256(bytes(res.encode(encoding='utf-8'))).digest() |
|
|
result = "SHA-256=" + base64.b64encode(m).decode(encoding='utf-8') |
|
|
return result |
|
|
|
|
|
def httpdate(dt): |
|
|
""" |
|
|
Return a string representation of a date according to RFC 1123 |
|
|
(HTTP/1.1). |
|
|
|
|
|
The supplied date must be in UTC. |
|
|
|
|
|
""" |
|
|
weekday = ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"][dt.weekday()] |
|
|
month = ["Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", |
|
|
"Oct", "Nov", "Dec"][dt.month - 1] |
|
|
return "%s, %02d %s %04d %02d:%02d:%02d GMT" % (weekday, dt.day, month, |
|
|
dt.year, dt.hour, dt.minute, dt.second) |
|
|
|
|
|
def translate(Text, From, To, APPID, Secret, APIKey, Host="itrans.xfyun.cn"): |
|
|
RequestUri = "/v2/its" |
|
|
url="https://"+Host+RequestUri |
|
|
HttpMethod = "POST" |
|
|
Algorithm = "hmac-sha256" |
|
|
HttpProto = "HTTP/1.1" |
|
|
|
|
|
|
|
|
curTime_utc = datetime.datetime.utcnow() |
|
|
Date = httpdate(curTime_utc) |
|
|
|
|
|
|
|
|
BusinessArgs={ |
|
|
"from":From, |
|
|
"to": To, |
|
|
} |
|
|
|
|
|
content = str(base64.b64encode(Text.encode('utf-8')), 'utf-8') |
|
|
postdata = { |
|
|
"common": {"app_id": APPID}, |
|
|
"business": BusinessArgs, |
|
|
"data": { |
|
|
"text": content, |
|
|
} |
|
|
} |
|
|
body = json.dumps(postdata) |
|
|
|
|
|
digest = hashlib_256(body) |
|
|
signatureStr = "host: " + Host + "\n" |
|
|
signatureStr += "date: " + Date + "\n" |
|
|
signatureStr += HttpMethod + " " + RequestUri \ |
|
|
+ " " + HttpProto + "\n" |
|
|
signatureStr += "digest: " + digest |
|
|
signature = hmac.new(bytes(Secret.encode(encoding='utf-8')), |
|
|
bytes(signatureStr.encode(encoding='utf-8')), |
|
|
digestmod=hashlib.sha256).digest() |
|
|
sign = base64.b64encode(signature).decode(encoding='utf-8') |
|
|
|
|
|
authHeader = 'api_key="%s", algorithm="%s", ' \ |
|
|
'headers="host date request-line digest", ' \ |
|
|
'signature="%s"' \ |
|
|
% (APIKey, Algorithm, sign) |
|
|
headers = { |
|
|
"Content-Type": "application/json", |
|
|
"Accept": "application/json", |
|
|
"Method": "POST", |
|
|
"Host": Host, |
|
|
"Date": Date, |
|
|
"Digest": digest, |
|
|
"Authorization": authHeader |
|
|
} |
|
|
|
|
|
response = requests.post(url, data=body, headers=headers,timeout=60) |
|
|
return response |
|
|
|
