stable-diffusion-webui/modules/localization.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

38 lines
1.1 KiB
Python
Raw Normal View History

2022-10-17 12:15:32 -06:00
import json
import os
2023-09-05 01:08:59 -06:00
from modules import errors, scripts
2022-10-17 12:15:32 -06:00
localizations = {}
def list_localizations(dirname):
localizations.clear()
for file in os.listdir(dirname):
fn, ext = os.path.splitext(file)
if ext.lower() != ".json":
continue
2023-09-05 01:06:47 -06:00
localizations[fn] = [os.path.join(dirname, file)]
2022-10-17 12:15:32 -06:00
for file in scripts.list_scripts("localizations", ".json"):
fn, ext = os.path.splitext(file.filename)
2023-09-05 01:06:47 -06:00
if fn not in localizations:
localizations[fn] = []
localizations[fn].append(file.path)
2022-10-17 12:15:32 -06:00
def localization_js(current_localization_name: str) -> str:
2023-09-05 01:06:47 -06:00
fns = localizations.get(current_localization_name, None)
2022-10-17 12:15:32 -06:00
data = {}
2023-09-05 01:06:47 -06:00
if fns is not None:
for fn in fns:
try:
with open(fn, "r", encoding="utf8") as file:
data.update(json.load(file))
except Exception:
errors.report(f"Error loading localization from {fn}", exc_info=True)
2022-10-17 12:15:32 -06:00
return f"window.localization = {json.dumps(data)}"