stable-diffusion-webui
31 строка · 992.0 Байт
1import os
2import importlib.util
3
4from modules import errors
5
6
7def load_module(path):
8module_spec = importlib.util.spec_from_file_location(os.path.basename(path), path)
9module = importlib.util.module_from_spec(module_spec)
10module_spec.loader.exec_module(module)
11
12return module
13
14
15def preload_extensions(extensions_dir, parser, extension_list=None):
16if not os.path.isdir(extensions_dir):
17return
18
19extensions = extension_list if extension_list is not None else os.listdir(extensions_dir)
20for dirname in sorted(extensions):
21preload_script = os.path.join(extensions_dir, dirname, "preload.py")
22if not os.path.isfile(preload_script):
23continue
24
25try:
26module = load_module(preload_script)
27if hasattr(module, 'preload'):
28module.preload(parser)
29
30except Exception:
31errors.report(f"Error running preload() for {preload_script}", exc_info=True)
32