mirror of https://github.com/jumpserver/jumpserver
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
|
import json
|
||
|
import os
|
||
|
|
||
|
from .base import BaseTranslateManager
|
||
|
from .const import RED, GREEN
|
||
|
|
||
|
|
||
|
class OtherTranslateManager(BaseTranslateManager):
|
||
|
|
||
|
@staticmethod
|
||
|
def get_need_trans_dict(zh_dict, other_dict):
|
||
|
diff_keys = set(zh_dict.keys()) - set(other_dict.keys())
|
||
|
need_trans_dict = {k: zh_dict[k] for k in diff_keys if k}
|
||
|
return need_trans_dict
|
||
|
|
||
|
def load_json_as_dict(self, file_prefix='zh'):
|
||
|
file_path = os.path.join(self._dir, f'{file_prefix}.json')
|
||
|
if not os.path.exists(file_path):
|
||
|
return {}
|
||
|
try:
|
||
|
with open(file_path, 'r', encoding='utf-8') as f:
|
||
|
return json.load(f)
|
||
|
except Exception as e:
|
||
|
print(f'{RED}File: {file_path} load error: {e}{RED}')
|
||
|
return {}
|
||
|
|
||
|
def save_dict_as_json(self, data, file_prefix='ja'):
|
||
|
file_path = os.path.join(self._dir, f'{file_prefix}.json')
|
||
|
try:
|
||
|
with open(file_path, 'w', encoding='utf-8') as f:
|
||
|
json.dump(data, f, ensure_ascii=False, sort_keys=True, indent=4)
|
||
|
except Exception as e:
|
||
|
print(f'{RED}File: {file_path} save error: {e}{RED}')
|
||
|
|
||
|
async def run(self):
|
||
|
zh_dict = self.load_json_as_dict()
|
||
|
|
||
|
for file_prefix, target_lang in self.LANG_MAPPER.items():
|
||
|
other_dict = self.load_json_as_dict(file_prefix)
|
||
|
need_trans_dict = self.get_need_trans_dict(zh_dict, other_dict)
|
||
|
print(f'{GREEN}Translate: {self.dir_name} {file_prefix} need to translate '
|
||
|
f'{len(need_trans_dict)}{GREEN}\n')
|
||
|
if not need_trans_dict:
|
||
|
continue
|
||
|
translated_dict = await self.bulk_translate(need_trans_dict, target_lang)
|
||
|
other_dict.update(translated_dict)
|
||
|
self.save_dict_as_json(other_dict, file_prefix)
|