Commit 37a23011 authored by space-nuko's avatar space-nuko
Browse files

Expose the compiled class module of scripts to extensions

parent 9cfd10cd
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -152,7 +152,7 @@ def basedir():

scripts_data = []
ScriptFile = namedtuple("ScriptFile", ["basedir", "filename", "path"])
ScriptClassData = namedtuple("ScriptClassData", ["script_class", "path", "basedir"])
ScriptClassData = namedtuple("ScriptClassData", ["script_class", "path", "basedir", "module"])


def list_scripts(scriptdirname, extension):
@@ -206,7 +206,7 @@ def load_scripts():

            for key, script_class in module.__dict__.items():
                if type(script_class) == type and issubclass(script_class, Script):
                    scripts_data.append(ScriptClassData(script_class, scriptfile.path, scriptfile.basedir))
                    scripts_data.append(ScriptClassData(script_class, scriptfile.path, scriptfile.basedir, module))

        except Exception:
            print(f"Error loading script: {scriptfile.filename}", file=sys.stderr)
@@ -241,7 +241,7 @@ class ScriptRunner:
        self.alwayson_scripts.clear()
        self.selectable_scripts.clear()

        for script_class, path, basedir in scripts_data:
        for script_class, path, basedir, script_module in scripts_data:
            script = script_class()
            script.filename = path
            script.is_txt2img = not is_img2img