链接模型的首选方式

from neo4j import GraphDatabase

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

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

    def create_node_and_relationship(self, node1_label, node1_properties, node2_label, node2_properties, relationship_type):
        with self._driver.session() as session:
            result = session.write_transaction(self._create_and_link, node1_label, node1_properties, node2_label, node2_properties, relationship_type)
            print(result)

    @staticmethod
    def _create_and_link(tx, node1_label, node1_properties, node2_label, node2_properties, relationship_type):
        query = (
            f"CREATE (a:{node1_label} $props1) "
            f"CREATE (b:{node2_label} $props2) "
            f"CREATE (a)-[:{relationship_type}]->(b) "
            "RETURN a, b"
        )
        result = tx.run(query, props1=node1_properties, props2=node2_properties)
        record = result.single()
        return record["a"]._properties, record["b"]._properties

# 示例用法
uri = "bolt://localhost:7687"
user = "neo4j"
password = "password"

conn = Neo4jConnection(uri, user, password)
conn.create_node_and_relationship("Person", {"name": "Alice"}, "Company", {"name": "Tech Corp"}, "WORKS_FOR")
conn.close()