Commit 0c1c9e74 authored by liubo0902's avatar liubo0902 Committed by GitHub

Update localization.py

parent 5ef669de
import json import json
import os import os
from modules import errors, scripts from modules import errors
localizations = {} localizations = {}
...@@ -14,21 +14,27 @@ def list_localizations(dirname): ...@@ -14,21 +14,27 @@ def list_localizations(dirname):
if ext.lower() != ".json": if ext.lower() != ".json":
continue continue
localizations[fn] = os.path.join(dirname, file) fn = fn.replace(" ", "").replace("(", "_").replace(")","")
localizations[fn] = [os.path.join(dirname, file)]
from modules import scripts
for file in scripts.list_scripts("localizations", ".json"): for file in scripts.list_scripts("localizations", ".json"):
fn, ext = os.path.splitext(file.filename) fn, ext = os.path.splitext(file.filename)
localizations[fn] = file.path fn = fn.replace(" ", "").replace("(", "_").replace(")","")
if fn not in localizations:
localizations[fn] = []
localizations[fn].append(file.path)
def localization_js(current_localization_name: str) -> str: def localization_js(current_localization_name: str) -> str:
fn = localizations.get(current_localization_name, None) fns = localizations.get(current_localization_name, None)
data = {} data = {}
if fn is not None: if fns is not None:
try: for fn in fns:
with open(fn, "r", encoding="utf8") as file: try:
data = json.load(file) with open(fn, "r", encoding="utf8") as file:
except Exception: data.update(json.load(file))
errors.report(f"Error loading localization from {fn}", exc_info=True) except Exception:
errors.report(f"Error loading localization from {fn}", exc_info=True)
return f"window.localization = {json.dumps(data)}" return f"window.localization = {json.dumps(data)}"
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment