| import importlib.util | |
| import glob | |
| import os | |
| import sys | |
| from .pysssss import init, get_ext_dir | |
| NODE_CLASS_MAPPINGS = {} | |
| NODE_DISPLAY_NAME_MAPPINGS = {} | |
| if init(): | |
| py = get_ext_dir("py") | |
| files = glob.glob(os.path.join(py, "*.py"), recursive=False) | |
| for file in files: | |
| name = os.path.splitext(file)[0] | |
| spec = importlib.util.spec_from_file_location(name, file) | |
| module = importlib.util.module_from_spec(spec) | |
| sys.modules[name] = module | |
| spec.loader.exec_module(module) | |
| if hasattr(module, "NODE_CLASS_MAPPINGS") and getattr(module, "NODE_CLASS_MAPPINGS") is not None: | |
| NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS) | |
| if hasattr(module, "NODE_DISPLAY_NAME_MAPPINGS") and getattr(module, "NODE_DISPLAY_NAME_MAPPINGS") is not None: | |
| NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS) | |
| WEB_DIRECTORY = "./web" | |
| __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"] | |