52 lines
1022 B
Python
52 lines
1022 B
Python
WEB_DIRECTORY = "./js"
|
|
|
|
class FlowManager:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {}
|
|
}
|
|
|
|
RETURN_TYPES = ("NULL",)
|
|
FUNCTION = "run"
|
|
CATEGORY = "Comfy-Flow"
|
|
OUTPUT_NODE = True
|
|
|
|
def run(self):
|
|
return (None,)
|
|
|
|
class FlowStepPointer:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"Step": ("INT", {
|
|
"default": 0,
|
|
"min": 0,
|
|
"step": 1
|
|
}),
|
|
"Name": ("STRING", {
|
|
"default": "Step X"
|
|
})
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("NULL",)
|
|
FUNCTION = "run"
|
|
CATEGORY = "Comfy-Flow"
|
|
OUTPUT_NODE = True
|
|
|
|
def run(self):
|
|
return (None,)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FlowManager": FlowManager,
|
|
"FlowStepPointer": FlowStepPointer
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FlowManager": "Flow-Manager",
|
|
"FlowStepPointer": "Flow-StepPointer"
|
|
}
|