58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
import os
|
|
import json
|
|
|
|
class I18n:
|
|
_instance = None
|
|
_translations = {}
|
|
_current_language = 'zh_CN' # 默认语言
|
|
|
|
@classmethod
|
|
def get_app_language(cls):
|
|
"""获取 ComfyUI 的当前语言设置"""
|
|
try:
|
|
# 尝试从 ComfyUI 配置文件获取语言设置
|
|
config_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), '..', '..', 'config.json')
|
|
if os.path.exists(config_path):
|
|
with open(config_path, 'r', encoding='utf-8') as f:
|
|
config = json.load(f)
|
|
return config.get('language', 'en_US')
|
|
except:
|
|
pass
|
|
return 'en_US'
|
|
|
|
@classmethod
|
|
def get_text(cls, key, default_text="", app_language=False):
|
|
"""
|
|
获取翻译文本
|
|
:param key: 翻译键
|
|
:param default_text: 默认文本
|
|
:param app_language: 是否使用 ComfyUI 的语言设置
|
|
:return: 翻译后的文本
|
|
"""
|
|
if app_language:
|
|
language = cls.get_app_language()
|
|
else:
|
|
language = cls._current_language
|
|
|
|
# 如果语言不存在，加载对应的翻译文件
|
|
if language not in cls._translations:
|
|
cls.load_translations(language)
|
|
|
|
# 获取翻译
|
|
translations = cls._translations.get(language, {})
|
|
return translations.get(key, default_text)
|
|
|
|
@classmethod
|
|
def load_translations(cls, language):
|
|
"""加载指定语言的翻译文件"""
|
|
try:
|
|
translation_file = os.path.join(
|
|
os.path.dirname(__file__),
|
|
'translations',
|
|
f'{language}.json'
|
|
)
|
|
if os.path.exists(translation_file):
|
|
with open(translation_file, 'r', encoding='utf-8') as f:
|
|
cls._translations[language] = json.load(f)
|
|
except:
|
|
cls._translations[language] = {} |