71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
import time
|
|
|
|
class CooldownStringNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"STRING" : ("STRING", {"forceInput": True}),
|
|
"sleep" : ("INT", {"default": 1, "min": 0, "max": 300})
|
|
},
|
|
}
|
|
|
|
OUTPUT_NODE = True
|
|
RETURN_TYPES = ("STRING",)
|
|
CATEGORY = "Cooldown"
|
|
FUNCTION = "main"
|
|
|
|
def main(self, STRING, sleep):
|
|
time.sleep(sleep)
|
|
return (STRING,)
|
|
|
|
class CooldownIntNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"INT" : ("INT", {"forceInput": True}),
|
|
"sleep" : ("INT", {"default": 1, "min": 0, "max": 300})
|
|
},
|
|
}
|
|
|
|
OUTPUT_NODE = True
|
|
RETURN_TYPES = ("INT",)
|
|
CATEGORY = "Cooldown"
|
|
FUNCTION = "main"
|
|
|
|
def main(self, INT, sleep):
|
|
time.sleep(sleep)
|
|
return (INT,)
|
|
|
|
class CooldownImageNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"IMAGE" : ("IMAGE", {"forceInput": True}),
|
|
"sleep" : ("INT", {"default": 1, "min": 0, "max": 300})
|
|
},
|
|
}
|
|
|
|
OUTPUT_NODE = True
|
|
RETURN_TYPES = ("IMAGE",)
|
|
CATEGORY = "Cooldown"
|
|
FUNCTION = "main"
|
|
|
|
def main(self, IMAGE, sleep):
|
|
time.sleep(sleep)
|
|
return (IMAGE,)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CooldownStringNode": CooldownStringNode,
|
|
"CooldownIntNode": CooldownIntNode,
|
|
"CooldownImageNode": CooldownImageNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CooldownStringNode": "Cooldown -STRING-",
|
|
"CooldownIntNode": "Cooldown -INT-",
|
|
"CooldownImageNode": "Cooldown -IMAGE-",
|
|
}
|