OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-01 03:03:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix to make scripts that failed to compile not crash the program

Browse Source
This commit is contained in:
AUTOMATIC 2022-09-03 19:59:11 +03:00
parent 226de5e503
commit e67a56dbb4
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
modules/scripts.py
View File

@ -41,17 +41,21 @@ def load_scripts(basedir):
with open(path, "r", encoding="utf8") as file: with open(path, "r", encoding="utf8") as file:
text = file.read() text = file.read()
from types import ModuleType try:
compiled = compile(text, path, 'exec') from types import ModuleType
module = ModuleType(filename) compiled = compile(text, path, 'exec')
exec(compiled, module.__dict__) module = ModuleType(filename)
exec(compiled, module.__dict__)
for key, script_class in module.__dict__.items(): for key, script_class in module.__dict__.items():
if type(script_class) == type and issubclass(script_class, Script): if type(script_class) == type and issubclass(script_class, Script):
obj = script_class() obj = script_class()
obj.filename = path obj.filename = path
scripts.append(obj) scripts.append(obj)
except Exception:
print(f"Error loading script: {filename}", file=sys.stderr)
print(traceback.format_exc(), file=sys.stderr)
def wrap_call(func, filename, funcname, *args, default=None, **kwargs): def wrap_call(func, filename, funcname, *args, default=None, **kwargs):