sqlspec 0.17.0__py3-none-any.whl → 0.18.0__py3-none-any.whl

sqlspec/migrations/tracker.py

@@ -7,15 +7,18 @@ import os
 from typing import TYPE_CHECKING, Any, Optional
 
 from sqlspec.migrations.base import BaseMigrationTracker
+from sqlspec.utils.logging import get_logger
 
 if TYPE_CHECKING:
     from sqlspec.driver import AsyncDriverAdapterBase, SyncDriverAdapterBase
 
 __all__ = ("AsyncMigrationTracker", "SyncMigrationTracker")
 
+logger = get_logger("migrations.tracker")
+
 
 class SyncMigrationTracker(BaseMigrationTracker["SyncDriverAdapterBase"]):
-    """Tracks applied migrations in the database."""
+    """Synchronous migration version tracker."""
 
     def ensure_tracking_table(self, driver: "SyncDriverAdapterBase") -> None:
         """Create the migration tracking table if it doesn't exist.
@@ -24,6 +27,7 @@ class SyncMigrationTracker(BaseMigrationTracker["SyncDriverAdapterBase"]):
             driver: The database driver to use.
         """
         driver.execute(self._get_create_table_sql())
+        self._safe_commit(driver)
 
     def get_current_version(self, driver: "SyncDriverAdapterBase") -> Optional[str]:
         """Get the latest applied migration version.
@@ -66,6 +70,7 @@ class SyncMigrationTracker(BaseMigrationTracker["SyncDriverAdapterBase"]):
                 version, description, execution_time_ms, checksum, os.environ.get("USER", "unknown")
             )
         )
+        self._safe_commit(driver)
 
     def remove_migration(self, driver: "SyncDriverAdapterBase", version: str) -> None:
         """Remove a migration record (used during downgrade).
@@ -75,10 +80,30 @@ class SyncMigrationTracker(BaseMigrationTracker["SyncDriverAdapterBase"]):
             version: Version number to remove.
         """
         driver.execute(self._get_remove_migration_sql(version))
+        self._safe_commit(driver)
+
+    def _safe_commit(self, driver: "SyncDriverAdapterBase") -> None:
+        """Safely commit a transaction only if autocommit is disabled.
+
+        Args:
+            driver: The database driver to use.
+        """
+        try:
+            connection = getattr(driver, "connection", None)
+            if connection and hasattr(connection, "autocommit") and getattr(connection, "autocommit", False):
+                return
+
+            driver_features = getattr(driver, "driver_features", {})
+            if driver_features and driver_features.get("autocommit", False):
+                return
+
+            driver.commit()
+        except Exception:
+            logger.debug("Failed to commit transaction, likely due to autocommit being enabled")
 
 
 class AsyncMigrationTracker(BaseMigrationTracker["AsyncDriverAdapterBase"]):
-    """Tracks applied migrations in the database."""
+    """Asynchronous migration version tracker."""
 
     async def ensure_tracking_table(self, driver: "AsyncDriverAdapterBase") -> None:
         """Create the migration tracking table if it doesn't exist.
@@ -87,6 +112,7 @@ class AsyncMigrationTracker(BaseMigrationTracker["AsyncDriverAdapterBase"]):
             driver: The database driver to use.
         """
         await driver.execute(self._get_create_table_sql())
+        await self._safe_commit_async(driver)
 
     async def get_current_version(self, driver: "AsyncDriverAdapterBase") -> Optional[str]:
         """Get the latest applied migration version.
@@ -129,6 +155,7 @@ class AsyncMigrationTracker(BaseMigrationTracker["AsyncDriverAdapterBase"]):
                 version, description, execution_time_ms, checksum, os.environ.get("USER", "unknown")
             )
         )
+        await self._safe_commit_async(driver)
 
     async def remove_migration(self, driver: "AsyncDriverAdapterBase", version: str) -> None:
         """Remove a migration record (used during downgrade).
@@ -138,3 +165,23 @@ class AsyncMigrationTracker(BaseMigrationTracker["AsyncDriverAdapterBase"]):
             version: Version number to remove.
         """
         await driver.execute(self._get_remove_migration_sql(version))
+        await self._safe_commit_async(driver)
+
+    async def _safe_commit_async(self, driver: "AsyncDriverAdapterBase") -> None:
+        """Safely commit a transaction only if autocommit is disabled.
+
+        Args:
+            driver: The database driver to use.
+        """
+        try:
+            connection = getattr(driver, "connection", None)
+            if connection and hasattr(connection, "autocommit") and getattr(connection, "autocommit", False):
+                return
+
+            driver_features = getattr(driver, "driver_features", {})
+            if driver_features and driver_features.get("autocommit", False):
+                return
+
+            await driver.commit()
+        except Exception:
+            logger.debug("Failed to commit transaction, likely due to autocommit being enabled")

sqlspec/protocols.py

@@ -1,7 +1,7 @@
-"""Runtime-checkable protocols for SQLSpec to replace duck typing.
+"""Runtime-checkable protocols for type safety and runtime checks.
 
 This module provides protocols that can be used for static type checking
-and runtime isinstance() checks, replacing defensive hasattr() patterns.
+and runtime isinstance() checks.
 """
 
 from typing import TYPE_CHECKING, Any, ClassVar, Optional, Protocol, Union, runtime_checkable
@@ -20,6 +20,7 @@ if TYPE_CHECKING:
 __all__ = (
     "BytesConvertibleProtocol",
     "DictProtocol",
+    "ExpressionWithAliasProtocol",
     "FilterAppenderProtocol",
     "FilterParameterProtocol",
     "HasExpressionProtocol",
@@ -172,6 +173,15 @@ class BytesConvertibleProtocol(Protocol):
         ...
 
 
+@runtime_checkable
+class ExpressionWithAliasProtocol(Protocol):
+    """Protocol for SQL expressions that support aliasing with as_() method."""
+
+    def as_(self, alias: str, **kwargs: Any) -> "exp.Alias":
+        """Create an aliased expression."""
+        ...
+
+
 @runtime_checkable
 class ObjectStoreItemProtocol(Protocol):
     """Protocol for object store items with path/key attributes."""
@@ -182,10 +192,7 @@ class ObjectStoreItemProtocol(Protocol):
 
 @runtime_checkable
 class ObjectStoreProtocol(Protocol):
-    """Protocol for object storage operations.
-
-    Defines the interface for storage backends with both sync and async operations.
-    """
+    """Protocol for object storage operations."""
 
     capabilities: ClassVar["StorageCapabilities"]
 

sqlspec/storage/__init__.py

@@ -1,10 +1,10 @@
 """Storage abstraction layer for SQLSpec.
 
-Provides a flexible storage system with:
+Provides a storage system with:
 - Multiple backend support (local, fsspec, obstore)
-- Lazy loading and configuration-based registration
-- URI scheme-based automatic backend resolution
-- Key-based named storage configurations
+- Configuration-based registration
+- URI scheme-based backend resolution
+- Named storage configurations
 - Capability-based backend selection
 """
 

sqlspec/storage/backends/fsspec.py

@@ -32,7 +32,7 @@ class _ArrowStreamer:
         return self
 
     async def _initialize(self) -> None:
-        """Initialize the paths iterator."""
+        """Initialize paths iterator."""
         if self.paths_iterator is None:
             paths = await async_(self.backend.glob)(self.pattern, **self.kwargs)
             self.paths_iterator = iter(paths)
@@ -59,9 +59,8 @@ class _ArrowStreamer:
 class FSSpecBackend(ObjectStoreBase):
     """Storage backend using fsspec.
 
-    Implements the ObjectStoreProtocol using fsspec,
-    providing support for various protocols including HTTP, HTTPS, FTP,
-    and cloud storage services.
+    Implements the ObjectStoreProtocol using fsspec for various protocols
+    including HTTP, HTTPS, FTP, and cloud storage services.
     """
 
     _default_capabilities: ClassVar[StorageCapabilities] = StorageCapabilities(
@@ -116,7 +115,7 @@ class FSSpecBackend(ObjectStoreBase):
         return path_str
 
     def _detect_capabilities(self) -> StorageCapabilities:
-        """Detect capabilities based on underlying filesystem protocol."""
+        """Detect capabilities based on filesystem protocol."""
         protocol = self.protocol.lower()
 
         if protocol in {"s3", "s3a", "s3n"}:
@@ -138,7 +137,7 @@ class FSSpecBackend(ObjectStoreBase):
 
     @property
     def capabilities(self) -> StorageCapabilities:
-        """Return instance-specific capabilities based on detected protocol."""
+        """Return capabilities based on detected protocol."""
         return getattr(self, "_instance_capabilities", self.__class__._default_capabilities)
 
     @classmethod

sqlspec/storage/backends/obstore.py

@@ -1,8 +1,7 @@
 """Object storage backend using obstore.
 
-Implements the ObjectStoreProtocol using obstore,
-providing native support for S3, GCS, Azure, and local file storage
-with Arrow support.
+Implements the ObjectStoreProtocol using obstore for S3, GCS, Azure,
+and local file storage with Arrow support.
 """
 
 from __future__ import annotations
@@ -56,11 +55,11 @@ DEFAULT_OPTIONS: Final[dict[str, Any]] = {"connect_timeout": "30s", "request_tim
 class ObStoreBackend(ObjectStoreBase, HasStorageCapabilities):
     """Object storage backend using obstore.
 
-    Uses obstore's Rust-based implementation for storage operations,
-    providing native support for AWS S3, Google Cloud Storage, Azure Blob Storage,
+    Uses obstore's Rust-based implementation for storage operations.
+    Supports AWS S3, Google Cloud Storage, Azure Blob Storage,
     local filesystem, and HTTP endpoints.
 
-    Includes native Arrow support.
+    Includes Arrow support.
     """
 
     capabilities: ClassVar[StorageCapabilities] = StorageCapabilities(
@@ -203,7 +202,7 @@ class ObStoreBackend(ObjectStoreBase, HasStorageCapabilities):
             raise StorageOperationFailedError(msg) from exc
 
     def glob(self, pattern: str, **kwargs: Any) -> list[str]:
-        """Find objects matching pattern using obstore.
+        """Find objects matching pattern.
 
         Lists all objects and filters them client-side using the pattern.
         """
@@ -330,7 +329,7 @@ class ObStoreBackend(ObjectStoreBase, HasStorageCapabilities):
             raise StorageOperationFailedError(msg) from exc
 
     def stream_arrow(self, pattern: str, **kwargs: Any) -> Iterator[ArrowRecordBatch]:
-        """Stream Arrow record batches using obstore.
+        """Stream Arrow record batches.
 
         Yields:
             Iterator of Arrow record batches from matching objects.

sqlspec/storage/registry.py

@@ -1,8 +1,8 @@
-"""Unified Storage Registry for ObjectStore backends.
+"""Storage Registry for ObjectStore backends.
 
-Provides a flexible, lazy-loading storage registry that supports URI-first access
+Provides a storage registry that supports URI-first access
 pattern with automatic backend detection, ObStore preferred with FSSpec fallback,
-intelligent scheme-based routing, and named aliases for common configurations.
+scheme-based routing, and named aliases for common configurations.
 """
 
 import logging

sqlspec/utils/__init__.py

@@ -1,3 +1,3 @@
-from sqlspec.utils import deprecation, fixtures, module_loader, singleton, sync_tools, text, type_guards
+from sqlspec.utils import deprecation, fixtures, logging, module_loader, singleton, sync_tools, text, type_guards
 
-__all__ = ("deprecation", "fixtures", "module_loader", "singleton", "sync_tools", "text", "type_guards")
+__all__ = ("deprecation", "fixtures", "logging", "module_loader", "singleton", "sync_tools", "text", "type_guards")

sqlspec/utils/logging.py

@@ -5,23 +5,19 @@ Users should configure their own logging handlers and levels as needed.
 SQLSpec provides StructuredFormatter for JSON-formatted logs if desired.
 """
 
-from __future__ import annotations
-
 import logging
 from contextvars import ContextVar
-from typing import TYPE_CHECKING, Any
+from logging import LogRecord
+from typing import Any, Optional
 
 from sqlspec._serialization import encode_json
 
-if TYPE_CHECKING:
-    from logging import LogRecord
-
 __all__ = ("StructuredFormatter", "correlation_id_var", "get_correlation_id", "get_logger", "set_correlation_id")
 
-correlation_id_var: ContextVar[str | None] = ContextVar("correlation_id", default=None)
+correlation_id_var: "ContextVar[Optional[str]]" = ContextVar("correlation_id", default=None)
 
 
-def set_correlation_id(correlation_id: str | None) -> None:
+def set_correlation_id(correlation_id: "Optional[str]") -> None:
     """Set the correlation ID for the current context.
 
     Args:
@@ -30,7 +26,7 @@ def set_correlation_id(correlation_id: str | None) -> None:
     correlation_id_var.set(correlation_id)
 
 
-def get_correlation_id() -> str | None:
+def get_correlation_id() -> "Optional[str]":
     """Get the current correlation ID.
 
     Returns:
@@ -90,7 +86,7 @@ class CorrelationIDFilter(logging.Filter):
         return True
 
 
-def get_logger(name: str | None = None) -> logging.Logger:
+def get_logger(name: "Optional[str]" = None) -> logging.Logger:
     """Get a logger instance with standardized configuration.
 
     Args:

sqlspec/utils/sync_tools.py

@@ -1,3 +1,10 @@
+"""Utilities for async/sync interoperability in SQLSpec.
+
+This module provides utilities for converting between async and sync functions,
+managing concurrency limits, and handling context managers. Used primarily
+for adapter implementations that need to support both sync and async patterns.
+"""
+
 import asyncio
 import functools
 import inspect
@@ -26,7 +33,15 @@ class CapacityLimiter:
     """Limits the number of concurrent operations using a semaphore."""
 
     def __init__(self, total_tokens: int) -> None:
-        self._semaphore = asyncio.Semaphore(total_tokens)
+        self._total_tokens = total_tokens
+        self._semaphore_instance: Optional[asyncio.Semaphore] = None
+
+    @property
+    def _semaphore(self) -> asyncio.Semaphore:
+        """Lazy initialization of asyncio.Semaphore for Python 3.9 compatibility."""
+        if self._semaphore_instance is None:
+            self._semaphore_instance = asyncio.Semaphore(self._total_tokens)
+        return self._semaphore_instance
 
     async def acquire(self) -> None:
         await self._semaphore.acquire()
@@ -36,11 +51,12 @@ class CapacityLimiter:
 
     @property
     def total_tokens(self) -> int:
-        return self._semaphore._value
+        return self._total_tokens
 
     @total_tokens.setter
     def total_tokens(self, value: int) -> None:
-        self._semaphore = asyncio.Semaphore(value)
+        self._total_tokens = value
+        self._semaphore_instance = None
 
     async def __aenter__(self) -> None:
         await self.acquire()
@@ -76,7 +92,14 @@ def run_(async_function: "Callable[ParamSpecT, Coroutine[Any, Any, ReturnT]]") -
             loop = None
 
         if loop is not None:
-            return asyncio.run(partial_f())
+            if loop.is_running():
+                import concurrent.futures
+
+                with concurrent.futures.ThreadPoolExecutor() as executor:
+                    future = executor.submit(asyncio.run, partial_f())
+                    return future.result()
+            else:
+                return asyncio.run(partial_f())
         if uvloop and sys.platform != "win32":
             uvloop.install()  # pyright: ignore[reportUnknownMemberType]
         return asyncio.run(partial_f())

sqlspec/utils/text.py

@@ -1,4 +1,9 @@
-"""General utility functions."""
+"""Text processing utilities for SQLSpec.
+
+Provides functions for string manipulation including case conversion,
+slugification, and email validation. Used primarily for identifier
+generation and data validation.
+"""
 
 import re
 import unicodedata

{sqlspec-0.17.0.dist-info → sqlspec-0.18.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlspec
-Version: 0.17.0
+Version: 0.18.0
 Summary: SQL Experiments in Python
 Project-URL: Discord, https://discord.gg/litestar
 Project-URL: Issue, https://github.com/litestar-org/sqlspec/issues/

sqlspec-0.18.0.dist-info/RECORD

@@ -0,0 +1,138 @@
+sqlspec/__init__.py,sha256=JkL9cp1h19tz1rCl-3WAep6kbZs8JLxNlb1SrQl8PWc,2114
+sqlspec/__main__.py,sha256=lXBKZMOXA1uY735Rnsb-GS7aXy0nt22tYmd2X9FcxrY,253
+sqlspec/__metadata__.py,sha256=IUw6MCTy1oeUJ1jAVYbuJLkOWbiAWorZ5W-E-SAD9N4,395
+sqlspec/_serialization.py,sha256=6U5-smk2h2yl0i6am2prtOLJTdu4NJQdcLlSfSUMaUQ,2590
+sqlspec/_sql.py,sha256=j9WljOgCme4jTfL6NegEWOhK-Rr3JEmhtbneh8ZN1bQ,45221
+sqlspec/_typing.py,sha256=jv-7QHGLrJLfnP86bR-Xcmj3PDoddNZEKDz_vYRBiAU,22684
+sqlspec/base.py,sha256=-5hxBpjEvbn30aXksbErkw3lH0Q3-u7h6D2JB9BSAmU,25429
+sqlspec/cli.py,sha256=3ZxPwl4neNWyrAkM9J9ccC_gaFigDJbhuZfx15JVE7E,9903
+sqlspec/config.py,sha256=QJMR46YC-6Kd4GUnqQm0s679Br4R-QEEruuKpJxHEec,15969
+sqlspec/exceptions.py,sha256=zBnzQOfYAgqX04GoaC9Io6ardzinldkEuZ3YtR5vr9U,6071
+sqlspec/loader.py,sha256=HDfMZDj7l9aLAxeZR2Rv-HC-dWah4CPkwZ6HKQ7yP-Y,23398
+sqlspec/protocols.py,sha256=Of6uJyxvawExCEyR3u7jbxOckUcwG0HHOEXmfHyev40,13106
+sqlspec/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/typing.py,sha256=yj8D8O-pkfUVZDfVHEgQaB95-5alwgQbp_sqNJOVhvQ,6301
+sqlspec/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/adapters/adbc/__init__.py,sha256=CqBaEahvHbQ5NxmBJuuCtZ8v5wuaPgbqSSe9WUhglWY,336
+sqlspec/adapters/adbc/_types.py,sha256=htTZ20qVo6Iod_ooUTLGd0xAL8AUYA4qeOexTaLQG3Q,351
+sqlspec/adapters/adbc/config.py,sha256=gekx9tzOA5vpcMWqczX5B5xKMbXCEnyYWavrrYll7j0,13583
+sqlspec/adapters/adbc/driver.py,sha256=E36PI499cMsi-CGeRNe21lkiIwZKNSHGusmxLxy6mIQ,18809
+sqlspec/adapters/aiosqlite/__init__.py,sha256=rgooFeHm7IsnC9fOnRQwMvbJdgkTvhnNTYphayVz9q8,869
+sqlspec/adapters/aiosqlite/_types.py,sha256=M8VqaW6iCigfExhdJM2yPAN09Ue2yBoBL59-QXGBObo,355
+sqlspec/adapters/aiosqlite/config.py,sha256=MmCl_0Be89OijlcY1EUgA87A1NNtF1CHjl6p_R6Dfks,8112
+sqlspec/adapters/aiosqlite/driver.py,sha256=pqrrRNmKKeEPX8871TzLwdehFB6-6ULwpM2Ke9Tsl-o,9458
+sqlspec/adapters/aiosqlite/pool.py,sha256=cuzkXA2hMMk16aX28aNRaTmLDmqlepka9QhcWsdUA2E,17140
+sqlspec/adapters/asyncmy/__init__.py,sha256=wBgak4MA3ySaGQHUxrnv3HdSVQKKRWf1_DmENL0LkWs,531
+sqlspec/adapters/asyncmy/_types.py,sha256=WbwREzJkLYmqd4c7A7cu04WaD5g7-n35ZETHZvbP1Z8,250
+sqlspec/adapters/asyncmy/config.py,sha256=DHk-MTkBcGPPmYI4RMJbUWWKbcu1ZaoG_8LS0aZR0P8,6603
+sqlspec/adapters/asyncmy/driver.py,sha256=DitgufSL3N4LlDOq_UB741f4czPPPTJrusLbuiSjagY,9911
+sqlspec/adapters/asyncpg/__init__.py,sha256=h3lg3pLBLrS9ABBt5yKFKhGL69RpOlASYSqVCuAuURY,567
+sqlspec/adapters/asyncpg/_types.py,sha256=p65WrqfUl2cx5Md759_aV8m_gHP-PyCzNk48bLhDQkc,425
+sqlspec/adapters/asyncpg/config.py,sha256=WrP-PZeWbumpKb8VrJYD-8hpBkjrNG1PQ5kfR2vFWo4,8867
+sqlspec/adapters/asyncpg/driver.py,sha256=ZEuCeyrpaIQme0cF-hupGlfxbkN9ReZBjflpgbEJEw8,11073
+sqlspec/adapters/bigquery/__init__.py,sha256=1nzkWubO4ZObuK27QoeGfS05ed5v02NU4GaXrvqjlw0,504
+sqlspec/adapters/bigquery/_types.py,sha256=gECmbiSmZmnR-xJpmznhnva4SbhhdcYQ09jBdqMAmOs,255
+sqlspec/adapters/bigquery/config.py,sha256=Dp0LKg9p88n4C4WQMaf9_A7YicIJT9DgcCXci46NTFk,9434
+sqlspec/adapters/bigquery/driver.py,sha256=gAQWd8R3NaqYzbPkNr59kxKVKW8Yz4KmU7BjUJXesoo,17745
+sqlspec/adapters/duckdb/__init__.py,sha256=c5GCYxhTFhvB8fUTt_d-dipaQfM8X5kv4GsU47SwJ6E,693
+sqlspec/adapters/duckdb/_types.py,sha256=4p5nuD1yNs0fmQ5sGxKFx2ru6jUCPrpsYgqfh-uZHM8,270
+sqlspec/adapters/duckdb/config.py,sha256=7CWXpKWX1NrzO1sORVFfD3TzdVxj62MbujnsBBCGSHQ,10037
+sqlspec/adapters/duckdb/driver.py,sha256=TiQ1HEWVJUwxENZFca62slYMqvFzVa11P3nhf8TggTA,12491
+sqlspec/adapters/duckdb/pool.py,sha256=2P4iAhJQ070wA_0O_ZXNfOCiXFxaW7zeyoL05A7izL4,8413
+sqlspec/adapters/oracledb/__init__.py,sha256=AUsZ8m9tDgNwxv9m4LtkIs9Ib6hOtBrGNm-Ee4HWNq0,843
+sqlspec/adapters/oracledb/_types.py,sha256=7Oe0DmA1ixptQjPMioQF7r2i451zzN2BMut2jTi2c3U,808
+sqlspec/adapters/oracledb/config.py,sha256=xdVObosNsGVhtuiWRfMQ9_78pO25vCOUIy_ktM8sYWc,12037
+sqlspec/adapters/oracledb/driver.py,sha256=XwkQGEWksXmS5wT13-vDuwqN5D203Z7C1KzuWWOr55g,19275
+sqlspec/adapters/oracledb/migrations.py,sha256=FwjkMBbzEQRkvZdfvpPUyPiWHCP6A_IVg6_6RIYSXYA,9136
+sqlspec/adapters/psqlpy/__init__.py,sha256=ABve2Oj-G-fxMO8WRJ0XzxEw2cs5H3INDnmUI99l8gc,526
+sqlspec/adapters/psqlpy/_types.py,sha256=tG4jwQtBB6mCX5KU5x3hAhxsQUQlQEzbCsYhU3IouVc,269
+sqlspec/adapters/psqlpy/config.py,sha256=ZDZKscjDViyID78SNfOeY9IZZFYQD__hJoYLFBm4TCo,7908
+sqlspec/adapters/psqlpy/driver.py,sha256=Rxpoezw6mrnqlhEB0DrYUHQK7ed-wLtgXYLV2gtJlwI,16146
+sqlspec/adapters/psycopg/__init__.py,sha256=swmTz8xlj6LvB-i78tBSE_T-sez2e6lFJW2cMzNJEVE,862
+sqlspec/adapters/psycopg/_types.py,sha256=UJBKDWgARoSR6UxSy8zcIP7HSHTpXU5moJTsjDj0M4M,563
+sqlspec/adapters/psycopg/config.py,sha256=PP6iwIKvINzOXZE4b99auCgR6eEfzSX-sFyHcBMb-xg,16558
+sqlspec/adapters/psycopg/driver.py,sha256=6Xp3MmMc7xZBpL8mdVlR5FMInXsREkob588t_oNHoVE,26762
+sqlspec/adapters/sqlite/__init__.py,sha256=hGZX0D6vHJMSah60_hgWHoLmcqKGm1lYcz2r02wc0p0,574
+sqlspec/adapters/sqlite/_types.py,sha256=4Nqolhk8s4mwLw13BwUjuyhAbM9BsKwJCvcmjMWkhaY,246
+sqlspec/adapters/sqlite/config.py,sha256=zXKipKuTPjFSUgdc5tiHAZF5hW33jvrrElNJAQhpOQM,5069
+sqlspec/adapters/sqlite/driver.py,sha256=52WzO8cGIERBv6fWYQYp6Myti9wYHyBzuLN8dJxsNZY,9409
+sqlspec/adapters/sqlite/pool.py,sha256=bKIrf3hgach6bMyFQORCdqnpN-sTfMagB3l61uPCGJI,4435
+sqlspec/builder/__init__.py,sha256=o328O3dNRj4vGXmOclIxDLzmiwkUCE_Aqmqv-LT5lWY,1665
+sqlspec/builder/_base.py,sha256=SkyDJYZ7QCBi4CHZQ_0e1SvA05-Hhfbvi_qS7BwjcVA,18022
+sqlspec/builder/_column.py,sha256=GfTMFPzJIhwDVKefMbyom-HoEj-KSKB92-XoZfchnEo,13289
+sqlspec/builder/_ddl.py,sha256=AyL0d1BnHkESMoZQhdFQTgGEVYRiRt-cgDiX4IKE5Vg,56533
+sqlspec/builder/_delete.py,sha256=xWA5nQB3UB8kpEGXN2k5ynt4cGZ7blkNoURpI0bKoeg,2264
+sqlspec/builder/_expression_wrappers.py,sha256=HTl8qAFD4sZNXLD2akkJWGCPse2HWS237mTGE4Cx_7I,1244
+sqlspec/builder/_insert.py,sha256=kqjjIfGcy49xDTTOMmb4GBvWsJqT1GOr_1Y0syAIsok,16863
+sqlspec/builder/_merge.py,sha256=95PLQSKA3zjk0wTZG3m817fTZpsS95PrS2qF34iLAP8,2004
+sqlspec/builder/_parsing_utils.py,sha256=RH8OFBFAetalEgHW5JLcEyyCdW_awVdy07MjboOkqL4,8383
+sqlspec/builder/_select.py,sha256=m5sfyuAssjlNimLLNBAeFooVIfM2FgKN1boPfdsOkaA,5785
+sqlspec/builder/_update.py,sha256=QieiguEq9T_UECv10f1xwQJp58gc3w246cvtCDpPwuw,6020
+sqlspec/builder/mixins/__init__.py,sha256=YXhAzKmQbQtne5j26SKWY8PUxwosl0RhlhLoahAdkj0,1885
+sqlspec/builder/mixins/_cte_and_set_ops.py,sha256=p5O9m_jvpaWxv1XP9Ys2DRI-qOTq30rr2EwYjAbIT8o,9088
+sqlspec/builder/mixins/_delete_operations.py,sha256=l0liajnoAfRgtWtyStuAIfxreEFRkNO4DtBwyGqAfic,1198
+sqlspec/builder/mixins/_insert_operations.py,sha256=ECFuX52fNYYxWt11e1n8jHx9wu5_RJSnmm9Dl_aXnIY,10326
+sqlspec/builder/mixins/_join_operations.py,sha256=8o_aApK5cmJbNCNfWa4bs5fR2zgQUjon5p-oyiW41Qw,11440
+sqlspec/builder/mixins/_merge_operations.py,sha256=KnvXxHKEJu3ofuQsLJlPf7H4mdkH4mKwGHoXtwzdK1o,24191
+sqlspec/builder/mixins/_order_limit_operations.py,sha256=KrTXE0HmKCMn4AkNFyMhEZrmJ2Dl-o5kSiCjk2jNIKM,5499
+sqlspec/builder/mixins/_pivot_operations.py,sha256=j5vdzXuEqB1Jn3Ie_QjVwSH2_OEi65oZ64bQJHd3jXo,6108
+sqlspec/builder/mixins/_select_operations.py,sha256=7CwKu5pt5Nu4tdFIXciQOnd6IcbiRVO8YcOFWHePafg,35239
+sqlspec/builder/mixins/_update_operations.py,sha256=rcmnmSSwHSujqe0gd-dKsWHLX9nPMS1we1Y3_h4S7G4,8654
+sqlspec/builder/mixins/_where_clause.py,sha256=1iz7Y2x_ooG2bOCu2zX0v5_bkGFpAckVQKvnyrR1JNQ,36373
+sqlspec/core/__init__.py,sha256=4pD6ymg_sZxh9QyAmsk3oNyvUWPoQj9qktg0uNJVp6k,5264
+sqlspec/core/cache.py,sha256=835tHhe3fan9_h_Y9CJ-4aAl8K-0cNMwfQzcxMzjlqQ,24837
+sqlspec/core/compiler.py,sha256=CdD5WqkUTUspetAN009mXF6aYmHCa1NdG6HbHAhjUMY,12066
+sqlspec/core/filters.py,sha256=JQRFw7ND68tWvWQpLIZbDNUGk3mdcw9Yg-ldzLcfVqQ,30475
+sqlspec/core/hashing.py,sha256=s8aY6xY0fOwYNIWYgUqmfkALt3BBK1s5Eq30fcnscL8,8529
+sqlspec/core/parameters.py,sha256=pOSjWQqmuSgi4-eS3SMIfTcgXLVlM9Qd21G-vfVDymY,49083
+sqlspec/core/result.py,sha256=yjc7mbntsrx7atrxb8CAdrvtUNLmSuCCM0N4J_zpzQA,20490
+sqlspec/core/splitter.py,sha256=Qb4cyhs8ll3Wm9z11-WWsvpkUYnyEcmBz2KuYMG2evI,27914
+sqlspec/core/statement.py,sha256=Ydk_OImRpmbu5EnvM0zgnOWw-H7tCWhoeJHlLZQBHbk,25906
+sqlspec/driver/__init__.py,sha256=QVpDRQGd1GreIP199en6qDbq-cZJcEF5go68DINagUk,569
+sqlspec/driver/_async.py,sha256=aS5AwY5IYqnVT8ldDLDwz2AMDN94CI9hfsOz-1k-Nus,18992
+sqlspec/driver/_common.py,sha256=6vm4MY_CJW3Q8_roIG9hwK18tIMT_zIESmUoUyRSreg,23576
+sqlspec/driver/_sync.py,sha256=wCBV9QfAH8BPjrrVCQc2eM90ai5-FYbKDd81L5sZMS0,18767
+sqlspec/driver/mixins/__init__.py,sha256=gN4pQyJXxNy0xi91dcMJGA7DQ7TbjGjQI24SSpZc6Go,248
+sqlspec/driver/mixins/_result_tools.py,sha256=4dLU-2uL6qCS-s7cve7DmHRZXKGPqdpWtl9OKMOgbR4,6599
+sqlspec/driver/mixins/_sql_translator.py,sha256=wRj7neNbMFvzNfy7v0_acBAoZusKHh25UW2WsD_hSlQ,3005
+sqlspec/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/extensions/aiosql/__init__.py,sha256=-9cefc9pYPf9vCgALoB-y1DtmcgRjKe2azfl6RIarAA,414
+sqlspec/extensions/aiosql/adapter.py,sha256=CXkNZaZq86ZhfYFGx4IFbkHmbIFQKMd9CS6Q2jkMCok,16009
+sqlspec/extensions/litestar/__init__.py,sha256=tOmQ7RHSWOot7p30gk0efxxuP0OCq1opyyZqNmQY7FE,320
+sqlspec/extensions/litestar/_utils.py,sha256=o-FuUj1_WkDrLxQxiP6hXDak66XfyRP3QLyEVKrIRjI,1954
+sqlspec/extensions/litestar/cli.py,sha256=X4DlAx3Ry-ccOjAQSxe8SMtyJKCFJVLTbENPU_efKuU,1356
+sqlspec/extensions/litestar/config.py,sha256=3UI_vhtbupCLsf1nhUgUpRlCoUS5c0GsAjWvegT0c3c,4462
+sqlspec/extensions/litestar/handlers.py,sha256=3LreU8rZvuHaJnKlN09ttu4wSorWJedsuKgeLT-cOEc,9993
+sqlspec/extensions/litestar/plugin.py,sha256=I0aRnL4oZPUYk7pYhZSL3yikl7ViM0kr33kVmH4W-MU,5637
+sqlspec/extensions/litestar/providers.py,sha256=5LRb5JvRV_XZdNOKkdaIy3j5x-dFCcAi1ea1pgwuapI,18882
+sqlspec/migrations/__init__.py,sha256=RiDi_HkUIgXtu_33QnRdvYNqcCn-euHUiWwTiPr5IGc,1055
+sqlspec/migrations/base.py,sha256=vIzQzUtQrNKDec6XUeRHcCBuWU1KNtRCFpOvVxsp3sQ,13093
+sqlspec/migrations/commands.py,sha256=-De-1RnqjI-BwgOuFvULkJYZccPtmxPPAQTCLlU2TvM,19122
+sqlspec/migrations/loaders.py,sha256=WSP3du5tLy9o5SsfvCCU9vhOiOA3ds7L_VzakrSfnmA,12932
+sqlspec/migrations/runner.py,sha256=y6fyZi02n8MseKR8XFWXUEOOYQNG_w_DikHVxH9p20M,10730
+sqlspec/migrations/tracker.py,sha256=hfrZGz8M70SfFniw4aXVtHNg4p8EPFm67vthjfUMUys,6843
+sqlspec/migrations/utils.py,sha256=gWnCOdr8pwfkgG-FSUJgRz4q9TlCgOXY_B7n59NrgVA,3746
+sqlspec/storage/__init__.py,sha256=xWSsq5QXrY7wCsjQYPldfdlm8UEJ-kojU-tWsoldSy0,645
+sqlspec/storage/capabilities.py,sha256=vyousxls9jISsykgoybpNHlGWN6Hq_pKcsZ5DmKGWvU,3045
+sqlspec/storage/registry.py,sha256=-H2ttz41M-DNfkEwANS704BDs1rtyii69G7iqsLGRG4,9424
+sqlspec/storage/backends/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sqlspec/storage/backends/base.py,sha256=KS2JRZILoH_R_xsfKtYkqQ5a1r5OOBDSE5KbibTmhGY,5730
+sqlspec/storage/backends/fsspec.py,sha256=8AX8ULwlApMd6WtHDVJBomdvk5UjPtfMf78TJ86gTC8,15994
+sqlspec/storage/backends/obstore.py,sha256=CXad0ug_b3snVNx50SzvvFsDXpkK4LV3jROIrwf6dhg,19783
+sqlspec/utils/__init__.py,sha256=4Yd2HnY-DJ6wf4jBrNTM3B41KlCcaFefP7bnux7Q17g,231
+sqlspec/utils/correlation.py,sha256=2jvkAY3nkU3UxNU_9pbBR6cz3A1Q1cGG9IaWSSOIb1Q,4195
+sqlspec/utils/deprecation.py,sha256=L5ylPlkrWahXZ3q2Yta2ReFh4pA8rZapLNw9u_mOOEE,3794
+sqlspec/utils/fixtures.py,sha256=Qm2uNMaL_6l6tlij-Pm3tLwD906iFK_OXhwZqOx3WgY,1807
+sqlspec/utils/logging.py,sha256=zAM7rHJ-KsmAj1yjvU9QFoiwf4Q2hKTere2J62FlllI,3664
+sqlspec/utils/module_loader.py,sha256=m5PSN9NwOLd0ZJBuqMVYVi-vaIQMBCVd25vnM3-rv3k,2823
+sqlspec/utils/serializers.py,sha256=TKsRryRcYMnb8Z8MGkYGClIxcYvC8CW7MsrPQTJqEcY,154
+sqlspec/utils/singleton.py,sha256=SKnszJi1NPeERgX7IjVIGYAYx4XqR1E_rph3bU6olAU,1047
+sqlspec/utils/sync_tools.py,sha256=f_bjHTXUaDcsNQY63-L7j5xn2MHfzXro-ShzJM-WLUI,8356
+sqlspec/utils/text.py,sha256=W97aX77A3NzG795AHjhdX6zqOBDmvLaXLCno2JIugCo,3081
+sqlspec/utils/type_guards.py,sha256=9C4SRebO4JiQrMzcJZFUA0KjSU48G26RmX6lbijyjBg,30476
+sqlspec-0.18.0.dist-info/METADATA,sha256=SNUm1W2FagyGjCmPrfUcCNBRSyurCMbg60IUMf9iA1E,16822
+sqlspec-0.18.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlspec-0.18.0.dist-info/entry_points.txt,sha256=G-ZqY1Nuuw3Iys7nXw23f6ILenk_Lt47VdK2mhJCWHg,53
+sqlspec-0.18.0.dist-info/licenses/LICENSE,sha256=MdujfZ6l5HuLz4mElxlu049itenOR3gnhN1_Nd3nVcM,1078
+sqlspec-0.18.0.dist-info/licenses/NOTICE,sha256=Lyir8ozXWov7CyYS4huVaOCNrtgL17P-bNV-5daLntQ,1634
+sqlspec-0.18.0.dist-info/RECORD,,

sqlspec/builder/_ddl_utils.py

@@ -1,103 +0,0 @@
-"""DDL builder utilities."""
-
-from typing import TYPE_CHECKING, Optional
-
-from sqlglot import exp
-
-if TYPE_CHECKING:
-    from sqlspec.builder._ddl import ColumnDefinition, ConstraintDefinition
-
-__all__ = ("build_column_expression", "build_constraint_expression")
-
-
-def build_column_expression(col: "ColumnDefinition") -> "exp.Expression":
-    """Build SQLGlot expression for a column definition."""
-    col_def = exp.ColumnDef(this=exp.to_identifier(col.name), kind=exp.DataType.build(col.dtype))
-
-    constraints: list[exp.ColumnConstraint] = []
-
-    if col.not_null:
-        constraints.append(exp.ColumnConstraint(kind=exp.NotNullColumnConstraint()))
-
-    if col.primary_key:
-        constraints.append(exp.ColumnConstraint(kind=exp.PrimaryKeyColumnConstraint()))
-
-    if col.unique:
-        constraints.append(exp.ColumnConstraint(kind=exp.UniqueColumnConstraint()))
-
-    if col.default is not None:
-        default_expr: Optional[exp.Expression] = None
-        if isinstance(col.default, str):
-            if col.default.upper() in {"CURRENT_TIMESTAMP", "CURRENT_DATE", "CURRENT_TIME"} or "(" in col.default:
-                default_expr = exp.maybe_parse(col.default)
-            else:
-                default_expr = exp.convert(col.default)
-        else:
-            default_expr = exp.convert(col.default)
-
-        constraints.append(exp.ColumnConstraint(kind=default_expr))
-
-    if col.check:
-        check_expr = exp.Check(this=exp.maybe_parse(col.check))
-        constraints.append(exp.ColumnConstraint(kind=check_expr))
-
-    if col.comment:
-        constraints.append(exp.ColumnConstraint(kind=exp.CommentColumnConstraint(this=exp.convert(col.comment))))
-
-    if col.generated:
-        generated_expr = exp.GeneratedAsIdentityColumnConstraint(this=exp.maybe_parse(col.generated))
-        constraints.append(exp.ColumnConstraint(kind=generated_expr))
-
-    if col.collate:
-        constraints.append(exp.ColumnConstraint(kind=exp.CollateColumnConstraint(this=exp.to_identifier(col.collate))))
-
-    if constraints:
-        col_def.set("constraints", constraints)
-
-    return col_def
-
-
-def build_constraint_expression(constraint: "ConstraintDefinition") -> "Optional[exp.Expression]":
-    """Build SQLGlot expression for a table constraint."""
-    if constraint.constraint_type == "PRIMARY KEY":
-        pk_cols = [exp.to_identifier(col) for col in constraint.columns]
-        pk_constraint = exp.PrimaryKey(expressions=pk_cols)
-
-        if constraint.name:
-            return exp.Constraint(this=exp.to_identifier(constraint.name), expression=pk_constraint)
-        return pk_constraint
-
-    if constraint.constraint_type == "FOREIGN KEY":
-        fk_cols = [exp.to_identifier(col) for col in constraint.columns]
-        ref_cols = [exp.to_identifier(col) for col in constraint.references_columns]
-
-        fk_constraint = exp.ForeignKey(
-            expressions=fk_cols,
-            reference=exp.Reference(
-                this=exp.to_table(constraint.references_table) if constraint.references_table else None,
-                expressions=ref_cols,
-                on_delete=constraint.on_delete,
-                on_update=constraint.on_update,
-            ),
-        )
-
-        if constraint.name:
-            return exp.Constraint(this=exp.to_identifier(constraint.name), expression=fk_constraint)
-        return fk_constraint
-
-    if constraint.constraint_type == "UNIQUE":
-        unique_cols = [exp.to_identifier(col) for col in constraint.columns]
-        unique_constraint = exp.UniqueKeyProperty(expressions=unique_cols)
-
-        if constraint.name:
-            return exp.Constraint(this=exp.to_identifier(constraint.name), expression=unique_constraint)
-        return unique_constraint
-
-    if constraint.constraint_type == "CHECK":
-        check_expr = exp.Check(this=exp.maybe_parse(constraint.condition) if constraint.condition else None)
-
-        if constraint.name:
-            return exp.Constraint(this=exp.to_identifier(constraint.name), expression=check_expr)
-        return check_expr
-
-    return None