Yusarina
93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
import os
|
|
import json
|
|
import bpy
|
|
from bpy.app.translations import locale
|
|
from ..core.register import register_wrap
|
|
from typing import Dict, List, Tuple
|
|
|
|
main_dir: str = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
resources_dir: str = os.path.join(main_dir, "resources")
|
|
translations_dir: str = os.path.join(resources_dir, "translations")
|
|
|
|
dictionary: Dict[str, str] = dict()
|
|
languages: List[str] = []
|
|
verbose: bool = True
|
|
|
|
def load_translations() -> None:
|
|
global dictionary, languages
|
|
|
|
dictionary = dict()
|
|
languages = ["auto"]
|
|
|
|
# Populate languages list
|
|
for i in os.listdir(translations_dir):
|
|
lang = i.split(".")[0]
|
|
if lang != "auto":
|
|
languages.append(lang)
|
|
|
|
# Check if the context and scene are available
|
|
if hasattr(bpy.context, "scene"):
|
|
# Check if the property exists before trying to access it
|
|
if hasattr(bpy.context.scene, "avatar_toolkit_language"):
|
|
language_index = bpy.context.scene.avatar_toolkit_language
|
|
if isinstance(language_index, str):
|
|
language_index = int(language_index)
|
|
if language_index == 0: # "auto"
|
|
language = bpy.context.preferences.view.language
|
|
else:
|
|
language = languages[language_index]
|
|
else:
|
|
language = bpy.context.preferences.view.language
|
|
else:
|
|
# Set a default language if the context or scene is not available
|
|
language = "en_US"
|
|
|
|
translation_file: str = os.path.join(translations_dir, language + ".json")
|
|
if os.path.exists(translation_file):
|
|
with open(translation_file, 'r', encoding='utf-8') as file:
|
|
dictionary = json.load(file)["messages"]
|
|
else:
|
|
custom_language: str = language.split("_")[0]
|
|
custom_translation_file: str = os.path.join(translations_dir, custom_language + ".json")
|
|
if os.path.exists(custom_translation_file):
|
|
with open(custom_translation_file, 'r', encoding='utf-8') as file:
|
|
dictionary = json.load(file)["messages"]
|
|
else:
|
|
print(f"Translation file not found for language: {language}")
|
|
default_file: str = os.path.join(translations_dir, "en_US.json")
|
|
if os.path.exists(default_file):
|
|
with open(default_file, 'r', encoding='utf-8') as file:
|
|
dictionary = json.load(file)["messages"]
|
|
else:
|
|
print("Default translation file 'en_US.json' not found.")
|
|
|
|
def t(phrase: str, *args, **kwargs) -> str:
|
|
output: str = dictionary.get(phrase)
|
|
if output is None:
|
|
if verbose:
|
|
print('Warning: Unknown phrase: ' + phrase)
|
|
return phrase
|
|
return output.format(*args, **kwargs)
|
|
|
|
def get_languages_list(self, context) -> List[Tuple[str, str, str]]:
|
|
return [(str(i), lang, f"Use {lang} language") for i, lang in enumerate(languages)]
|
|
|
|
def refresh_translations():
|
|
load_translations()
|
|
# Force a full UI update
|
|
for window in bpy.context.window_manager.windows:
|
|
for area in window.screen.areas:
|
|
area.tag_redraw()
|
|
|
|
def update_ui(self, context):
|
|
refresh_translations()
|
|
# Force Blender to redraw all UI elements
|
|
for screen in bpy.data.screens:
|
|
for area in screen.areas:
|
|
area.tag_redraw()
|
|
# Update the Scene to trigger a full UI refresh
|
|
bpy.context.scene.update_tag()
|
|
|
|
# Initial load of translations
|
|
load_translations()
|