Fooocus/modules/localization.py
lllyasviel adf134d446 100% cover all texts
100% cover all texts
2023-10-20 17:28:33 -07:00

60 lines
1.9 KiB
Python

import json
import os
localization_root = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'language')
def localization_js(filename):
data = {}
if isinstance(filename, str):
full_name = os.path.abspath(os.path.join(localization_root, filename + '.json'))
if os.path.exists(full_name):
try:
with open(full_name, encoding='utf-8') as f:
data = json.load(f)
assert isinstance(data, dict)
for k, v in data.items():
assert isinstance(k, str)
assert isinstance(v, str)
except Exception as e:
print(str(e))
print(f'Failed to load localization file {full_name}')
# data = {k: 'XXX' for k in data.keys()} # use this to see if all texts are covered
return f"window.localization = {json.dumps(data)}"
def dump_english_config(components):
all_texts = []
for c in components:
label = getattr(c, 'label', None)
value = getattr(c, 'value', None)
choices = getattr(c, 'choices', None)
info = getattr(c, 'info', None)
if isinstance(label, str):
all_texts.append(label)
if isinstance(value, str):
all_texts.append(value)
if isinstance(info, str):
all_texts.append(info)
if isinstance(choices, list):
for x in choices:
if isinstance(x, str):
all_texts.append(x)
if isinstance(x, tuple):
for y in x:
if isinstance(y, str):
all_texts.append(y)
config_dict = {k: k for k in all_texts if k != "" and 'progress-container' not in k}
full_name = os.path.abspath(os.path.join(localization_root, 'en.json'))
with open(full_name, "w", encoding="utf-8") as json_file:
json.dump(config_dict, json_file, indent=4)
return