|
|
"""Neo4j client for graph database operations.""" |
|
|
|
|
|
from neo4j import AsyncGraphDatabase |
|
|
|
|
|
from app.core.config import settings |
|
|
|
|
|
|
|
|
class Neo4jClient: |
|
|
"""Async Neo4j client for spatial and graph queries.""" |
|
|
|
|
|
def __init__(self, uri: str, user: str, password: str): |
|
|
"""Initialize Neo4j driver.""" |
|
|
self._driver = AsyncGraphDatabase.driver(uri, auth=(user, password)) |
|
|
|
|
|
async def close(self) -> None: |
|
|
"""Close the driver connection.""" |
|
|
await self._driver.close() |
|
|
|
|
|
async def run_cypher( |
|
|
self, |
|
|
query: str, |
|
|
params: dict | None = None, |
|
|
) -> list[dict]: |
|
|
""" |
|
|
Execute a Cypher query and return results. |
|
|
|
|
|
Args: |
|
|
query: Cypher query string |
|
|
params: Optional query parameters |
|
|
|
|
|
Returns: |
|
|
List of result records as dictionaries |
|
|
""" |
|
|
async with self._driver.session() as session: |
|
|
result = await session.run(query, params or {}) |
|
|
return await result.data() |
|
|
|
|
|
async def verify_connectivity(self) -> bool: |
|
|
"""Verify connection to Neo4j.""" |
|
|
try: |
|
|
await self._driver.verify_connectivity() |
|
|
return True |
|
|
except Exception: |
|
|
return False |
|
|
|
|
|
|
|
|
|
|
|
neo4j_client = Neo4jClient( |
|
|
settings.neo4j_uri, |
|
|
settings.neo4j_username, |
|
|
settings.neo4j_password, |
|
|
) |
|
|
|
