You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.9 KiB
63 lines
1.9 KiB
import json
|
|
from Config.Config import *
|
|
from Neo4j.Neo4jExecutor import *
|
|
|
|
def json_to_cypher(data):
|
|
"""将知识体系JSON转换为Cypher插入脚本"""
|
|
cypher = []
|
|
seen = set()
|
|
|
|
def process_node(node, parent_id=None):
|
|
nonlocal seen
|
|
|
|
node_id = node['value']
|
|
node_name = node['title'].replace("'", "''")
|
|
|
|
# 创建当前节点
|
|
if node_id not in seen:
|
|
cypher.append(f"MERGE (n:KnowledgePoint {{id: '{node_id}'}}) SET n.name = '{node_name}';")
|
|
seen.add(node_id)
|
|
|
|
# 创建父子关系
|
|
if parent_id:
|
|
cypher.append(f"""
|
|
MATCH (parent:KnowledgePoint {{id: '{parent_id}'}}),
|
|
(child:KnowledgePoint {{id: '{node_id}'}})
|
|
MERGE (parent)-[:HAS_SUB_POINT]->(child);""")
|
|
|
|
# 递归处理子节点
|
|
if 'children' in node and not node['isLeaf']:
|
|
for child in node['children']:
|
|
process_node(child, parent_id=node_id)
|
|
|
|
# 处理根节点
|
|
for root in data['data']['tree']:
|
|
process_node(root)
|
|
|
|
# 处理根节点的父级关系（如果有）
|
|
if root.get('parentValue'):
|
|
cypher.append(f"MERGE (parent:KnowledgePoint {{id: '{root['parentValue']}'}});")
|
|
cypher.append(f"""
|
|
MATCH (parent:KnowledgePoint {{id: '{root['parentValue']}'}}),
|
|
(child:KnowledgePoint {{id: '{root['value']}'}})
|
|
MERGE (parent)-[:HAS_SUB_POINT]->(child);""")
|
|
|
|
return '\n'.join(cypher)
|
|
|
|
|
|
# 使用示例
|
|
if __name__ == '__main__':
|
|
executor = Neo4jExecutor(
|
|
uri=NEO4J_URI,
|
|
auth=NEO4J_AUTH
|
|
)
|
|
# 清库
|
|
clear(executor.graph)
|
|
|
|
# 这里替换成你的JSON数据变量
|
|
with open('小学数学知识点体系.json', 'r',encoding='utf-8') as f:
|
|
your_json_data = json.load(f)
|
|
cypherText=json_to_cypher(your_json_data)
|
|
executor.execute_cypher_text(cypherText)
|
|
print("数据插入成功！")
|