56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
|
# =========================================================
|
|||
|
|
# =============== 为插件提供简易的i18n翻译API ===============
|
|||
|
|
# =========================================================
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import csv
|
|||
|
|
|
|||
|
|
from umi_log import logger
|
|||
|
|
|
|||
|
|
LangCode = ""
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 设置语言
|
|||
|
|
def setLangCode(langCode):
|
|||
|
|
global LangCode
|
|||
|
|
LangCode = langCode
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Translator:
|
|||
|
|
# 载入翻译。每个模块调用tr前，必须载入翻译。
|
|||
|
|
# file: 与翻译文件同目录的任意文件的路径
|
|||
|
|
# name: 翻译文件名
|
|||
|
|
def __init__(self, file="", name=""):
|
|||
|
|
self._trDict = {}
|
|||
|
|
if not file or not name:
|
|||
|
|
return
|
|||
|
|
path = os.path.dirname(os.path.abspath(file)) + "/" + name
|
|||
|
|
try:
|
|||
|
|
if os.path.exists(path):
|
|||
|
|
with open(path, "r", encoding="utf-8") as file:
|
|||
|
|
csv_reader = csv.reader(file)
|
|||
|
|
headers = next(csv_reader)
|
|||
|
|
index = -1
|
|||
|
|
for i, h in enumerate(headers): # 搜索语言
|
|||
|
|
if h == LangCode:
|
|||
|
|
index = i
|
|||
|
|
if index == -1: # 找不到对应语言
|
|||
|
|
if LangCode.startswith("zh_"): # 中文
|
|||
|
|
index = 0
|
|||
|
|
else:
|
|||
|
|
index = 1 # 找不到对应语言，默认英文
|
|||
|
|
for row in csv_reader:
|
|||
|
|
self._trDict[row[0]] = row[index]
|
|||
|
|
except Exception:
|
|||
|
|
logger.error(
|
|||
|
|
f"加载插件翻译失败。 file: {file}, name: {name}",
|
|||
|
|
exc_info=True,
|
|||
|
|
stack_info=True,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 进行翻译。输入原文，返回译文
|
|||
|
|
def __call__(self, original):
|
|||
|
|
if original in self._trDict:
|
|||
|
|
return self._trDict[original]
|
|||
|
|
return original
|