Python 入门指南——打造 Neo4j 图数据查询公共类

后端

2023-03-27 14:47:26

Neo4j 查询公共类：简化图数据查询

图数据处理项目中，大量查询操作是不可避免的。为了提升开发效率，创建 Python 公共类封装查询十分必要。本文将逐步指导您构建这样一个公共类，涵盖连接、断开连接、执行查询、获取单行和所有结果等方法，并提供代码示例和使用说明。

1. Neo4j 查询语法简介

Neo4j 使用专为图数据查询设计的 Cypher 查询语言。其语法主要包括：

匹配：匹配图中的节点和关系。
返回：指定查询结果。
过滤：筛选查询结果。
排序：对查询结果排序。
限制：限制查询结果数量。

2. Python Neo4j 查询公共类设计

我们的公共类将包含以下方法：

connect()：连接到 Neo4j 数据库。
disconnect()：断开与 Neo4j 数据库的连接。
execute()：执行 Cypher 查询并返回结果。
fetch_one()：获取查询结果的第一行数据。
fetch_all()：获取查询结果的所有数据。

3. Python Neo4j 查询公共类实现

import neo4j

class Neo4jConnector:
    def __init__(self, uri, user, password):
        self.uri = uri
        self.user = user
        self.password = password
        self.driver = None

    def connect(self):
        self.driver = neo4j.GraphDatabase.driver(self.uri, auth=(self.user, self.password))

    def disconnect(self):
        if self.driver:
            self.driver.close()

    def execute(self, query, parameters=None):
        with self.driver.session() as session:
            result = session.run(query, parameters=parameters)
            return result

    def fetch_one(self, query, parameters=None):
        result = self.execute(query, parameters)
        return result.single()

    def fetch_all(self, query, parameters=None):
        result = self.execute(query, parameters)
        return result.data()

4. Python Neo4j 查询公共类使用示例

假设我们有一个名为 Movie 的图数据库，其中包含电影节点和演员节点。要查询 2000 年上映电影的所有主演演员，可以使用以下代码：

connector = Neo4jConnector("bolt://localhost:7687", "neo4j", "password")
connector.connect()

query = """
MATCH (movie:Movie)<-[:ACTED_IN]-(actor:Actor)
WHERE movie.year = 2000
RETURN actor.name
"""

result = connector.fetch_all(query)
for actor in result:
    print(actor["name"])

connector.disconnect()