Made proper None checks in V1 translation class properties for ComfyNodeV3

This commit is contained in:
kosinkadink1@gmail.com 2025-05-31 04:14:01 -07:00
parent 937d2d5325
commit 8b331c5ca2

View File

@ -360,7 +360,6 @@ class Hidden(str, Enum):
"""API_KEY_COMFY_ORG is an API Key generated by ComfyOrg that allows skipping signing into a ComfyOrg account on frontend.""" """API_KEY_COMFY_ORG is an API Key generated by ComfyOrg that allows skipping signing into a ComfyOrg account on frontend."""
@dataclass @dataclass
class NodeInfoV1: class NodeInfoV1:
input: dict=None input: dict=None
@ -508,77 +507,77 @@ class ComfyNodeV3(ABC):
_DESCRIPTION = None _DESCRIPTION = None
@classproperty @classproperty
def DESCRIPTION(cls): def DESCRIPTION(cls):
if not cls._DESCRIPTION: if cls._DESCRIPTION is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._DESCRIPTION return cls._DESCRIPTION
_CATEGORY = None _CATEGORY = None
@classproperty @classproperty
def CATEGORY(cls): def CATEGORY(cls):
if not cls._CATEGORY: if cls._CATEGORY is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._CATEGORY return cls._CATEGORY
_EXPERIMENTAL = None _EXPERIMENTAL = None
@classproperty @classproperty
def EXPERIMENTAL(cls): def EXPERIMENTAL(cls):
if not cls._EXPERIMENTAL: if cls._EXPERIMENTAL is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._EXPERIMENTAL return cls._EXPERIMENTAL
_DEPRECATED = None _DEPRECATED = None
@classproperty @classproperty
def DEPRECATED(cls): def DEPRECATED(cls):
if not cls._DEPRECATED: if cls._DEPRECATED is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._DEPRECATED return cls._DEPRECATED
_API_NODE = None _API_NODE = None
@classproperty @classproperty
def API_NODE(cls): def API_NODE(cls):
if not cls._API_NODE: if cls._API_NODE is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._API_NODE return cls._API_NODE
_OUTPUT_NODE = None _OUTPUT_NODE = None
@classproperty @classproperty
def OUTPUT_NODE(cls): def OUTPUT_NODE(cls):
if not cls._OUTPUT_NODE: if cls._OUTPUT_NODE is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._OUTPUT_NODE return cls._OUTPUT_NODE
_INPUT_IS_LIST = None _INPUT_IS_LIST = None
@classproperty @classproperty
def INPUT_IS_LIST(cls): def INPUT_IS_LIST(cls):
if not cls._INPUT_IS_LIST: if cls._INPUT_IS_LIST is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._INPUT_IS_LIST return cls._INPUT_IS_LIST
_OUTPUT_IS_LIST = None _OUTPUT_IS_LIST = None
@classproperty @classproperty
def OUTPUT_IS_LIST(cls): def OUTPUT_IS_LIST(cls):
if not cls._OUTPUT_IS_LIST: if cls._OUTPUT_IS_LIST is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._OUTPUT_IS_LIST return cls._OUTPUT_IS_LIST
_RETURN_TYPES = None _RETURN_TYPES = None
@classproperty @classproperty
def RETURN_TYPES(cls): def RETURN_TYPES(cls):
if not cls._RETURN_TYPES: if cls._RETURN_TYPES is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._RETURN_TYPES return cls._RETURN_TYPES
_RETURN_NAMES = None _RETURN_NAMES = None
@classproperty @classproperty
def RETURN_NAMES(cls): def RETURN_NAMES(cls):
if not cls._RETURN_NAMES: if cls._RETURN_NAMES is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._RETURN_NAMES return cls._RETURN_NAMES
_OUTPUT_TOOLTIPS = None _OUTPUT_TOOLTIPS = None
@classproperty @classproperty
def OUTPUT_TOOLTIPS(cls): def OUTPUT_TOOLTIPS(cls):
if not cls._OUTPUT_TOOLTIPS: if cls._OUTPUT_TOOLTIPS is None:
cls.GET_SCHEMA() cls.GET_SCHEMA()
return cls._OUTPUT_TOOLTIPS return cls._OUTPUT_TOOLTIPS