from logging import log import os import sys import importlib.util import glob from .arikan import init, get_ext_dir NODE_CLASS_MAPPINGS = {} NODE_DISPLAY_NAME_MAPPINGS = {} WEB_DIRECTORY = "./js" 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) __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']