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
|