hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update localization.py

Browse Source
This commit is contained in:
liubo0902
2023-09-05 15:06:47 +08:00
committed by GitHub
parent 5ef669de08
commit 0c1c9e74cd
1 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
modules/localization.py
View File

@@ -1,7 +1,7 @@
import json import json
import os import os
from modules import errors, scripts from modules import errors
localizations = {} localizations = {}
@@ -14,20 +14,26 @@ 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:
for fn in fns:
try: try:
with open(fn, "r", encoding="utf8") as file: with open(fn, "r", encoding="utf8") as file:
data = json.load(file) data.update(json.load(file))
except Exception: except Exception:
errors.report(f"Error loading localization from {fn}", exc_info=True) errors.report(f"Error loading localization from {fn}", exc_info=True)