54 lines
1.1 KiB
Python
54 lines
1.1 KiB
Python
import time
|
|
|
|
class AnyType(str):
|
|
"""A special type that compares equal to everything, allowing any connection."""
|
|
def __ne__(self, __value: object) -> bool:
|
|
return False
|
|
|
|
any_type = AnyType("*")
|
|
|
|
|
|
class AddDelay:
|
|
"""
|
|
Adds a delay (sleep) in seconds, then passes the input through unchanged.
|
|
Default delay = 6.0 seconds.
|
|
"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input": (any_type,),
|
|
"delay": ("FLOAT", {
|
|
"default": 6.0,
|
|
"min": 0.0,
|
|
"max": 3600.0,
|
|
"step": 0.1
|
|
}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = (any_type,)
|
|
RETURN_NAMES = ("output",)
|
|
FUNCTION = "run"
|
|
CATEGORY = "flow-assistor"
|
|
|
|
def run(self, input, delay=6.0):
|
|
try:
|
|
d = float(delay)
|
|
except Exception:
|
|
d = 6.0
|
|
|
|
if d > 0:
|
|
time.sleep(d)
|
|
|
|
return (input,)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AddDelay": AddDelay,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AddDelay": "Add Delay",
|
|
} |