|
|
import json
|
|
|
|
|
|
from Config.Config import NEO4J_URI, NEO4J_AUTH
|
|
|
from Util.Neo4jExecutor import *
|
|
|
|
|
|
# Neo4j连接配置
|
|
|
graph = Graph(NEO4J_URI, auth=NEO4J_AUTH)
|
|
|
|
|
|
|
|
|
class Py2neoLoader:
|
|
|
def __init__(self):
|
|
|
self.node_cache = {}
|
|
|
self.tx = graph.begin() # 初始化事务
|
|
|
|
|
|
def create_literacy_nodes(self, data):
|
|
|
try:
|
|
|
print("Received data structure keys:", data.keys())
|
|
|
|
|
|
# 创建核心素养节点
|
|
|
if "tree" in data and len(data["tree"]) > 0:
|
|
|
self._create_core_node(data["tree"][0])
|
|
|
else:
|
|
|
raise ValueError("Invalid data structure: missing 'tree' array")
|
|
|
|
|
|
# 创建关联体系节点（处理第二个节点）
|
|
|
if len(data["tree"]) > 1:
|
|
|
self._create_relation_system(data["tree"][1])
|
|
|
graph.commit(self.tx) # 替换 self.tx.commit()
|
|
|
print(f"数据导入成功！共处理{len(self.node_cache)}个节点")
|
|
|
except Exception as e:
|
|
|
graph.rollback(self.tx) # 替换 self.tx.rollback()
|
|
|
print("主流程错误:", str(e))
|
|
|
raise
|
|
|
|
|
|
def _create_core_node(self, node):
|
|
|
try:
|
|
|
core_node = Node("CoreLiteracy",
|
|
|
title=node.get("title", "未命名节点"),
|
|
|
value=node.get("value", "CORE_DEFAULT"),
|
|
|
key=node.get("key", "ROOT_DEFAULT"))
|
|
|
self.tx.merge(core_node, "CoreLiteracy", "value")
|
|
|
self.node_cache[node.get("value")] = core_node
|
|
|
print(f"创建核心节点: {core_node}")
|
|
|
|
|
|
if "children" in node:
|
|
|
for child in node["children"]:
|
|
|
self._create_child_node(child, core_node)
|
|
|
except Exception as e:
|
|
|
print(f"创建核心节点出错: {str(e)}")
|
|
|
print("问题节点数据:", node)
|
|
|
raise
|
|
|
|
|
|
def _create_child_node(self, node, parent):
|
|
|
try:
|
|
|
if "value" not in node:
|
|
|
node["value"] = f"AUTO_VALUE_{len(self.node_cache) + 1}"
|
|
|
print(f"为节点生成默认value: {node['value']}")
|
|
|
|
|
|
child_node = Node("LiteracyNode",
|
|
|
title=node.get("title", "未命名子节点"),
|
|
|
value=node["value"],
|
|
|
key=node.get("key", ""),
|
|
|
parentValue=node.get("parentValue", ""),
|
|
|
preSkills=node.get("preSkills", []))
|
|
|
self.tx.merge(child_node, "LiteracyNode", "value")
|
|
|
self.node_cache[node["value"]] = child_node
|
|
|
|
|
|
if parent is not None:
|
|
|
rel = Relationship(parent, "HAS_SUB_NODE", child_node)
|
|
|
self.tx.merge(rel)
|
|
|
print(f"创建关系: {parent['value']} -> {child_node['value']}")
|
|
|
|
|
|
# 处理前置技能关系
|
|
|
for pre_skill in node.get("preSkills", []):
|
|
|
if pre_skill in self.node_cache:
|
|
|
pre_node = self.node_cache[pre_skill]
|
|
|
rel = Relationship(pre_node, "REQUIRES", child_node)
|
|
|
self.tx.merge(rel)
|
|
|
else:
|
|
|
print(f"警告: 前置技能节点 {pre_skill} 不存在")
|
|
|
|
|
|
# 递归处理子节点
|
|
|
if "children" in node:
|
|
|
for sub_child in node.get("children", []):
|
|
|
self._create_child_node(sub_child, child_node)
|
|
|
except Exception as e:
|
|
|
print(f"创建子节点出错: {str(e)}")
|
|
|
print("问题节点数据:", node)
|
|
|
raise
|
|
|
|
|
|
def _create_relation_system(self, relation_system):
|
|
|
try:
|
|
|
# 创建关联体系根节点
|
|
|
sys_node = Node("RelationSystem",
|
|
|
title=relation_system.get("title", "未命名关联体系"),
|
|
|
value=relation_system.get("value", "REL_SYS_DEFAULT"),
|
|
|
path=relation_system.get("path", []))
|
|
|
self.tx.merge(sys_node, "RelationSystem", "value")
|
|
|
print(f"创建关联体系节点: {sys_node}")
|
|
|
|
|
|
# 处理关联链
|
|
|
for chain in relation_system.get("children", []):
|
|
|
chain_node = Node("RelationChain",
|
|
|
title=chain.get("title", "未命名关联链"),
|
|
|
value=chain.get("value", f"CHAIN_{len(self.node_cache) + 1}"),
|
|
|
path=chain.get("path", []))
|
|
|
self.tx.merge(chain_node, "RelationChain", "value")
|
|
|
self.tx.merge(Relationship(sys_node, "HAS_CHAIN", chain_node))
|
|
|
|
|
|
# 处理路径关系
|
|
|
for path_str in chain.get("path", []):
|
|
|
nodes = [n.strip() for n in path_str.replace("+", "→").split("→")]
|
|
|
for i in range(len(nodes) - 1):
|
|
|
source = self.node_cache.get(nodes[i])
|
|
|
target = self.node_cache.get(nodes[i + 1])
|
|
|
if source and target:
|
|
|
rel = Relationship(source, "NEXT_STEP", target)
|
|
|
self.tx.merge(rel)
|
|
|
else:
|
|
|
print(f"路径节点缺失: {nodes[i]}→{nodes[i + 1]}")
|
|
|
except Exception as e:
|
|
|
print(f"创建关联体系出错: {str(e)}")
|
|
|
print("问题数据:", relation_system)
|
|
|
raise
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
try:
|
|
|
with open("../小学数学学科素养点体系.json", 'r', encoding='utf-8') as f:
|
|
|
raw_data = json.load(f)
|
|
|
print("JSON文件加载成功，顶层键:", raw_data.keys())
|
|
|
|
|
|
if "data" not in raw_data:
|
|
|
raise ValueError("JSON缺少'data'字段")
|
|
|
|
|
|
loader = Py2neoLoader()
|
|
|
loader.create_literacy_nodes(raw_data["data"])
|
|
|
except Exception as e:
|
|
|
print("主程序错误:", str(e)) |
