Unverified Commit 0c1c9e74 authored by liubo0902's avatar liubo0902 Committed by GitHub
Browse files

Update localization.py

parent 5ef669de
Loading
Loading
Loading
Loading
+16 −10
Original line number Diff line number Diff line
import json
import os

from modules import errors, scripts
from modules import errors

localizations = {}

@@ -14,20 +14,26 @@ def list_localizations(dirname):
        if ext.lower() != ".json":
            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"):
        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:
    fn = localizations.get(current_localization_name, None)
    fns = localizations.get(current_localization_name, None)
    data = {}
    if fn is not None:
    if fns is not None:
        for fn in fns:
            try:
                with open(fn, "r", encoding="utf8") as file:
                data = json.load(file)
                    data.update(json.load(file))
            except Exception:
                errors.report(f"Error loading localization from {fn}", exc_info=True)