78 lines
3.5 KiB
Python
78 lines
3.5 KiB
Python
import volcenginesdkcore
|
||
import volcenginesdktranslate20250301
|
||
from volcenginesdkcore.rest import ApiException
|
||
from TranslateParent import TranslateParent
|
||
from concurrent.futures import ThreadPoolExecutor, as_completed
|
||
|
||
|
||
class VolcEngine(TranslateParent):
|
||
def __init__(self, ):
|
||
from main import server
|
||
self.access_id, self.access_key = server.tell_volc_api()
|
||
|
||
def _set_configure(self):
|
||
configuration = volcenginesdkcore.Configuration()
|
||
configuration.ak = self.access_id
|
||
configuration.sk = self.access_key
|
||
configuration.region = "cn-beijing"
|
||
volcenginesdkcore.Configuration.set_default(configuration)
|
||
|
||
@staticmethod
|
||
def _start_translate_threads(tag_list: list, target_language: str, api_instance) -> dict:
|
||
index = -1
|
||
if type(tag_list[0] == dict):
|
||
index = tag_list.pop(0)["index"]
|
||
translate_text_request = volcenginesdktranslate20250301.TranslateTextRequest(
|
||
source_language="en",
|
||
target_language=target_language,
|
||
text_list=tag_list,
|
||
)
|
||
try:
|
||
response = api_instance.translate_text(translate_text_request)
|
||
return {"thread_status": True, "thread_data": response.translation_list, "index": index}
|
||
except ApiException as e:
|
||
# 复制代码运行示例，请自行打印API错误信息。
|
||
print("Exception when calling api: %s\n" % e)
|
||
return {"thread_status": False}
|
||
|
||
def translate_list(self, tag_list: list, target_language: str = "zh") -> dict | None:
|
||
self._set_configure()
|
||
api_instance = volcenginesdktranslate20250301.TRANSLATE20250301Api()
|
||
list_max = 15
|
||
if len(tag_list) <= list_max:
|
||
thread_result_dict = self._start_translate_threads(tag_list, target_language, api_instance)
|
||
if not thread_result_dict["thread_status"]:
|
||
return {"status": False}
|
||
translate_list = []
|
||
thread_result = thread_result_dict["thread_data"]
|
||
for item in thread_result:
|
||
translate_list.append(item.translation)
|
||
return {"status": True, "translate_list": translate_list}
|
||
else:
|
||
threads_result = []
|
||
chunks = [tag_list[i:i + list_max] for i in range(0, len(tag_list), list_max)]
|
||
open_threads_count = len(chunks)
|
||
index = 0
|
||
for item in chunks:
|
||
item.insert(0, {"index": index})
|
||
index += 1
|
||
with ThreadPoolExecutor(max_workers=min(open_threads_count, 5)) as executor:
|
||
futures = [executor.submit(self._start_translate_threads, thread_tag_list,
|
||
target_language, api_instance) for thread_tag_list in chunks]
|
||
for future in as_completed(futures):
|
||
try:
|
||
if not future.result().get("thread_status"):
|
||
return {"status": False}
|
||
threads_result.append(future.result())
|
||
except Exception as e:
|
||
print(e)
|
||
return None
|
||
translate_list = []
|
||
for index in range(len(threads_result)):
|
||
for item in threads_result:
|
||
if item["index"] == index:
|
||
thread_result = item["thread_data"]
|
||
for translate_data in thread_result:
|
||
translate_list.append(translate_data.translation)
|
||
return {"status": True, "translate_list": translate_list}
|