sqlspec 0.63.1__py3-none-any.whl → 0.65.0__py3-none-any.whl

sqlspec/__init__.py

@@ -13,79 +13,98 @@ _warnings.filterwarnings(
 )
 del _warnings
 
+from importlib import import_module
+from importlib.machinery import EXTENSION_SUFFIXES
 from typing import TYPE_CHECKING, Any
 
-from sqlspec import adapters, base, builder, core, driver, exceptions, extensions, loader, migrations, typing, utils
 from sqlspec.__metadata__ import __version__
-from sqlspec.base import SQLSpec
-from sqlspec.builder import (
-    Column,
-    ColumnExpression,
-    CreateTable,
-    Delete,
-    DropTable,
-    FunctionColumn,
-    Insert,
-    Merge,
-    QueryBuilder,
-    Select,
-    SQLFactory,
-    Update,
-    sql,
-)
-from sqlspec.config import AsyncDatabaseConfig, SyncDatabaseConfig
-from sqlspec.core import (
-    SQL,
-    ArrowResult,
-    CacheConfig,
-    CacheStats,
-    ParameterConverter,
-    ParameterDeclaration,
-    ParameterProcessor,
-    ParameterStyle,
-    ParameterStyleConfig,
-    ParamTypeMatcher,
-    ProcessedState,
-    SQLResult,
-    StackOperation,
-    StackResult,
-    Statement,
-    StatementConfig,
-    StatementStack,
-    matches_param_type,
-    register_param_type,
-    resolve_param_type,
-)
-from sqlspec.core import filters as filters
-from sqlspec.driver import AsyncDriverAdapterBase, ExecutionResult, SyncDriverAdapterBase
-from sqlspec.exceptions import StackExecutionError
-from sqlspec.extensions.events import (
-    AsyncEventChannel,
-    AsyncEventListener,
-    EventMessage,
-    SyncEventChannel,
-    SyncEventListener,
-)
-from sqlspec.loader import SQLFile, SQLFileLoader
-from sqlspec.observability import (
-    LoggingConfig,
-    ObservabilityConfig,
-    ObservabilityRuntime,
-    RedactionConfig,
-    StatementEvent,
-    StatementObserver,
-    TelemetryConfig,
-    create_event,
-    create_statement_observer,
-    default_statement_observer,
-    format_statement_event,
-)
-from sqlspec.typing import ConnectionT, PoolT, SchemaT, StatementParameters, SupportedSchemaModel
+from sqlspec.utils import logging as _logging
 from sqlspec.utils.logging import suppress_erroneous_sqlglot_log_messages
-from sqlspec.utils.uuids import nanoid, uuid4, uuid6, uuid7
+
+_COMPILED = (_logging.__file__ or "").endswith(tuple(EXTENSION_SUFFIXES))
 
 if TYPE_CHECKING:
-    from sqlspec import dialects
+    from sqlspec import (
+        adapters,
+        base,
+        builder,
+        core,
+        dialects,
+        driver,
+        exceptions,
+        extensions,
+        loader,
+        migrations,
+        typing,
+        utils,
+    )
+    from sqlspec import config as config
+    from sqlspec import observability as observability
+    from sqlspec.base import SQLSpec
+    from sqlspec.builder import (
+        Column,
+        ColumnExpression,
+        CreateTable,
+        Delete,
+        DropTable,
+        FunctionColumn,
+        Insert,
+        Merge,
+        QueryBuilder,
+        Select,
+        SQLFactory,
+        Update,
+        sql,
+    )
+    from sqlspec.config import AsyncDatabaseConfig, SyncDatabaseConfig
+    from sqlspec.core import (
+        SQL,
+        ArrowResult,
+        CacheConfig,
+        CacheStats,
+        ParameterConverter,
+        ParameterDeclaration,
+        ParameterProcessor,
+        ParameterStyle,
+        ParameterStyleConfig,
+        ParamTypeMatcher,
+        ProcessedState,
+        SQLResult,
+        StackOperation,
+        StackResult,
+        Statement,
+        StatementConfig,
+        StatementStack,
+        matches_param_type,
+        register_param_type,
+        resolve_param_type,
+    )
+    from sqlspec.core import filters as filters
+    from sqlspec.driver import AsyncDriverAdapterBase, ExecutionResult, SyncDriverAdapterBase
+    from sqlspec.exceptions import StackExecutionError
+    from sqlspec.extensions.events import (
+        AsyncEventChannel,
+        AsyncEventListener,
+        EventMessage,
+        SyncEventChannel,
+        SyncEventListener,
+    )
+    from sqlspec.loader import SQLFile, SQLFileLoader
+    from sqlspec.observability import (
+        LoggingConfig,
+        ObservabilityConfig,
+        ObservabilityRuntime,
+        RedactionConfig,
+        StatementEvent,
+        StatementObserver,
+        TelemetryConfig,
+        create_event,
+        create_statement_observer,
+        default_statement_observer,
+        format_statement_event,
+    )
+    from sqlspec.typing import ConnectionT, PoolT, SchemaT, StatementParameters, SupportedSchemaModel
+    from sqlspec.utils.uuids import nanoid, uuid4, uuid6, uuid7
 
 __all__ = (
     "SQL",
@@ -174,10 +193,111 @@ __all__ = (
 suppress_erroneous_sqlglot_log_messages()
 
 
-def __getattr__(name: str) -> "Any":
-    if name == "dialects":
-        import importlib
+_EXPORTS: dict[str, tuple[str, str | None]] = {
+    "ArrowResult": ("sqlspec.core", "ArrowResult"),
+    "AsyncDatabaseConfig": ("sqlspec.config", "AsyncDatabaseConfig"),
+    "AsyncDriverAdapterBase": ("sqlspec.driver", "AsyncDriverAdapterBase"),
+    "AsyncEventChannel": ("sqlspec.extensions.events", "AsyncEventChannel"),
+    "AsyncEventListener": ("sqlspec.extensions.events", "AsyncEventListener"),
+    "CacheConfig": ("sqlspec.core", "CacheConfig"),
+    "CacheStats": ("sqlspec.core", "CacheStats"),
+    "Column": ("sqlspec.builder", "Column"),
+    "ColumnExpression": ("sqlspec.builder", "ColumnExpression"),
+    "ConnectionT": ("sqlspec.typing", "ConnectionT"),
+    "CreateTable": ("sqlspec.builder", "CreateTable"),
+    "Delete": ("sqlspec.builder", "Delete"),
+    "DropTable": ("sqlspec.builder", "DropTable"),
+    "EventMessage": ("sqlspec.extensions.events", "EventMessage"),
+    "ExecutionResult": ("sqlspec.driver", "ExecutionResult"),
+    "FunctionColumn": ("sqlspec.builder", "FunctionColumn"),
+    "Insert": ("sqlspec.builder", "Insert"),
+    "LoggingConfig": ("sqlspec.observability", "LoggingConfig"),
+    "Merge": ("sqlspec.builder", "Merge"),
+    "ObservabilityConfig": ("sqlspec.observability", "ObservabilityConfig"),
+    "ObservabilityRuntime": ("sqlspec.observability", "ObservabilityRuntime"),
+    "ParamTypeMatcher": ("sqlspec.core", "ParamTypeMatcher"),
+    "ParameterConverter": ("sqlspec.core", "ParameterConverter"),
+    "ParameterDeclaration": ("sqlspec.core", "ParameterDeclaration"),
+    "ParameterProcessor": ("sqlspec.core", "ParameterProcessor"),
+    "ParameterStyle": ("sqlspec.core", "ParameterStyle"),
+    "ParameterStyleConfig": ("sqlspec.core", "ParameterStyleConfig"),
+    "PoolT": ("sqlspec.typing", "PoolT"),
+    "ProcessedState": ("sqlspec.core", "ProcessedState"),
+    "QueryBuilder": ("sqlspec.builder", "QueryBuilder"),
+    "RedactionConfig": ("sqlspec.observability", "RedactionConfig"),
+    "SQL": ("sqlspec.core", "SQL"),
+    "SQLFactory": ("sqlspec.builder", "SQLFactory"),
+    "SQLFile": ("sqlspec.loader", "SQLFile"),
+    "SQLFileLoader": ("sqlspec.loader", "SQLFileLoader"),
+    "SQLResult": ("sqlspec.core", "SQLResult"),
+    "SQLSpec": ("sqlspec.base", "SQLSpec"),
+    "SchemaT": ("sqlspec.typing", "SchemaT"),
+    "Select": ("sqlspec.builder", "Select"),
+    "StackExecutionError": ("sqlspec.exceptions", "StackExecutionError"),
+    "StackOperation": ("sqlspec.core", "StackOperation"),
+    "StackResult": ("sqlspec.core", "StackResult"),
+    "Statement": ("sqlspec.core", "Statement"),
+    "StatementConfig": ("sqlspec.core", "StatementConfig"),
+    "StatementEvent": ("sqlspec.observability", "StatementEvent"),
+    "StatementObserver": ("sqlspec.observability", "StatementObserver"),
+    "StatementParameters": ("sqlspec.typing", "StatementParameters"),
+    "StatementStack": ("sqlspec.core", "StatementStack"),
+    "SupportedSchemaModel": ("sqlspec.typing", "SupportedSchemaModel"),
+    "SyncDatabaseConfig": ("sqlspec.config", "SyncDatabaseConfig"),
+    "SyncDriverAdapterBase": ("sqlspec.driver", "SyncDriverAdapterBase"),
+    "SyncEventChannel": ("sqlspec.extensions.events", "SyncEventChannel"),
+    "SyncEventListener": ("sqlspec.extensions.events", "SyncEventListener"),
+    "TelemetryConfig": ("sqlspec.observability", "TelemetryConfig"),
+    "Update": ("sqlspec.builder", "Update"),
+    "adapters": ("sqlspec.adapters", None),
+    "base": ("sqlspec.base", None),
+    "builder": ("sqlspec.builder", None),
+    "config": ("sqlspec.config", None),
+    "core": ("sqlspec.core", None),
+    "create_event": ("sqlspec.observability", "create_event"),
+    "create_statement_observer": ("sqlspec.observability", "create_statement_observer"),
+    "default_statement_observer": ("sqlspec.observability", "default_statement_observer"),
+    "dialects": ("sqlspec.dialects", None),
+    "driver": ("sqlspec.driver", None),
+    "exceptions": ("sqlspec.exceptions", None),
+    "extensions": ("sqlspec.extensions", None),
+    "filters": ("sqlspec.core", "filters"),
+    "format_statement_event": ("sqlspec.observability", "format_statement_event"),
+    "loader": ("sqlspec.loader", None),
+    "matches_param_type": ("sqlspec.core", "matches_param_type"),
+    "migrations": ("sqlspec.migrations", None),
+    "nanoid": ("sqlspec.utils.uuids", "nanoid"),
+    "observability": ("sqlspec.observability", None),
+    "register_param_type": ("sqlspec.core", "register_param_type"),
+    "resolve_param_type": ("sqlspec.core", "resolve_param_type"),
+    "sql": ("sqlspec.builder._factory", "sql"),
+    "typing": ("sqlspec.typing", None),
+    "utils": ("sqlspec.utils", None),
+    "uuid4": ("sqlspec.utils.uuids", "uuid4"),
+    "uuid6": ("sqlspec.utils.uuids", "uuid6"),
+    "uuid7": ("sqlspec.utils.uuids", "uuid7"),
+}
+
+
+def __getattr__(name: str) -> Any:
+    target = _EXPORTS.get(name)
+    if target is None:
+        msg = f"module {__name__!r} has no attribute {name!r}"
+        raise AttributeError(msg)
+    module_name, attribute = target
+    module = import_module(module_name)
+    value = module if attribute is None else getattr(module, attribute)
+    globals()[name] = value
+    return value
+
+
+def __dir__() -> list[str]:
+    return sorted(set(globals()) | _EXPORTS.keys())
+
 
-        return importlib.import_module("sqlspec.dialects")
-    msg = f"module {__name__!r} has no attribute {name!r}"
-    raise AttributeError(msg)
+# Native cross-module initialization is not protected by Python import locks.
+# Retain eager exports in compiled wheels until mypyc supports lazy concurrency.
+if _COMPILED:
+    for _name in __all__:
+        if _name not in globals():
+            __getattr__(_name)

sqlspec/_typing.py

@@ -50,6 +50,7 @@ __all__ = (
     "CATTRS_INSTALLED",
     "CLOUD_SQL_CONNECTOR_INSTALLED",
     "FSSPEC_INSTALLED",
+    "IBM_DB_INSTALLED",
     "LITESTAR_INSTALLED",
     "MSGSPEC_INSTALLED",
     "NANOID_INSTALLED",
@@ -336,10 +337,6 @@ def attrs_define_stub(*args: Any, **kwargs: Any) -> Any:  # noqa: ARG001
     return _attrs_define_identity
 
 
-def _attrs_define_identity(cls: Any) -> Any:
-    return cls
-
-
 def attrs_field_stub(*args: Any, **kwargs: Any) -> Any:  # noqa: ARG001
     """Placeholder implementation"""
     return None
@@ -670,6 +667,7 @@ ATTRS_INSTALLED = dependency_flag("attrs")
 CATTRS_INSTALLED = dependency_flag("cattrs")
 CLOUD_SQL_CONNECTOR_INSTALLED = dependency_flag("google.cloud.sql.connector")
 FSSPEC_INSTALLED = dependency_flag("fsspec")
+IBM_DB_INSTALLED = dependency_flag("ibm_db")
 LITESTAR_INSTALLED = dependency_flag("litestar")
 MSGSPEC_INSTALLED = dependency_flag("msgspec")
 NUMPY_INSTALLED = dependency_flag("numpy")
@@ -751,3 +749,7 @@ def __dir__() -> "list[str]":
     """Expose the public surface for autocomplete and ``dir()``."""
 
     return sorted(set(globals()) | set(__all__))
+
+
+def _attrs_define_identity(cls: Any) -> Any:
+    return cls

sqlspec/adapters/adbc/_typing.py

@@ -9,9 +9,13 @@ import contextlib
 from typing import TYPE_CHECKING, Any
 
 from adbc_driver_manager import Error as _AdbcNativeError
+from adbc_driver_manager import NotSupportedError as AdbcNotSupportedError
+from adbc_driver_manager import OperationalError as AdbcOperationalError
 from adbc_driver_manager.dbapi import Connection
 from adbc_driver_manager.dbapi import Cursor as _AdbcRawCursor
 
+from sqlspec.typing import import_optional_attr
+
 _AdbcConnection = Connection
 
 if TYPE_CHECKING:
@@ -19,19 +23,29 @@ if TYPE_CHECKING:
     from types import TracebackType
     from typing import TypeAlias
 
+    from adbc_driver_flightsql import DatabaseOptions as AdbcFlightSqlDatabaseOptions
+
     from sqlspec.adapters.adbc.driver import AdbcDriver
     from sqlspec.core import StatementConfig
 
     AdbcConnection: TypeAlias = _AdbcConnection
     AdbcRawCursor: TypeAlias = _AdbcRawCursor
     AdbcNativeError: TypeAlias = _AdbcNativeError
-
-if not TYPE_CHECKING:
+else:
     AdbcConnection = _AdbcConnection
     AdbcRawCursor = _AdbcRawCursor
     AdbcNativeError = _AdbcNativeError
 
-__all__ = ("AdbcConnection", "AdbcCursor", "AdbcNativeError", "AdbcRawCursor", "AdbcSessionContext")
+__all__ = (
+    "AdbcConnection",
+    "AdbcCursor",
+    "AdbcFlightSqlDatabaseOptions",
+    "AdbcNativeError",
+    "AdbcNotSupportedError",
+    "AdbcOperationalError",
+    "AdbcRawCursor",
+    "AdbcSessionContext",
+)
 
 
 class AdbcCursor:
@@ -77,7 +91,7 @@ class AdbcSessionContext:
     def __init__(
         self,
         acquire_connection: "Callable[[], Any]",
-        release_connection: "Callable[[Any], Any]",
+        release_connection: "Callable[..., Any]",
         statement_config: "StatementConfig",
         driver_features: "dict[str, Any]",
         prepare_driver: "Callable[[AdbcDriver], AdbcDriver]",
@@ -108,6 +122,18 @@ class AdbcSessionContext:
         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._release_connection(self._connection, exc_type=exc_type, exc_val=exc_val, exc_tb=exc_tb)
             self._connection = None
         return None
+
+
+def __getattr__(name: str) -> Any:
+    """Load Flight SQL options only when that optional driver is used."""
+    if name != "AdbcFlightSqlDatabaseOptions":
+        msg = f"module {__name__!r} has no attribute {name!r}"
+        raise AttributeError(msg)
+    value = import_optional_attr("adbc_driver_flightsql", "DatabaseOptions")
+    if value is None:
+        msg = "Cannot import DatabaseOptions from adbc_driver_flightsql"
+        raise ImportError(msg)
+    return value