-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathadd_delay_node.py
43 lines (35 loc) · 1.09 KB
/
add_delay_node.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import time
class AnyType(str):
def __ne__(self, __value: object) -> bool:
return False
any_type = AnyType("*")
class add_delay_node:
"""Node that adds a configurable delay between operations"""
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"input": (any_type, {"defaultInput": True}),
"delay_seconds": ("FLOAT", {
"default": 1.0,
"min": 0.0,
"step": 0.1
}),
},
}
RETURN_TYPES = (any_type,)
RETURN_NAMES = ("output",)
FUNCTION = "add_delay"
CATEGORY = "utils"
def add_delay(self, input, delay_seconds):
delay_text = f"{delay_seconds:.1f} second{'s' if delay_seconds != 1 else ''}"
print(f"[Delay Node] Starting delay of {delay_text}")
time.sleep(delay_seconds)
print(f"[Delay Node] Delay of {delay_text} completed")
return (input,)
NODE_CLASS_MAPPINGS = {
"Add Delay": add_delay_node
}
NODE_DISPLAY_NAME_MAPPINGS = {
"Add Delay": "Add Delay"
}