40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
import uuid
|
||
|
||
|
||
class TaskIdGenerate:
|
||
"""TaskID生成器：用户可传入或自动生成TaskID"""
|
||
|
||
@classmethod
|
||
def INPUT_TYPES(cls):
|
||
return {
|
||
"required": {},
|
||
"optional": {
|
||
"custom_task_id": ("STRING", {"default": "", "placeholder": "留空则自动生成"}),
|
||
},
|
||
}
|
||
|
||
RETURN_TYPES = ("STRING",)
|
||
RETURN_NAMES = ("task_id",)
|
||
FUNCTION = "generate_task_id"
|
||
OUTPUT_NODE = False
|
||
CATEGORY = "utils"
|
||
|
||
def generate_task_id(self, custom_task_id=""):
|
||
if custom_task_id and custom_task_id.strip():
|
||
# 用户输入了自定义ID
|
||
task_id = custom_task_id.strip()
|
||
print(f"📝 使用自定义TaskID: {task_id}")
|
||
else:
|
||
# 自动生成UUID
|
||
task_id = str(uuid.uuid4())
|
||
print(f"🎲 自动生成TaskID: {task_id}")
|
||
|
||
return (task_id,)
|
||
|
||
|
||
NODE_CLASS_MAPPINGS = {
|
||
"TaskIdGenerate": TaskIdGenerate
|
||
}
|
||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||
"TaskIdGenerate": "TaskID生成器"
|
||
} |