202 lines
7.4 KiB
Python
202 lines
7.4 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
简单的模板数据插入脚本
|
|||
|
|
直接使用数据库连接，不依赖app模块
|
|||
|
|
"""
|
|||
|
|
import pymysql
|
|||
|
|
import json
|
|||
|
|
import uuid
|
|||
|
|
from datetime import datetime, timedelta
|
|||
|
|
import random
|
|||
|
|
|
|||
|
|
# 数据库配置（从config.py中获取）
|
|||
|
|
DB_CONFIG = {
|
|||
|
|
'host': 'gz-cynosdbmysql-grp-d26pzce5.sql.tencentcdb.com',
|
|||
|
|
'port': 24936,
|
|||
|
|
'user': 'root',
|
|||
|
|
'password': '!Rjb12191',
|
|||
|
|
'database': 'agent_db',
|
|||
|
|
'charset': 'utf8mb4'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 模板数据
|
|||
|
|
TEMPLATES = [
|
|||
|
|
{
|
|||
|
|
"name": "智能客服工作流",
|
|||
|
|
"description": "一个完整的智能客服工作流，支持自动回复、问题分类和转人工功能。",
|
|||
|
|
"category": "llm",
|
|||
|
|
"tags": ["客服", "AI", "自动化"],
|
|||
|
|
"is_featured": True
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "数据清洗工作流",
|
|||
|
|
"description": "自动清洗和转换数据，支持多种数据源格式。",
|
|||
|
|
"category": "data_processing",
|
|||
|
|
"tags": ["数据处理", "ETL", "数据清洗"],
|
|||
|
|
"is_featured": False
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "定时数据同步",
|
|||
|
|
"description": "定时从数据库同步数据到另一个系统，支持增量同步。",
|
|||
|
|
"category": "automation",
|
|||
|
|
"tags": ["定时任务", "数据同步", "自动化"],
|
|||
|
|
"is_featured": True
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "API数据聚合",
|
|||
|
|
"description": "从多个API获取数据并聚合处理，支持错误重试。",
|
|||
|
|
"category": "integration",
|
|||
|
|
"tags": ["API", "数据聚合", "集成"],
|
|||
|
|
"is_featured": False
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "邮件通知工作流",
|
|||
|
|
"description": "根据条件发送邮件通知，支持HTML格式和附件。",
|
|||
|
|
"category": "automation",
|
|||
|
|
"tags": ["邮件", "通知", "自动化"],
|
|||
|
|
"is_featured": True
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "内容生成工作流",
|
|||
|
|
"description": "使用AI生成文章、摘要等内容，支持多种格式输出。",
|
|||
|
|
"category": "llm",
|
|||
|
|
"tags": ["AI", "内容生成", "LLM"],
|
|||
|
|
"is_featured": False
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "文件处理工作流",
|
|||
|
|
"description": "批量处理文件，支持上传、下载、格式转换等功能。",
|
|||
|
|
"category": "data_processing",
|
|||
|
|
"tags": ["文件处理", "批量操作"],
|
|||
|
|
"is_featured": False
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "Webhook触发器",
|
|||
|
|
"description": "接收外部Webhook请求并触发工作流，支持多种认证方式。",
|
|||
|
|
"category": "integration",
|
|||
|
|
"tags": ["Webhook", "触发器", "集成"],
|
|||
|
|
"is_featured": True
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 简单的节点和边配置
|
|||
|
|
SIMPLE_NODES = [
|
|||
|
|
{"id": "start", "type": "start", "position": {"x": 100, "y": 100}, "data": {"label": "开始"}},
|
|||
|
|
{"id": "input", "type": "input", "position": {"x": 300, "y": 100}, "data": {"label": "输入"}},
|
|||
|
|
{"id": "process", "type": "transform", "position": {"x": 500, "y": 100}, "data": {"label": "处理"}},
|
|||
|
|
{"id": "output", "type": "output", "position": {"x": 700, "y": 100}, "data": {"label": "输出"}},
|
|||
|
|
{"id": "end", "type": "end", "position": {"x": 900, "y": 100}, "data": {"label": "结束"}}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
SIMPLE_EDGES = [
|
|||
|
|
{"id": "e1", "source": "start", "target": "input"},
|
|||
|
|
{"id": "e2", "source": "input", "target": "process"},
|
|||
|
|
{"id": "e3", "source": "process", "target": "output"},
|
|||
|
|
{"id": "e4", "source": "output", "target": "end"}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
def insert_templates():
|
|||
|
|
"""插入模板数据"""
|
|||
|
|
try:
|
|||
|
|
# 连接数据库
|
|||
|
|
connection = pymysql.connect(**DB_CONFIG)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
with connection.cursor() as cursor:
|
|||
|
|
# 获取第一个用户ID
|
|||
|
|
cursor.execute("SELECT id FROM users LIMIT 1")
|
|||
|
|
user_result = cursor.fetchone()
|
|||
|
|
|
|||
|
|
if not user_result:
|
|||
|
|
print("❌ 没有找到用户，请先创建用户")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
user_id = user_result[0]
|
|||
|
|
print(f"✅ 使用用户ID: {user_id}")
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 检查模板是否已存在
|
|||
|
|
cursor.execute("SELECT COUNT(*) FROM workflow_templates WHERE user_id = %s", (user_id,))
|
|||
|
|
existing_count = cursor.fetchone()[0]
|
|||
|
|
|
|||
|
|
if existing_count > 0:
|
|||
|
|
print(f"ℹ️ 已存在 {existing_count} 个模板，跳过插入")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 插入模板
|
|||
|
|
added_count = 0
|
|||
|
|
for template_info in TEMPLATES:
|
|||
|
|
template_id = str(uuid.uuid4())
|
|||
|
|
nodes_json = json.dumps(SIMPLE_NODES, ensure_ascii=False)
|
|||
|
|
edges_json = json.dumps(SIMPLE_EDGES, ensure_ascii=False)
|
|||
|
|
tags_json = json.dumps(template_info["tags"], ensure_ascii=False)
|
|||
|
|
|
|||
|
|
view_count = random.randint(0, 1000)
|
|||
|
|
use_count = random.randint(0, 100)
|
|||
|
|
rating_count = random.randint(0, 50)
|
|||
|
|
rating_avg = round(random.uniform(3.5, 5.0), 1)
|
|||
|
|
days_ago = random.randint(0, 30)
|
|||
|
|
created_at = datetime.now() - timedelta(days=days_ago)
|
|||
|
|
updated_at = datetime.now() - timedelta(days=random.randint(0, 7))
|
|||
|
|
|
|||
|
|
sql = """
|
|||
|
|
INSERT INTO workflow_templates (
|
|||
|
|
id, name, description, category, tags, nodes, edges,
|
|||
|
|
is_public, is_featured, view_count, use_count, rating_count, rating_avg,
|
|||
|
|
user_id, created_at, updated_at
|
|||
|
|
) VALUES (
|
|||
|
|
%s, %s, %s, %s, %s, %s, %s,
|
|||
|
|
%s, %s, %s, %s, %s, %s,
|
|||
|
|
%s, %s, %s
|
|||
|
|
)
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
cursor.execute(sql, (
|
|||
|
|
template_id,
|
|||
|
|
template_info["name"],
|
|||
|
|
template_info["description"],
|
|||
|
|
template_info["category"],
|
|||
|
|
tags_json,
|
|||
|
|
nodes_json,
|
|||
|
|
edges_json,
|
|||
|
|
True,
|
|||
|
|
template_info["is_featured"],
|
|||
|
|
view_count,
|
|||
|
|
use_count,
|
|||
|
|
rating_count,
|
|||
|
|
rating_avg,
|
|||
|
|
user_id,
|
|||
|
|
created_at,
|
|||
|
|
updated_at
|
|||
|
|
))
|
|||
|
|
added_count += 1
|
|||
|
|
print(f"✅ 添加模板: {template_info['name']}")
|
|||
|
|
|
|||
|
|
connection.commit()
|
|||
|
|
|
|||
|
|
print()
|
|||
|
|
print("=" * 60)
|
|||
|
|
print(f"✅ 数据添加完成！")
|
|||
|
|
print(f" 新增: {added_count} 个模板")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
# 查询总数
|
|||
|
|
cursor.execute("SELECT COUNT(*) FROM workflow_templates WHERE user_id = %s", (user_id,))
|
|||
|
|
total_count = cursor.fetchone()[0]
|
|||
|
|
print(f" 当前模板总数: {total_count} 个")
|
|||
|
|
|
|||
|
|
finally:
|
|||
|
|
connection.close()
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 执行失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("模板市场假数据生成工具")
|
|||
|
|
print("=" * 60)
|
|||
|
|
print()
|
|||
|
|
insert_templates()
|