#创建数据库连接
def link_Graph():
# Neo4j服务器地址、端口、用户名和密码
uri = "bolt://localhost:7687" # 使用bolt协议，默认端口是7687
user = "" # 用户名，默认是neo4j
password = "" # 替换为你的密码
# 创建驱动程序实例
driver = GraphDatabase.driver(uri, auth=(user, password))
return driver

#读取实体数据
def read_json_file(json_path):
# 打开并读取JSON文件
with open(json_path, 'r', encoding='utf-8') as file:
datas = json.load(file)
return datas

#读取实体数据
def read_json_file(json_path):
# 打开并读取JSON文件
with open(json_path, 'r', encoding='utf-8') as file:
datas = json.load(file)
return datas

#导入实体以及属性
def create_enity_with_properties(graph_db, label, properties):
with graph_db.session() as session:
try:
if label == 'Disease':
# 构建Cypher查询（使用参数）
cypher = "CREATE (n:{label} $props) RETURN n".format(label=label)
# 执行Cypher查询并传递属性作为参数
result = session.run(cypher, props=properties)
return 1
else:
cypher = "CREATE (n:{} {{name: $name}})".format(label)
result = session.run(cypher, name=properties)
return 1
except Exception as e:
return 0

#执行程序
if __name__=='__main__':
entity_json_path = "D:\知识图谱数据汇总\DiseaseKG基于cnSchma常见疾病信息知识图谱\diseasekg
\entities.json"
relationship_json_path="D:\知识图谱数据汇总\DiseaseKG基于cnSchma常见疾病信息知识图谱\diseasekg
\\relations.json"
graph_db = link_Graph()
datas=read_json_file(relationship_json_path)#读取数据，请在导入实体时选择entity_json_path参数;
导入关系时选择relationship_json_path参数
error_number=flag=index=0
error_data=[]
for data in datas:
# flag = create_enity_with_properties(graph_db, data['label'], data['name'])#导入实体，请
在进行关系导入时注释这一行
flag=add_relationship_between_nodes(graph_db,data)#关系导入，请在进行实体导入时注释这一行
if flag==1:
print('成功执行！！！')
else:
print('数据错误！！！')

3. 导入关系数据

#导入实体间关系
def add_relationship_between_nodes(driver,dict_relationship):
# 从字典中提取属性
start_entity_type=dict_relationship['start_entity_type']
end_entity_type=dict_relationship['end_entity_type']
relationship_type = dict_relationship['rel_type']
relationship_name = dict_relationship['rel_name']
all_entity_pairs=dict_relationship['rels']
try:
with driver.session() as session:
for entity_pair in all_entity_pairs:
start_entity_name = entity_pair['start_entity_name']
end_entity_name = entity_pair['end_entity_name']
# 构建Cypher查询，使用参数来引用节点和关系类型
# cypher = "MATCH (startNode:{start_label} {id: $startEntityId}), (endNode:{end_
label} {id:$endEntityId}) CREATE (startNode)-[:relationship]->(endNode) RETURN
startNode".replace('relationship',relationship_type).format(start_label=start_
entity_type,end_label=end_entity_type)
cypher = "MATCH (s:{} {{name: $startNodeName}}), (e:{} {{name: $endNodeName}})
CREATE (s)-[r:relationship {{name: $relName}}]->(e)".replace(
'relationship', relationship_type).format(start_entity_type, end_entity_type)
result = session.run(cypher, {"startNodeName": start_entity_name, "endNodeName":
end_entity_name,
'relName': relationship_name})
return 1
except Exception as e:
return 0
注：在导入关系时仅需要按照main函数中的注释要求即可

4. 数据可视化