存储图的对象

from neo4j import GraphDatabase

class GraphDB:
    def __init__(self, uri, user, password):
        self._driver = GraphDatabase.driver(uri, auth=(user, password))

    def close(self):
        self._driver.close()

    def add_node(self, label, properties):
        with self._driver.session() as session:
            session.run(f"CREATE (n:{label} $props)", props=properties)

    def add_edge(self, start_node_id, end_node_id, relationship_type, properties=None):
        with self._driver.session() as session:
            query = f"MATCH (a), (b) WHERE id(a) = $start_node_id AND id(b) = $end_node_id CREATE (a)-[r:{relationship_type}]->(b)"
            params = {"start_node_id": start_node_id, "end_node_id": end_node_id}
            if properties:
                query += " SET r += $props"
                params["props"] = properties
            session.run(query, **params)

# 示例使用
db = GraphDB("bolt://localhost:7687", "neo4j", "password")
db.add_node("Person", {"name": "Alice"})
db.add_node("Person", {"name": "Bob"})
db.add_edge(0, 1, "KNOWS", {"since": 2020})
db.close()