102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
import requests
|
|
import json
|
|
import os
|
|
import time
|
|
|
|
# 加载配置文件
|
|
def load_config():
|
|
config_path = os.path.join(os.path.dirname(__file__), "config.json")
|
|
default_config = {
|
|
"default_webhook_url": "https://example.com/webhook",
|
|
"enable_debug_logs": True
|
|
}
|
|
|
|
try:
|
|
if os.path.exists(config_path):
|
|
with open(config_path, "r") as f:
|
|
config = json.load(f)
|
|
# 确保config中包含必要的字段
|
|
for key, value in default_config.items():
|
|
if key not in config:
|
|
config[key] = value
|
|
return config
|
|
else:
|
|
# 如果配置文件不存在，创建一个默认配置
|
|
with open(config_path, "w") as f:
|
|
json.dump(default_config, f, indent=4)
|
|
return default_config
|
|
except Exception as e:
|
|
print(f"加载配置文件出错: {str(e)}")
|
|
return default_config
|
|
|
|
# 全局配置
|
|
CONFIG = load_config()
|
|
|
|
class WebhookNotifierNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"images": ("IMAGE",),
|
|
"webhook_url": ("STRING", {"default": CONFIG.get("default_webhook_url", "https://example.com/webhook")})
|
|
},
|
|
"optional": {
|
|
"workflow_name": ("STRING", {"default": "默认工作流"}),
|
|
"additional_info": ("STRING", {"default": "{}", "multiline": True})
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE",)
|
|
RETURN_NAMES = ("images",)
|
|
FUNCTION = "notify"
|
|
CATEGORY = "utils"
|
|
|
|
def notify(self, images, webhook_url, workflow_name="默认工作流", additional_info="{}", prompt=None):
|
|
try:
|
|
# 从ComfyUI获取prompt_id (execution_id)
|
|
prompt_id = "unknown"
|
|
if prompt is not None and "extra_pnginfo" in prompt and "workflow" in prompt["extra_pnginfo"]:
|
|
if "execution_id" in prompt["extra_pnginfo"]["workflow"]:
|
|
prompt_id = prompt["extra_pnginfo"]["workflow"]["execution_id"]
|
|
elif "client_id" in prompt["extra_pnginfo"]["workflow"]:
|
|
prompt_id = prompt["extra_pnginfo"]["workflow"]["client_id"]
|
|
|
|
# 准备基本信息
|
|
image_info = {
|
|
"image_count": len(images),
|
|
"dimensions": f"{images.shape[1]}x{images.shape[2]}"
|
|
}
|
|
|
|
# 尝试解析额外信息
|
|
try:
|
|
extra_info = json.loads(additional_info) if additional_info else {}
|
|
except json.JSONDecodeError:
|
|
extra_info = {}
|
|
|
|
# 构造payload
|
|
payload = {
|
|
"status": "completed",
|
|
"prompt_id": prompt_id,
|
|
"workflow_name": workflow_name,
|
|
"timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
|
|
"images": image_info,
|
|
**extra_info
|
|
}
|
|
|
|
# 发送webhook
|
|
response = requests.post(
|
|
webhook_url,
|
|
json=payload,
|
|
headers={"Content-Type": "application/json"}
|
|
)
|
|
|
|
if response.status_code >= 400:
|
|
print(f"Webhook通知失败: {response.status_code} - {response.text}")
|
|
else:
|
|
print(f"Webhook通知成功: {response.status_code}, prompt_id: {prompt_id}")
|
|
|
|
except Exception as e:
|
|
print(f"发送webhook时出错: {str(e)}")
|
|
|
|
# 返回原始图像
|
|
return (images,) |