69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
import torch
|
|
|
|
# --- 核心修复部分开始 ---
|
|
class AnyType(str):
|
|
"""
|
|
这是一个特殊的类，用来欺骗 ComfyUI 的类型检查系统。
|
|
无论传入什么类型，它都返回 True (匹配成功)。
|
|
"""
|
|
def __ne__(self, __value):
|
|
return False
|
|
def __eq__(self, __value):
|
|
return True
|
|
|
|
# 实例化一个万能类型对象
|
|
any_type = AnyType("*")
|
|
# --- 核心修复部分结束 ---
|
|
|
|
class SelectorIntNode:
|
|
"""
|
|
根据连接情况输出序号的节点。
|
|
修正了类型验证错误。
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
# 使用 any_type 代替之前的 "*"
|
|
"input_1": (any_type, {}),
|
|
"input_2": (any_type, {}),
|
|
"input_3": (any_type, {}),
|
|
"input_4": (any_type, {}),
|
|
"input_5": (any_type, {}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT",)
|
|
RETURN_NAMES = ("selected_index",)
|
|
FUNCTION = "get_index"
|
|
CATEGORY = "MyCustomNodes/Utils"
|
|
|
|
def get_index(self, **kwargs):
|
|
# 定义总端口数
|
|
total_inputs = 5
|
|
|
|
# 倒序检查：从 5 到 1
|
|
for i in range(total_inputs, 0, -1):
|
|
key = f"input_{i}"
|
|
|
|
# 检查是否有连接
|
|
if key in kwargs:
|
|
value = kwargs[key]
|
|
|
|
# 逻辑：如果连接了东西，且不是 None (即没有被 Mute/忽略)
|
|
# 注意：在某些 ComfyUI 版本中，被 Mute 的节点可能根本不会传参进 kwargs，
|
|
# 或者传进来是 None。这个判断能同时覆盖这两种情况。
|
|
if value is not None:
|
|
# 额外检查：防止某些特殊节点传空列表等情况
|
|
# 只要有值，就认为是有效连接
|
|
return (i,)
|
|
|
|
# 如果全都被忽略或没连接，返回 0
|
|
return (0,)
|
|
|