diff --git a/__init__.py b/__init__.py index ba5ad6c..dd5082d 100644 --- a/__init__.py +++ b/__init__.py @@ -1,3 +1,7 @@ -from .predefined_resolution_node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS +from .predefined_resolution_node import NODE_CLASS_MAPPINGS as RESOLUTION_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as RESOLUTION_DISPLAY_MAPPINGS +from .none_constant_node import NODE_CLASS_MAPPINGS as NONE_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as NONE_DISPLAY_MAPPINGS + +NODE_CLASS_MAPPINGS = {**RESOLUTION_MAPPINGS, **NONE_MAPPINGS} +NODE_DISPLAY_NAME_MAPPINGS = {**RESOLUTION_DISPLAY_MAPPINGS, **NONE_DISPLAY_MAPPINGS} __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] diff --git a/none_constant_node.py b/none_constant_node.py new file mode 100644 index 0000000..f1b1ed8 --- /dev/null +++ b/none_constant_node.py @@ -0,0 +1,23 @@ +class NoneConstantNode: + @classmethod + def INPUT_TYPES(cls): + return { + "required": {}, + "optional": {} + } + + RETURN_TYPES = ("*",) + RETURN_NAMES = ("value",) + FUNCTION = "get_none" + CATEGORY = "utils" + + def get_none(self): + return (None,) + +NODE_CLASS_MAPPINGS = { + "NoneConstant": NoneConstantNode +} + +NODE_DISPLAY_NAME_MAPPINGS = { + "NoneConstant": "None Constant" +}