Files
Merserk-ComfyUI-Flow-Assistor/__init__.py
T
2025-11-27 21:54:55 +02:00

29 lines
1.2 KiB
Python

# Import other nodes. Using try-except blocks to prevent crashes if
# sibling files are missing during testing/installation.
try:
from .camera_angle_control_node import NODE_CLASS_MAPPINGS as CAM_CLASSES, NODE_DISPLAY_NAME_MAPPINGS as CAM_DISPLAY
except ImportError:
CAM_CLASSES, CAM_DISPLAY = {}, {}
try:
from .prompt_queue_node import NODE_CLASS_MAPPINGS as QUEUE_CLASSES, NODE_DISPLAY_NAME_MAPPINGS as QUEUE_DISPLAY
except ImportError:
QUEUE_CLASSES, QUEUE_DISPLAY = {}, {}
try:
from .prompt_queue_folder_node import NODE_CLASS_MAPPINGS as FOLDER_CLASSES, NODE_DISPLAY_NAME_MAPPINGS as FOLDER_DISPLAY
except ImportError:
FOLDER_CLASSES, FOLDER_DISPLAY = {}, {}
try:
from .resolution_select_node import NODE_CLASS_MAPPINGS as RES_CLASSES, NODE_DISPLAY_NAME_MAPPINGS as RES_DISPLAY
except ImportError:
RES_CLASSES, RES_DISPLAY = {}, {}
# Combine all mappings
NODE_CLASS_MAPPINGS = {**CAM_CLASSES, **QUEUE_CLASSES, **FOLDER_CLASSES, **RES_CLASSES}
NODE_DISPLAY_NAME_MAPPINGS = {**CAM_DISPLAY, **QUEUE_DISPLAY, **FOLDER_DISPLAY, **RES_DISPLAY}
# WEB_DIRECTORY is removed because path_helper.js is deleted
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']