sqlspec 0.47.0__cp314-cp314-win_amd64.whl

sqlspec/adapters/adbc/__init__.py

@@ -0,0 +1,13 @@
+from sqlspec.adapters.adbc._typing import AdbcConnection
+from sqlspec.adapters.adbc.config import AdbcConfig, AdbcConnectionParams, AdbcDriverFeatures
+from sqlspec.adapters.adbc.driver import AdbcCursor, AdbcDriver, AdbcExceptionHandler
+
+__all__ = (
+    "AdbcConfig",
+    "AdbcConnection",
+    "AdbcConnectionParams",
+    "AdbcCursor",
+    "AdbcDriver",
+    "AdbcDriverFeatures",
+    "AdbcExceptionHandler",
+)

sqlspec/adapters/adbc/_typing.py

@@ -0,0 +1,106 @@
+# pyright: reportCallIssue=false, reportAttributeAccessIssue=false, reportArgumentType=false
+"""ADBC adapter type definitions.
+
+This module contains type aliases and classes that are excluded from mypyc
+compilation to avoid ABI boundary issues.
+"""
+
+import contextlib
+from typing import TYPE_CHECKING, Any
+
+from adbc_driver_manager.dbapi import Connection
+from adbc_driver_manager.dbapi import Cursor as _AdbcRawCursor
+
+_AdbcConnection = Connection
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+    from types import TracebackType
+    from typing import TypeAlias
+
+    from sqlspec.adapters.adbc.driver import AdbcDriver
+    from sqlspec.core import StatementConfig
+
+    AdbcConnection: TypeAlias = _AdbcConnection
+    AdbcRawCursor: TypeAlias = _AdbcRawCursor
+
+if not TYPE_CHECKING:
+    AdbcConnection = _AdbcConnection
+    AdbcRawCursor = _AdbcRawCursor
+
+
+class AdbcCursor:
+    """Context manager for cursor management."""
+
+    __slots__ = ("connection", "cursor")
+
+    def __init__(self, connection: "AdbcConnection") -> None:
+        self.connection = connection
+        self.cursor: AdbcRawCursor | None = None
+
+    def __enter__(self) -> "AdbcRawCursor":
+        self.cursor = self.connection.cursor()
+        return self.cursor
+
+    def __exit__(self, *_: Any) -> None:
+        if self.cursor is not None:
+            with contextlib.suppress(Exception):
+                self.cursor.close()
+
+
+class AdbcSessionContext:
+    """Sync context manager for ADBC sessions.
+
+    This class is intentionally excluded from mypyc compilation to avoid ABI
+    boundary issues. It receives callables from uncompiled config classes and
+    instantiates compiled Driver objects, acting as a bridge between compiled
+    and uncompiled code.
+
+    Uses callable-based connection management to decouple from config implementation.
+    """
+
+    __slots__ = (
+        "_acquire_connection",
+        "_connection",
+        "_driver",
+        "_driver_features",
+        "_prepare_driver",
+        "_release_connection",
+        "_statement_config",
+    )
+
+    def __init__(
+        self,
+        acquire_connection: "Callable[[], Any]",
+        release_connection: "Callable[[Any], Any]",
+        statement_config: "StatementConfig",
+        driver_features: "dict[str, Any]",
+        prepare_driver: "Callable[[AdbcDriver], AdbcDriver]",
+    ) -> None:
+        self._acquire_connection = acquire_connection
+        self._release_connection = release_connection
+        self._statement_config = statement_config
+        self._driver_features = driver_features
+        self._prepare_driver = prepare_driver
+        self._connection: Any = None
+        self._driver: AdbcDriver | None = None
+
+    def __enter__(self) -> "AdbcDriver":
+        from sqlspec.adapters.adbc.driver import AdbcDriver
+
+        self._connection = self._acquire_connection()
+        self._driver = AdbcDriver(
+            connection=self._connection, statement_config=self._statement_config, driver_features=self._driver_features
+        )
+        return self._prepare_driver(self._driver)
+
+    def __exit__(
+        self, exc_type: "type[BaseException] | None", exc_val: "BaseException | None", exc_tb: "TracebackType | None"
+    ) -> "bool | None":
+        if self._connection is not None:
+            self._release_connection(self._connection)
+            self._connection = None
+        return None
+
+
+__all__ = ("AdbcConnection", "AdbcCursor", "AdbcRawCursor", "AdbcSessionContext")

sqlspec/adapters/adbc/adk/__init__.py

@@ -0,0 +1,5 @@
+"""ADBC ADK integration for Google Agent Development Kit."""
+
+from sqlspec.adapters.adbc.adk.store import AdbcADKMemoryStore, AdbcADKStore
+
+__all__ = ("AdbcADKMemoryStore", "AdbcADKStore")