82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""
|
|
XB-ToolBox 引用任意节点
|
|
========================
|
|
融合 rgthree Fast Muter + Any Switch 的功能。
|
|
JS 端管理动态槽位和单选开关，Python 端直通选中的数据。
|
|
|
|
参照 rgthree-comfy 的 FlexibleOptionalInputType 模式处理动态输入。
|
|
"""
|
|
|
|
|
|
class AnyType(str):
|
|
"""Credit to pythongosssss / rgthree. Always equals any other type."""
|
|
def __ne__(self, other):
|
|
return False
|
|
|
|
|
|
class FlexibleOptionalInputType(dict):
|
|
"""Credit to rgthree-comfy.
|
|
Enables dynamic/flexible input types. __contains__ always returns True,
|
|
and unknown keys return the given type via __getitem__.
|
|
"""
|
|
|
|
def __init__(self, type_, data=None):
|
|
self.type_ = type_
|
|
self.data = data
|
|
if self.data is not None:
|
|
for k, v in self.data.items():
|
|
self[k] = v
|
|
|
|
def __getitem__(self, key):
|
|
if self.data is not None and key in self.data:
|
|
return self.data[key]
|
|
return (self.type_,)
|
|
|
|
def __contains__(self, key):
|
|
return True
|
|
|
|
|
|
any_type = AnyType("*")
|
|
|
|
|
|
class XB_ReferenceAny:
|
|
"""引用任意 — 动态输入槽，下拉菜单选择活跃的上游节点。
|
|
- 下拉菜单"选择"：手动选择
|
|
- "选择"端口可通过 defaultInput 接受上游 STRING 来控制选择"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"选择": ("STRING", {"default": "全部关闭", "defaultInput": True}),
|
|
},
|
|
"optional": FlexibleOptionalInputType(any_type),
|
|
}
|
|
|
|
RETURN_TYPES = (any_type,)
|
|
RETURN_NAMES = ("输出",)
|
|
FUNCTION = "execute"
|
|
CATEGORY = "XB_ToolBox/Utils"
|
|
OUTPUT_NODE = False
|
|
|
|
def execute(self, 选择="", **kwargs):
|
|
# JS 端通过 mode 控制上游节点静音/激活
|
|
# 激活的节点数据会到达 kwargs，静音的不会
|
|
active_input = None
|
|
active_value = None
|
|
for key, value in kwargs.items():
|
|
if value is not None:
|
|
active_input = key
|
|
active_value = value
|
|
break
|
|
|
|
if 选择 and 选择 != "全部关闭":
|
|
val_type = type(active_value).__name__ if active_value is not None else "?"
|
|
print(f"[引用任意] 选中 [{选择}] → 上游 [{active_input}] ({val_type}) → 输出")
|
|
elif active_value is not None:
|
|
print(f"[引用任意] 直通 → 上游 [{active_input}] → 输出")
|
|
|
|
if active_value is not None:
|
|
return (active_value,)
|
|
return (None,)
|