stable-diffusion-webui

1
import json
2
import os
3

4
from modules import errors, scripts
5

6
localizations = {}
7

8

9
def list_localizations(dirname):
10
    localizations.clear()
11

12
    for file in os.listdir(dirname):
13
        fn, ext = os.path.splitext(file)
14
        if ext.lower() != ".json":
15
            continue
16

17
        localizations[fn] = [os.path.join(dirname, file)]
18

19
    for file in scripts.list_scripts("localizations", ".json"):
20
        fn, ext = os.path.splitext(file.filename)
21
        if fn not in localizations:
22
            localizations[fn] = []
23
        localizations[fn].append(file.path)
24

25

26
def localization_js(current_localization_name: str) -> str:
27
    fns = localizations.get(current_localization_name, None)
28
    data = {}
29
    if fns is not None:
30
        for fn in fns:
31
            try:
32
                with open(fn, "r", encoding="utf8") as file:
33
                    data.update(json.load(file))
34
            except Exception:
35
                errors.report(f"Error loading localization from {fn}", exc_info=True)
36

37
    return f"window.localization = {json.dumps(data)}"
38