Public-Mirror/LunaTranslator
1
0
Fork 0
mirror of https://github.com/HIllya51/LunaTranslator.git synced 2024-12-28 08:04:13 +08:00
Code Issues Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
恍兮惚兮 2024-10-04 16:34:28 +08:00
parent 3b52d28f5b
commit 05df805d71
2 changed files with 44 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
LunaTranslator/LunaTranslator/ocrengines/baiduocr_X.py
View File

@ -143,22 +143,34 @@ class OCR(baseocr):
}
def initocr(self):
self.keys = {}
self.access = {}
if self.config["接口"] != 5:
self.getaccess()
def get_access_token(self, API_KEY, SECRET_KEY):
url = "https://aip.baidubce.com/oauth/2.0/token"
params = {
"grant_type": "client_credentials",
"client_id": API_KEY,
"client_secret": SECRET_KEY,
}
js = self.proxysession.post(url, params=params).json()
try:
return js["access_token"]
except:
raise Exception(js)
def getaccess(self):
self.checkempty(["API Key", "Secret Key"])
pair = (self.config["API Key"], self.config["Secret Key"])
if pair not in self.keys:
accstoken = self.proxysession.get(
"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id="
+ self.appid
+ "&client_secret="
+ self.secretKey
).json()["access_token"]
self.keys[pair] = accstoken
return self.keys[pair]
SECRET_KEY, API_KEY = (
self.config["Secret Key"],
self.config["API Key"],
)
if not self.access.get((API_KEY, SECRET_KEY)):
acss = self.get_access_token(API_KEY, SECRET_KEY)
self.access[(API_KEY, SECRET_KEY)] = acss
return self.access[(API_KEY, SECRET_KEY)]
def ocr(self, imagebinary):
if self.config["接口"] in [0, 1, 2, 3]:

33
LunaTranslator/LunaTranslator/translator/baiduapi.py
View File

@ -6,7 +6,7 @@ import random
class TS(basetrans):
def inittranslator(self):
self.keys = {}
self.access = {}
if self.config["interface"] == 1:
self.getaccess()
@ -31,21 +31,30 @@ class TS(basetrans):
return self.translate_bce(query)
raise Exception("unknown")
def get_access_token(self, API_KEY, SECRET_KEY):
url = "https://aip.baidubce.com/oauth/2.0/token"
params = {
"grant_type": "client_credentials",
"client_id": API_KEY,
"client_secret": SECRET_KEY,
}
js = self.proxysession.post(url, params=params).json()
try:
return js["access_token"]
except:
raise Exception(js)
def getaccess(self):
self.checkempty(["API Key", "Secret Key"])
pair = (
self.multiapikeycurrent["API Key"],
SECRET_KEY, API_KEY = (
self.multiapikeycurrent["Secret Key"],
self.multiapikeycurrent["API Key"],
)
if pair not in self.keys:
accstoken = self.proxysession.get(
"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id="
+ self.appid
+ "&client_secret="
+ self.secretKey
).json()["access_token"]
self.keys[pair] = accstoken
return self.keys[pair]
if not self.access.get((API_KEY, SECRET_KEY)):
acss = self.get_access_token(API_KEY, SECRET_KEY)
self.access[(API_KEY, SECRET_KEY)] = acss
return self.access[(API_KEY, SECRET_KEY)]
def translate_bce(self, q):
accstoken = self.getaccess()