#!/usr/bin/env python3
from neo4j import GraphDatabase

uri = "bolt://neo4j:7687"
user = "neo4j"
password = "Wecom@2026"

driver = GraphDatabase.driver(uri, auth=(user, password))

with driver.session() as session:
    # 清除旧数据
    session.run("MATCH (n) DETACH DELETE n")

    # 添加测试数据
    session.run("""
        CREATE (i1:Issue {name: "打印机驱动安装", category: "硬件问题", uuid: "issue-001"})
        CREATE (a1:Action {name: "打印机驱动安装步骤", description: "1. 打开控制面板 2. 添加打印机 3. 选择手动添加"})
        CREATE (i1)-[:HAS_ACTION]->(a1)
    """)

    session.run("""
        CREATE (i2:Issue {name: "网络连不上", category: "网络问题", uuid: "issue-002"})
        CREATE (a2:Action {name: "网络诊断步骤", description: "1. 检查网线 2. 重启路由器 3. 检查IP配置"})
        CREATE (i2)-[:HAS_ACTION]->(a2)
    """)

    session.run("""
        CREATE (i3:Issue {name: "邮箱无法收发", category: "软件问题", uuid: "issue-003"})
        CREATE (a3:Action {name: "邮箱故障排除", description: "1. 检查网络 2. 清除缓存 3. 重新登录"})
        CREATE (i3)-[:HAS_ACTION]->(a3)
    """)

    # 验证
    result = session.run("MATCH (i:Issue) RETURN i.uuid, i.name")
    for record in result:
        print(f"uuid: {record['i.uuid']}, name: {record['i.name']}")

driver.close()
print("Done!")