83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
"""A simple debug node used to quickly check output of nodes in ComfyUI"""
|
|
|
|
# AnyType was ruthlessly pilfered from https://github.com/pythongosssss/ComfyUI-Custom-Scripts
|
|
class AnyType(str):
|
|
def __ne__(self, __value: object) -> bool:
|
|
return False
|
|
|
|
ANY = AnyType("*")
|
|
|
|
class WTFDebugNode:
|
|
"""What's That Field? Displays string representation of node output to ease debugging
|
|
|
|
Displays some common attributes, if available:
|
|
- len()
|
|
- .shape
|
|
- if the input was an iterable, the type of the first child item
|
|
|
|
Displays str(Exception) on for unhandled exceptions"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"anything": (ANY,),
|
|
}
|
|
}
|
|
|
|
OUTPUT_NODE = True
|
|
|
|
INPUT_IS_LIST = True
|
|
# OUTPUT_IS_LIST = (False,)
|
|
RETURN_TYPES = ()
|
|
FUNCTION = "execute"
|
|
|
|
CATEGORY = "debug"
|
|
|
|
def execute(self, anything=None):
|
|
out = []
|
|
|
|
for item in anything:
|
|
info = {
|
|
"type": str(type(item)),
|
|
}
|
|
|
|
try:
|
|
info["len"] = len(item)
|
|
except TypeError:
|
|
pass
|
|
except Exception as e:
|
|
info["len"] = str(e)
|
|
|
|
if not isinstance(item, str):
|
|
try:
|
|
first_iter_item = next(iter(item))
|
|
info["firstIterItem"] = str(type(first_iter_item))
|
|
except TypeError:
|
|
pass
|
|
# info["firstIterItem"] = ["Wasn't a list"]
|
|
except StopIteration:
|
|
info["firstIterItem"] = "List had no items"
|
|
except Exception as e:
|
|
info["firstIterItem"] = f"Exception [firstIterItem]: {str(e)}"
|
|
|
|
try:
|
|
info["shape"] = str(item.shape)
|
|
except AttributeError:
|
|
pass
|
|
except Exception as e:
|
|
info["shape"] = str(e)
|
|
|
|
info["value"] = str(item) if item is not None else None
|
|
out.append(info)
|
|
|
|
return { "ui": { "items": out } }
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"WTFDebugNode": WTFDebugNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"WTFDebugNode": "🐜 WTF?",
|
|
}
|