Class Neo4jDriverWrapper

src/codegraphcontext/core/database.py:32–48

A simple wrapper around the Neo4j Driver to inject a database name into session() calls.

        return f"{base}\nSuggested fixes:{suggestion_block}"

class Neo4jDriverWrapper:
    """
    A simple wrapper around the Neo4j Driver to inject a database name into session() calls.
    """
    def __init__(self, driver: Driver, database: str = None):
        self._driver = driver
        self._database = database

    def session(self, **kwargs):
        """Proxy method to get a session from the underlying driver."""
        if self._database and 'database' not in kwargs:
            kwargs["database"] = self._database
        return self._driver.session(**kwargs)
    
    def close(self):
        """Proxy method to close the underlying driver."""
        self._driver.close()

class DatabaseManager:
    """