50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
# ABOUTME: Main entry point for ComfyUI Stillfront custom nodes
|
|
# ABOUTME: Dynamically loads all node modules from the py/ directory
|
|
|
|
import importlib.util
|
|
import os
|
|
import sys
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
|
|
def get_ext_dir(subpath=None, mkdir=False):
|
|
"""Get the extension directory path, optionally with a subpath."""
|
|
dir = os.path.dirname(__file__)
|
|
if subpath is not None:
|
|
dir = os.path.join(dir, subpath)
|
|
|
|
dir = os.path.abspath(dir)
|
|
|
|
if mkdir and not os.path.exists(dir):
|
|
os.makedirs(dir)
|
|
return dir
|
|
|
|
|
|
py = get_ext_dir("py")
|
|
files = os.listdir(py)
|
|
|
|
for file in files:
|
|
if not file.endswith(".py"):
|
|
continue
|
|
name = os.path.splitext(file)[0]
|
|
try:
|
|
imported_module = importlib.import_module(".py.{}".format(name), __name__)
|
|
if hasattr(imported_module, "NODE_CLASS_MAPPINGS"):
|
|
NODE_CLASS_MAPPINGS = {
|
|
**NODE_CLASS_MAPPINGS,
|
|
**imported_module.NODE_CLASS_MAPPINGS,
|
|
}
|
|
if hasattr(imported_module, "NODE_DISPLAY_NAME_MAPPINGS"):
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**NODE_DISPLAY_NAME_MAPPINGS,
|
|
**imported_module.NODE_DISPLAY_NAME_MAPPINGS,
|
|
}
|
|
except Exception as e:
|
|
print(f"[comfyui-stillfront] Failed to load {file}: {e}")
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|