319 lines
9.7 KiB
Python
319 lines
9.7 KiB
Python
#coding: utf-8
|
||
|
||
import os
|
||
import sys
|
||
import json
|
||
import re
|
||
import time
|
||
import random
|
||
import json
|
||
from hashlib import md5
|
||
|
||
try:
|
||
from googletrans import Translator
|
||
except ImportError:
|
||
print("未安装 googletrans==4.0.0-rc1，正在安装...")
|
||
os.system('btpip install googletrans==4.0.0-rc1')
|
||
from googletrans import Translator
|
||
|
||
try:
|
||
import requests
|
||
except ImportError:
|
||
print("未安装 requests，正在安装...")
|
||
os.system('btpip install requests')
|
||
import requests
|
||
|
||
|
||
|
||
# 解决 Windows 下编码问题，强制使用 utf-8 编码
|
||
if sys.platform == "win32":
|
||
sys.stdout.reconfigure(encoding='utf-8')
|
||
|
||
# 百度翻译
|
||
class baidu_translate:
|
||
endpoint = 'http://api.fanyi.baidu.com'
|
||
path = '/api/trans/vip/translate'
|
||
url = endpoint + path
|
||
appid = ''
|
||
appkey = ''
|
||
# todo
|
||
# from_lang = 'zh'
|
||
from_lang = 'en'
|
||
|
||
def make_md5(self,s, encoding='utf-8'):
|
||
'''
|
||
@name 生成md5
|
||
@param s 字符串
|
||
@param encoding 编码
|
||
@return string
|
||
'''
|
||
return md5(s.encode(encoding)).hexdigest()
|
||
|
||
|
||
def translate(self, query, to_lang='en'):
|
||
'''
|
||
@name 调用百度翻译API
|
||
@param query 原文
|
||
@param to_lang 目标语言
|
||
@return dict
|
||
'''
|
||
salt = random.randint(32768, 65536)
|
||
sign = self.make_md5(self.appid + query + str(salt) + self.appkey)
|
||
|
||
headers = {'Content-Type': 'application/x-www-form-urlencoded'}
|
||
|
||
payload = {'appid': self.appid, 'q': query, 'from': self.from_lang, 'to': to_lang, 'salt': salt, 'sign': sign}
|
||
|
||
r = requests.post(self.url, params=payload, headers=headers)
|
||
result = r.json()
|
||
return result
|
||
|
||
|
||
def google(self,text,dest,src):
|
||
'''
|
||
@name 调用googletrans API
|
||
@param text 原文
|
||
@param dest 目标语言
|
||
@param src 源语言
|
||
@return string
|
||
'''
|
||
print('原文:',text,'目标语言:',dest)
|
||
translater = Translator()
|
||
result = translater.translate(text, dest, src)
|
||
return result.text
|
||
|
||
|
||
|
||
class Lang:
|
||
client_extension = [".json",".js",".ts",".html",".css",".sh",".vue"] # 前端文件后缀
|
||
server_extension = [".py"] # 后端文件后缀
|
||
server_path = ['class', 'class_v2', 'plugin', 'mod'] # 后端代码 目录
|
||
# client_path = "../YakPanel"
|
||
exec_path = os.getcwd() # 语言包目录
|
||
panel_path = os.path.dirname(os.path.dirname(exec_path)) # 项目目录
|
||
|
||
|
||
|
||
def get_files(self, path, mark='client'):
|
||
'''
|
||
@name 获取指定目录下的文件列表
|
||
@param path 目录
|
||
@param mark 标记 区分前端与后端 前端 'client' 后端 'server'
|
||
@return list
|
||
'''
|
||
files = []
|
||
for root, dirs, file in os.walk(path):
|
||
for f in file:
|
||
if root.find('node_modules') != -1:
|
||
continue
|
||
if mark == 'client':
|
||
for ext in self.client_extension:
|
||
if f.endswith(ext):
|
||
files.append(os.path.join(root, f))
|
||
else:
|
||
for ext in self.server_extension:
|
||
if f.endswith(ext):
|
||
files.append(os.path.join(root, f))
|
||
return files
|
||
|
||
def read_file(self,filename):
|
||
'''
|
||
@name 读取文件
|
||
@param filename 文件名
|
||
@return any
|
||
'''
|
||
with open(filename, "r",encoding="utf-8") as file:
|
||
fbody = file.read()
|
||
return fbody
|
||
|
||
def write_file(self,filename, body):
|
||
'''
|
||
@name 写入文件
|
||
@param filename 文件名
|
||
@param body 内容
|
||
'''
|
||
with open(filename, "w",encoding="utf-8") as file:
|
||
file.write(body)
|
||
|
||
def get_lang(self, body):
|
||
'''
|
||
@name 解析文件中要翻译的文本
|
||
@param body 文件内容
|
||
@return list
|
||
'''
|
||
|
||
lang_list = []
|
||
# 匹配单引号 双引号
|
||
pattern = r"public\.lang\([\"'](.+?)[\"']"
|
||
for line in body.split("\n"):
|
||
if "public.lang(" in line:
|
||
# lang = re.findall(r"public\.lang\(\'(.+?)\'", line)
|
||
lang = re.findall(pattern, line)
|
||
|
||
if lang:
|
||
lang_list.extend(lang)
|
||
return lang_list
|
||
|
||
def get_all_lang_dict(self,to_lang='en'):
|
||
'''
|
||
@name 取已经翻译过的结果集
|
||
@param to_lang 目标语言
|
||
@return dict
|
||
'''
|
||
|
||
filename = os.path.join(self.exec_path,'all',to_lang + '.json')
|
||
if not os.path.exists(filename):
|
||
return {}
|
||
body = self.read_file(filename)
|
||
try:
|
||
return json.loads(body)
|
||
except:
|
||
return {}
|
||
|
||
def save_all_lang_dict(self,langs,to_lang='en'):
|
||
'''
|
||
@name 保存翻译结果集
|
||
@param langs dict
|
||
@param to_lang 目标语言
|
||
'''
|
||
save_path = os.path.join(self.exec_path,'all')
|
||
filename = os.path.join(save_path,to_lang + '.json')
|
||
if not os.path.exists(save_path):
|
||
os.makedirs(save_path)
|
||
|
||
self.write_file(filename,json.dumps(langs,indent=4,ensure_ascii=False))
|
||
|
||
# todo 改源语言为英文
|
||
def translate(self, langs,to_lang='en',name='server',to_lang_google='en'):
|
||
'''
|
||
@name 翻译
|
||
@param langs list
|
||
@param to_lang 目标语言
|
||
@param name 保存文件名
|
||
'''
|
||
baidu = baidu_translate()
|
||
to_lang_dict = {}
|
||
all_lang_dict = self.get_all_lang_dict(to_lang)
|
||
for lang in langs:
|
||
print('----- lang')
|
||
print(lang)
|
||
md5 = baidu.make_md5(lang)
|
||
# if to_lang == 'zh':
|
||
if to_lang == 'en':
|
||
to_lang_dict[md5] = lang
|
||
continue
|
||
|
||
# 先从已经翻译过的结果集中取
|
||
if md5 in all_lang_dict:
|
||
to_lang_dict[md5] = all_lang_dict[md5]
|
||
continue
|
||
|
||
# 调用googletrans API
|
||
trans_result = {}
|
||
try:
|
||
# trans_result['dst'] = baidu.google(lang,to_lang_google,'zh-cn')
|
||
trans_result['dst'] = baidu.google(lang,to_lang_google,'en')
|
||
except Exception as e:
|
||
print('Error:',e)
|
||
continue
|
||
|
||
to_lang_dict[md5] = trans_result['dst']
|
||
|
||
all_lang_dict[md5] = trans_result['dst']
|
||
time.sleep(0.1)
|
||
|
||
# 保存翻译结果集
|
||
self.save_all_lang_dict(all_lang_dict,to_lang)
|
||
|
||
# 保存翻译结果到语言文件
|
||
save_path = os.path.join(self.exec_path,to_lang)
|
||
if not os.path.exists(save_path):
|
||
os.makedirs(save_path)
|
||
filename = os.path.join(save_path,name+'.json')
|
||
|
||
self.write_file(filename,json.dumps(to_lang_dict,indent=4,ensure_ascii=False))
|
||
|
||
|
||
def get_settings(self):
|
||
'''
|
||
@name 获取配置文件
|
||
@return list
|
||
'''
|
||
default = [{
|
||
"name": "en",
|
||
"google": "en",
|
||
"title": "English",
|
||
"cn": "英语"
|
||
}]
|
||
filename = os.path.join(self.exec_path,'settings.json')
|
||
if not os.path.exists(filename):
|
||
return default
|
||
body = self.read_file(filename)
|
||
try:
|
||
return json.loads(body)['languages']
|
||
except:
|
||
return default
|
||
|
||
|
||
def start(self):
|
||
'''
|
||
@name 开始翻译
|
||
'''
|
||
|
||
# 获取需要翻译的文件列表(后端)
|
||
server_files = []
|
||
for directory in self.server_path:
|
||
directory_path = os.path.join(self.panel_path, directory)
|
||
files = self.get_files(directory_path, 'server')
|
||
server_files.extend(files)
|
||
print(files)
|
||
# return
|
||
# server_files = self.get_files(os.path.join(self.exec_path,self.server_path))
|
||
|
||
# 解析文件中需要翻译的文本
|
||
langs = []
|
||
for file in server_files:
|
||
body = self.read_file(file)
|
||
langs.extend(self.get_lang(body))
|
||
|
||
# 获取配置文件中的语言
|
||
config_langs = self.get_settings()
|
||
for lang in config_langs:
|
||
# 翻译
|
||
print('正在翻译:',lang['cn'],'=>',lang['name'],'=>',lang['title'],'...')
|
||
self.translate(langs,lang['name'],'server',lang['google'])
|
||
|
||
#-----------------------------------------------------------------------------------------
|
||
# # 获取需要翻译的文件列表（前端）
|
||
# client_files = self.get_files(self.client_path)
|
||
# # 解析文件中需要翻译的文本
|
||
# langs = []
|
||
# for file in client_files:
|
||
# print('解析文件:',file)
|
||
# body = self.read_file(file)
|
||
# langs.extend(self.get_lang(body))
|
||
|
||
# for lang in config_langs:
|
||
# # 翻译
|
||
# print('正在翻译:',lang['cn'],'=>',lang['name'],'=>',lang['title'],'...')
|
||
# self.translate(langs,lang['name'],'client',lang['google'])
|
||
|
||
print('翻译完成！')
|
||
|
||
|
||
|
||
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# PS: 在项目目录/YakPanel/languages/lang.py 中运行此脚本
|
||
# 需要先安装 googletrans==4.0.0-rc1 和 requests
|
||
# 设置代理（如果你已在本机开启了系统代理，可以注释下面两行代码，否则将其设置为你的HTTP代理地址）
|
||
os.environ["http_proxy"] = "http://192.168.168.162:7890"
|
||
os.environ["https_proxy"] = "http://192.168.168.162:7890"
|
||
|
||
# 开始翻译
|
||
lang = Lang()
|
||
lang.start()
|
||
|
||
|