Simon
184 lines
7.8 KiB
Python
184 lines
7.8 KiB
Python
# =============================================================================
|
||
# 企微IT智能服务台 — 会话 Mock 数据填充脚本
|
||
# =============================================================================
|
||
# 说明：往 SQLite 数据库 conversations 表写入 15 条模拟会话
|
||
# 用法：cd backend && ..\venv\Scripts\python.exe seed_conversations.py
|
||
# =============================================================================
|
||
|
||
import sqlite3
|
||
from datetime import datetime, timezone, timedelta
|
||
import json
|
||
import uuid
|
||
|
||
DB_PATH = r"C:\Users\simon\wecom_it_smart_desk\backend\it_smart_desk.db"
|
||
|
||
# 当前时间（UTC）
|
||
NOW = datetime.now(tz=timezone.utc)
|
||
ISO = lambda dt: dt.isoformat()
|
||
|
||
def make_conv(cid, user_id, name, dept, position, level,
|
||
status, is_vip, is_pinned, is_todo,
|
||
urgency, tags, assigned_agent, impact, is_blocking, emotion,
|
||
last_summary, mins_ago):
|
||
"""构造一条会话记录（tuple）"""
|
||
seq = int(cid.split("-")[1])
|
||
hours_ago = max(0, 48 - seq * 3)
|
||
created = NOW - timedelta(hours=hours_ago)
|
||
updated = NOW - timedelta(minutes=mins_ago) if mins_ago else created
|
||
last_msg = NOW - timedelta(minutes=mins_ago) if mins_ago else None
|
||
|
||
return (
|
||
cid, # id
|
||
"ww1234567890", # corp_id
|
||
user_id, # employee_id
|
||
name, # employee_name
|
||
dept, # department
|
||
position, # position
|
||
level, # level
|
||
status, # status
|
||
1 if is_vip else 0, # is_vip
|
||
1 if is_pinned else 0, # is_pinned
|
||
1 if is_todo else 0, # is_todo
|
||
urgency, # urgency_score
|
||
json.dumps(tags, ensure_ascii=False), # tags (JSON)
|
||
assigned_agent, # assigned_agent_id
|
||
json.dumps([], ensure_ascii=False), # collaborating_agent_ids
|
||
0, # ai_substantive_reply_count
|
||
impact, # impact_scope
|
||
1 if is_blocking else 0, # is_blocking
|
||
emotion, # emotion_state
|
||
None, # dify_conversation_id
|
||
ISO(last_msg) if last_msg else None, # last_message_at
|
||
last_summary, # last_message_summary
|
||
ISO(created), # created_at
|
||
ISO(updated), # updated_at
|
||
)
|
||
|
||
rows = [
|
||
# ── 排队中（queued）────
|
||
make_conv("conv-001", "user-001", "张伟", "财务部", "经理", "P6",
|
||
"queued", False, False, False,
|
||
5, {"hand_raise": True, "emotion": "urgent"},
|
||
None, 0, False, "urgent",
|
||
"VPN 连不上，报错 Error 691，今天要急着报税！", 5),
|
||
|
||
make_conv("conv-002", "user-002", "李娜", "设计部", "设计师", "P5",
|
||
"queued", False, False, True,
|
||
3, {"emotion": "worried"},
|
||
None, 0, False, "worried",
|
||
"PS 2026 安装后一直闪退，急需用", 12),
|
||
|
||
make_conv("conv-003", "user-003", "王强", "市场部", "总监", "P7",
|
||
"queued", True, False, False,
|
||
4, {"hand_raise": True, "need_intervene": True},
|
||
None, 8, True, "angry",
|
||
"邮箱满了，发送失败，影响了3个客户的报价单！", 1),
|
||
|
||
make_conv("conv-004", "user-004", "刘芳", "人事部", "HRBP", "P6",
|
||
"queued", False, False, False,
|
||
2, {},
|
||
None, 0, False, "neutral",
|
||
"OA 系统白屏，其他同事也有同样问题", 30),
|
||
|
||
make_conv("conv-005", "user-005", "陈明", "研发部", "高级工程师", "P7",
|
||
"queued", True, False, False,
|
||
4, {"need_intervene": True},
|
||
None, 2, True, "urgent",
|
||
"生产数据库只读权限申请，今天上线要用", 8),
|
||
|
||
# ── 服务中（serving）────
|
||
make_conv("conv-006", "user-006", "赵敏", "市场部", "专员", "P4",
|
||
"serving", False, False, False,
|
||
3, {"emotion": "worried"},
|
||
"agent-001", 0, False, "worried",
|
||
"打印机驱动装好了，但是打印出来是乱码", 60),
|
||
|
||
make_conv("conv-007", "user-007", "周婷", "行政部", "行政主管", "P5",
|
||
"serving", False, True, False,
|
||
2, {},
|
||
"agent-001", 0, False, "neutral",
|
||
"会议室投影仪怎么切换到HDMI模式？下午有客户来访", 120),
|
||
|
||
make_conv("conv-008", "user-008", "吴婷", "人事部", "薪酬经理", "P6",
|
||
"serving", False, False, True,
|
||
3, {"emotion": "angry"},
|
||
"agent-001", 1, False, "angry",
|
||
"新员工入职的笔记本磁盘只有 256G，完全不够用", 45),
|
||
|
||
make_conv("conv-009", "user-009", "刘军", "销售部", "大区经理", "P7",
|
||
"serving", True, False, False,
|
||
4, {"hand_raise": True},
|
||
"agent-001", 5, True, "urgent",
|
||
"CRM 系统卡死，正在和客户通话，急！！", 10),
|
||
|
||
# ── AI 处理中（ai_handling）────
|
||
make_conv("conv-010", "user-010", "孙磊", "运维部", "SRE", "P6",
|
||
"ai_handling", False, False, False,
|
||
2, {},
|
||
None, 0, False, "neutral",
|
||
"Prometheus 告警：磁盘使用率超过 85%，怎么清理？", 90),
|
||
|
||
make_conv("conv-011", "user-011", "马超", "研发部", "实习生", "P3",
|
||
"ai_handling", False, False, False,
|
||
1, {},
|
||
None, 0, False, "neutral",
|
||
"Git 提交时提示 'fatal: unable to access'，怎么解决？", 150),
|
||
|
||
# ── 已结单（resolved）────
|
||
make_conv("conv-012", "user-001", "张伟", "财务部", "经理", "P6",
|
||
"resolved", False, False, False,
|
||
3, {},
|
||
"agent-001", 0, False, "neutral",
|
||
"谢谢，VPN 问题已解决，是密码过期了", 300),
|
||
|
||
make_conv("conv-013", "user-007", "周婷", "行政部", "行政主管", "P5",
|
||
"resolved", False, False, False,
|
||
2, {},
|
||
"agent-001", 0, False, "neutral",
|
||
"IT 已来处理，投影仪换了新的HDMI线，正常了", 500),
|
||
|
||
make_conv("conv-014", "user-012", "杨阳", "公关部", "专员", "P4",
|
||
"resolved", False, False, False,
|
||
1, {},
|
||
"agent-001", 0, False, "neutral",
|
||
"耳机没声音的问题解决了，是静音键被误触了😅", 800),
|
||
|
||
make_conv("conv-015", "user-005", "刘芳", "人事部", "HRBP", "P6",
|
||
"resolved", False, False, False,
|
||
2, {},
|
||
"agent-001", 0, False, "neutral",
|
||
"OA 白屏问题已恢复，是浏览器缓存导致的，清掉就好了", 1200),
|
||
]
|
||
|
||
# 插入数据库（INSERT OR IGNORE 避免主键冲突，重复运行安全）
|
||
conn = sqlite3.connect(DB_PATH)
|
||
cur = conn.cursor()
|
||
|
||
cur.execute("SELECT COUNT(*) FROM conversations")
|
||
before = cur.fetchone()[0]
|
||
print(f"插入前已有 {before} 条记录")
|
||
|
||
sql = """
|
||
INSERT OR IGNORE INTO conversations (
|
||
id, corp_id, employee_id, employee_name, department, position, level,
|
||
status, is_vip, is_pinned, is_todo, urgency_score, tags,
|
||
assigned_agent_id, collaborating_agent_ids, ai_substantive_reply_count,
|
||
impact_scope, is_blocking, emotion_state, dify_conversation_id,
|
||
last_message_at, last_message_summary, created_at, updated_at
|
||
) VALUES (
|
||
?,?,?,?,?,?,?,
|
||
?,?,?,?,?,?,
|
||
?,?,?,
|
||
?,?,?,?,
|
||
?,?,?,?
|
||
)
|
||
"""
|
||
|
||
cur.executemany(sql, rows)
|
||
conn.commit()
|
||
|
||
cur.execute("SELECT COUNT(*) FROM conversations")
|
||
after = cur.fetchone()[0]
|
||
print(f"✅ 插入后共 {after} 条记录（新增 {after - before} 条会话 Mock 数据）！")
|
||
conn.close()
|