# -*- coding: utf-8 -*-
|
# neo4j.py
|
# @author: lyg
|
# @date: 2025-04-24
|
# @version: 0.0.1
|
# @description: neo4j数据库操作
|
|
from py2neo import Graph, Node, Relationship
|
import json
|
|
|
class Neo4jHelper:
|
def __init__(self):
|
self.graph = Graph('http://192.168.1.14:7474', user='neo4j', password='123456')
|
|
def create_page_node(self, page_info):
|
"""
|
创建页面节点
|
"""
|
# 创建节点
|
node = Node("Page", page_num=page_info.page_num,
|
chapter_info=json.dumps(page_info.chapter_info, ensure_ascii=False, indent=2),
|
text=page_info.text,
|
entities=json.dumps(page_info.entities, ensure_ascii=False, indent=2))
|
self.graph.create(node)
|
return node
|
def create_trunk_node(self, trunk, entities):
|
"""
|
创建页面节点
|
"""
|
# 创建节点
|
node = Node("Trunk", trunk=trunk,
|
entities=json.dumps(entities, ensure_ascii=False, indent=2))
|
self.graph.create(node)
|
return node
|
def create_entity_node(self, entity: str):
|
"""
|
创建实体节点
|
"""
|
node = self.graph.nodes.match("Entity", text=entity).first()
|
if node:
|
return node
|
node = Node("Entity", text=entity)
|
self.graph.create(node)
|
return node
|
|
|
def create_trunk_entity_relationship(self, page_node, entity_node):
|
"""
|
创建页面和实体节点的关系
|
"""
|
relationship = Relationship(page_node, "trunk_entity", entity_node)
|
self.graph.create(relationship)
|
return relationship
|
|
def create_page_entity_relationship(self, page_node, entity_node):
|
"""
|
创建页面和实体节点的关系
|
"""
|
relationship = Relationship(page_node, "page_entity", entity_node)
|
self.graph.create(relationship)
|
return relationship
|
|
def create_entity_relationship(self, entity_node1, entity_node2):
|
"""
|
创建实体节点的关系
|
"""
|
if entity_node1 == entity_node2:
|
return
|
relationship = Relationship(entity_node1, "entity_entity", entity_node2)
|
if self.graph.exists(relationship):
|
return
|
self.graph.create(relationship)
|
return relationship
|
