224 lines
7.8 KiB
Python
224 lines
7.8 KiB
Python
# -------------------------------
|
||
# Difficult Rocket
|
||
# Copyright © 2021-2022 by shenjackyuanjie 3695888@qq.com
|
||
# All rights reserved
|
||
# -------------------------------
|
||
|
||
"""
|
||
writen by shenjackyuanjie <3695888@qq.com>
|
||
mail: 3695888@qq.com
|
||
github: @shenjackyuanjie
|
||
gitee: @shenjackyuanjie
|
||
"""
|
||
|
||
import inspect
|
||
|
||
from dataclasses import dataclass
|
||
from typing import Union, Tuple, Any, Type, List, Dict, Hashable, Optional
|
||
|
||
from Difficult_Rocket import DR_runtime, DR_option
|
||
from Difficult_Rocket.utils import tools
|
||
from Difficult_Rocket.exception.language import *
|
||
|
||
|
||
@dataclass
|
||
class TranslateConfig:
|
||
raise_error: bool = False
|
||
# 引用错误时抛出错误
|
||
crack_normal: bool = False
|
||
# 出现错误引用后 将引用到的正确内容替换为引用路径
|
||
insert_crack: bool = True
|
||
# 加入引用的错误内容
|
||
is_final: bool = False
|
||
# 是否为最终内容
|
||
keep_get: bool = True
|
||
# 引用错误后是否继续引用
|
||
|
||
def set(self, item: str, value: bool) -> 'TranslateConfig':
|
||
assert getattr(self, item, None) is not None, f'Config {item} is not in TranslateConfig'
|
||
assert type(value) is bool
|
||
setattr(self, item, value)
|
||
return self
|
||
|
||
def __copy__(self) -> 'TranslateConfig':
|
||
return TranslateConfig(raise_error=self.raise_error,
|
||
crack_normal=self.crack_normal,
|
||
insert_crack=self.insert_crack,
|
||
is_final=self.is_final,
|
||
keep_get=self.keep_get)
|
||
|
||
def copy(self) -> 'TranslateConfig':
|
||
return self.__copy__()
|
||
|
||
|
||
class Translates:
|
||
def __init__(self,
|
||
value: Union[Dict[str, Any], list, tuple, str],
|
||
config: Optional[TranslateConfig] = None,
|
||
get_list: List[Tuple[int, str]] = None,
|
||
error_get_list: List[Tuple[int, str]] = None):
|
||
"""
|
||
一个用于翻译的东西
|
||
:param value: 翻译键节点
|
||
:param config: 配置
|
||
:param get_list: 获取列表
|
||
:param error_get_list:
|
||
"""
|
||
self.value: Union[Dict[str, Any], list, tuple] = value
|
||
self.config = config or TranslateConfig()
|
||
self.get_list = get_list or []
|
||
self.error_get_list = error_get_list or []
|
||
|
||
def set_option(self, option: Union[str, TranslateConfig],
|
||
value: Optional[Union[bool, List[str]]] = None) -> 'Translates':
|
||
assert type(option) is str or isinstance(option, TranslateConfig)
|
||
if isinstance(option, TranslateConfig):
|
||
self.config = option
|
||
return self
|
||
self.config.set(option, value)
|
||
|
||
def __getitem__(self, item: Union[str, int, Hashable]) -> Union["Translates", str, int]:
|
||
"""
|
||
一坨答辩
|
||
:param item: 取用的内容/小天才
|
||
:return:
|
||
"""
|
||
cache_get_list = self.get_list.copy()
|
||
cache_error_get_list = self.error_get_list.copy()
|
||
cache_get_list.append(item)
|
||
try:
|
||
if not self.final:
|
||
if item not in self.value and (type(item) is int and len(self.value) < item):
|
||
raise KeyError
|
||
cache = self.value[item]
|
||
cache_get_list.append(item)
|
||
except (KeyError, TypeError):
|
||
# 出现问题
|
||
if DR_option.report_translate_no_found:
|
||
frame = inspect.currentframe()
|
||
last_frame = frame.f_back
|
||
if last_frame.f_code == self.__getattr__.__code__:
|
||
last_frame = last_frame.f_back
|
||
call_info = f'Translate Not Found at {last_frame.f_code.co_name} by {".".join(cache_get_list)} at:' \
|
||
f'{last_frame.f_code.co_filename}:{last_frame.f_lineno}'
|
||
print(call_info)
|
||
# 如果不抛出错误
|
||
if self.config.raise_error:
|
||
raise TranslateKeyNotFound(item_names=cache_get_list) from None
|
||
cache_error_get_list.append(item)
|
||
if self.final: # 如果已经是翻译结果
|
||
return Translates(value='.'.join(cache_get_list), error_get_list=cache_error_get_list)
|
||
else:
|
||
if self.final:
|
||
return self
|
||
else:
|
||
return Translates(value=cache, get_list=cache_get_list)
|
||
|
||
def __getattr__(self, item: Union[str, Hashable]) -> Union["Translates"]:
|
||
if hasattr(object, item):
|
||
return getattr(object, item)
|
||
return self.__getitem__(item)
|
||
|
||
def __str__(self):
|
||
if self.final: # 如果是字符串
|
||
return f'{self.value}.{".".join(self.get_list)}'
|
||
return str(self.value)
|
||
|
||
|
||
class Tr:
|
||
"""
|
||
我不装了，我就抄了tr
|
||
GOOD
|
||
"""
|
||
|
||
def __init__(self, language: str = None, config: Optional[TranslateConfig] = None):
|
||
"""
|
||
诶嘿，我抄的MCDR
|
||
:param language: Tr 所使用的的语言
|
||
:param config: 配置
|
||
"""
|
||
self.language_name = language or DR_runtime.language
|
||
self.translates: Dict = tools.load_file(f'configs/lang/{self.language_name}.toml')
|
||
self.default_translate: Dict = tools.load_file(f'configs/lang/{DR_runtime.default_language}.toml')
|
||
self.default_config = config or TranslateConfig()
|
||
self.translates_cache = Translates(value=self.translates, config=TranslateConfig().copy())
|
||
|
||
# def __call__(self, ):
|
||
# ...
|
||
|
||
def __getitem__(self, item):
|
||
if item in self.translates:
|
||
return getattr(self.translates_cache, item)
|
||
|
||
|
||
class Lang:
|
||
"""
|
||
用于创建一个对应语言的翻译类
|
||
感谢Fallen的MCDR提供idea
|
||
https://github.com/Fallen-Breath/MCDReforged
|
||
可以用
|
||
lang['language'] = 'abc' 或
|
||
lang['lang'] = 'abc'
|
||
的方式直接更改并刷新翻译
|
||
用
|
||
lang.lang(xxx, xxx)来获取翻译过的值
|
||
"""
|
||
|
||
def __init__(self) -> None:
|
||
self.translates = tools.load_file(f'configs/lang/{DR_runtime.language}.toml')
|
||
self.default_translates = tools.load_file('configs/lang/zh-CN.toml')
|
||
self.直接返回原始数据 = True
|
||
|
||
def __str__(self) -> str:
|
||
return DR_option.language
|
||
|
||
def __getitem__(self, item) -> Union[int, str, list, dict]:
|
||
try:
|
||
return self.translates[item]
|
||
except KeyError:
|
||
try:
|
||
return self.default_translates[item]
|
||
except KeyError:
|
||
raise TranslateKeyNotFound
|
||
# raise TranslateKeyNotFound(f'there\'s no key {item} in both {DR_option.language} and zh-CN')
|
||
|
||
def lang(self, *args) -> Union[int, str, list, dict, tuple]:
|
||
# frame = inspect.currentframe()
|
||
# # print("调用当前log的文件名:", frame.f_back.f_code.co_filename)
|
||
# objprint.objprint(frame.f_back.f_code,
|
||
# honor_existing=False,
|
||
# depth=2)
|
||
try:
|
||
result = self.translates
|
||
for option in args:
|
||
result = result[option]
|
||
return result
|
||
except KeyError:
|
||
try:
|
||
result = self.default_translates
|
||
for option in args:
|
||
result = result[option]
|
||
return result
|
||
except KeyError as e:
|
||
if self.直接返回原始数据:
|
||
return args
|
||
raise TranslateKeyNotFound from e
|
||
# raise TranslateKeyNotFound(f'there\'s no key {args} in both {DR_option.language} and zh-CN') from e
|
||
|
||
def 翻译(self, *args) -> Union[int, str, list, dict]:
|
||
return self.lang(args)
|
||
|
||
def _update_lang(self) -> str:
|
||
"""
|
||
用于更新语言(内部调用)
|
||
:return: 设置完成后的语言
|
||
"""
|
||
self.translates = tools.load_file(f'configs/lang/{DR_option.language}.toml')
|
||
return DR_option.language
|
||
|
||
|
||
if not __name__ == '__main__':
|
||
tr = Lang()
|
||
else:
|
||
tr_ = Tr()
|