32 lines
927 B
Python
32 lines
927 B
Python
# "Design-focused node suite for ComfyUI."
|
|
# Copyright 2026 Augment Studio
|
|
# Augmentstudio.app
|
|
|
|
from comfy_execution.graph_utils import ExecutionBlocker
|
|
|
|
|
|
class SwitchNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input": ("BOOLEAN", {"forceInput": True}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("TRIGGER", "TRIGGER", "TRIGGER")
|
|
RETURN_NAMES = ("on_true", "on_false", "trigger")
|
|
OUTPUT_NODE = True
|
|
FUNCTION = "execute"
|
|
CATEGORY = "Augment/Utils"
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, **kwargs):
|
|
return True
|
|
|
|
def execute(self, input):
|
|
on_true = "done" if input else ExecutionBlocker(None)
|
|
on_false = ExecutionBlocker(None) if input else "done"
|
|
ui_text = f"Input: {input}\nPath: {'on_true' if input else 'on_false'}"
|
|
return {"ui": {"text": [ui_text]}, "result": (on_true, on_false, "done")}
|