WebUI/modules/script_loading.py

import os
import importlib.util

from modules import errors


def load_module(path):
    module_spec = importlib.util.spec_from_file_location(os.path.basename(path), path)
    module = importlib.util.module_from_spec(module_spec)
    module_spec.loader.exec_module(module)

    return module


def preload_extensions(extensions_dir, parser):
    if not os.path.isdir(extensions_dir):
        return

    for dirname in sorted(os.listdir(extensions_dir)):
        preload_script = os.path.join(extensions_dir, dirname, "preload.py")
        if not os.path.isfile(preload_script):
            continue

        try:
            module = load_module(preload_script)
            if hasattr(module, 'preload'):
                module.preload(parser)

        except Exception:
            errors.report(f"Error running preload() for {preload_script}", exc_info=True)
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py 2022-11-12 07:56:06 +00:00			`import os`
use python importlib to load and execute extension modules previously module attributes like __file__ where not set correctly, leading to scripts getting the directory of the stable-diffusion repo location instead of their own script. This causes problem when loading user data from an external location using the --data-dir flag, as extensions would look for their own code in the stable-diffusion repo location instead of the data dir location. Using pythons importlib functions sets the modules specs correctly and executes them. But this will break extensions if they build paths based on the previously incorrect __file__ attribute. 2023-01-25 16:55:59 +00:00			`import importlib.util`
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py 2022-11-12 07:56:06 +00:00
rename print_error to report, use it with together with package name 2023-05-31 16:56:37 +00:00			`from modules import errors`
Add & use modules.errors.print_error where currently printing exception info by hand 2023-05-29 05:54:13 +00:00
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py 2022-11-12 07:56:06 +00:00
			`def load_module(path):`
use python importlib to load and execute extension modules previously module attributes like __file__ where not set correctly, leading to scripts getting the directory of the stable-diffusion repo location instead of their own script. This causes problem when loading user data from an external location using the --data-dir flag, as extensions would look for their own code in the stable-diffusion repo location instead of the data dir location. Using pythons importlib functions sets the modules specs correctly and executes them. But this will break extensions if they build paths based on the previously incorrect __file__ attribute. 2023-01-25 16:55:59 +00:00			`module_spec = importlib.util.spec_from_file_location(os.path.basename(path), path)`
			`module = importlib.util.module_from_spec(module_spec)`
			`module_spec.loader.exec_module(module)`
make existing script loading and new preload code use same code for loading modules limit extension preload scripts to just one file named preload.py 2022-11-12 07:56:06 +00:00
			`return module`


			`def preload_extensions(extensions_dir, parser):`
			`if not os.path.isdir(extensions_dir):`
			`return`

			`for dirname in sorted(os.listdir(extensions_dir)):`
			`preload_script = os.path.join(extensions_dir, dirname, "preload.py")`
			`if not os.path.isfile(preload_script):`
			`continue`

			`try:`
			`module = load_module(preload_script)`
			`if hasattr(module, 'preload'):`
			`module.preload(parser)`

			`except Exception:`
rename print_error to report, use it with together with package name 2023-05-31 16:56:37 +00:00			`errors.report(f"Error running preload() for {preload_script}", exc_info=True)`