18 lines
575 B
Python
18 lines
575 B
Python
from .py.log import log
|
|
# import importlib.util # for auto import
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
def AddMapping(node_class, display_name):
|
|
NODE_CLASS_MAPPINGS[node_class.NAME] = node_class
|
|
NODE_DISPLAY_NAME_MAPPINGS[node_class.NAME] = display_name
|
|
|
|
# Load Image Nodes
|
|
try:
|
|
from .py.nodes_image import ImageResizeTo, Image2BW
|
|
AddMapping(ImageResizeTo, "Image Resize To 🐦🔥")
|
|
AddMapping(Image2BW, "Image To Black and White 🐦🔥")
|
|
except ImportError:
|
|
log("Failed to load image nodes", msg_color="BRIGHT_GREEN")
|