25 lines
873 B
Python
25 lines
873 B
Python
"""Lazy Neo4j adapter used only by controlled relationship services."""
|
|
|
|
from collections.abc import Sequence
|
|
from typing import Any
|
|
|
|
from neo4j import AsyncGraphDatabase
|
|
|
|
from app.core.config import Settings
|
|
|
|
|
|
class Neo4jGraphDriver:
|
|
def __init__(self, settings: Settings) -> None:
|
|
self.settings = settings
|
|
self._driver: Any = None
|
|
|
|
async def execute_query(self, query: str, **parameters: Any) -> Sequence[Any]:
|
|
if self._driver is None:
|
|
self._driver = AsyncGraphDatabase.driver(
|
|
self.settings.neo4j_uri,
|
|
auth=(self.settings.neo4j_username, self.settings.neo4j_password),
|
|
)
|
|
async with self._driver.session(database=self.settings.neo4j_database) as session:
|
|
result = await session.run(query, parameters)
|
|
return [record.data() async for record in result]
|