sqlspec 0.16.1__cp39-cp39-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl

sqlspec/__init__.py

@@ -0,0 +1,92 @@
+"""SQLSpec: Safe and elegant SQL query building for Python."""
+
+from sqlspec import adapters, base, builder, core, driver, exceptions, extensions, loader, migrations, typing, utils
+from sqlspec.__metadata__ import __version__
+from sqlspec._sql import SQLFactory, sql
+from sqlspec.base import SQLSpec
+from sqlspec.builder import (
+    Column,
+    ColumnExpression,
+    CreateTable,
+    Delete,
+    DropTable,
+    FunctionColumn,
+    Insert,
+    Merge,
+    QueryBuilder,
+    Select,
+    Update,
+)
+from sqlspec.config import AsyncDatabaseConfig, SyncDatabaseConfig
+from sqlspec.core import (
+    SQL,
+    ArrowResult,
+    CacheConfig,
+    CacheStats,
+    ParameterConverter,
+    ParameterProcessor,
+    ParameterStyle,
+    ParameterStyleConfig,
+    SQLResult,
+    Statement,
+    StatementConfig,
+)
+from sqlspec.core import filters as filters
+from sqlspec.driver import AsyncDriverAdapterBase, ExecutionResult, SyncDriverAdapterBase
+from sqlspec.loader import SQLFile, SQLFileLoader
+from sqlspec.typing import ConnectionT, DictRow, ModelDTOT, ModelT, RowT, StatementParameters, SupportedSchemaModel
+
+__all__ = (
+    "SQL",
+    "ArrowResult",
+    "AsyncDatabaseConfig",
+    "AsyncDriverAdapterBase",
+    "CacheConfig",
+    "CacheStats",
+    "Column",
+    "ColumnExpression",
+    "ConnectionT",
+    "CreateTable",
+    "Delete",
+    "DictRow",
+    "DropTable",
+    "ExecutionResult",
+    "FunctionColumn",
+    "Insert",
+    "Merge",
+    "ModelDTOT",
+    "ModelT",
+    "ParameterConverter",
+    "ParameterProcessor",
+    "ParameterStyle",
+    "ParameterStyleConfig",
+    "QueryBuilder",
+    "RowT",
+    "SQLFactory",
+    "SQLFile",
+    "SQLFileLoader",
+    "SQLResult",
+    "SQLSpec",
+    "Select",
+    "Statement",
+    "StatementConfig",
+    "StatementParameters",
+    "SupportedSchemaModel",
+    "SyncDatabaseConfig",
+    "SyncDriverAdapterBase",
+    "Update",
+    "__version__",
+    "adapters",
+    "base",
+    "builder",
+    "core",
+    "driver",
+    "exceptions",
+    "extensions",
+    "filters",
+    "loader",
+    "migrations",
+    "sql",
+    "typing",
+    "utils",
+)

sqlspec/__main__.py

@@ -0,0 +1,12 @@
+from sqlspec.cli import add_migration_commands as build_cli_interface
+
+__all__ = ("run_cli",)
+
+
+def run_cli() -> None:  # pragma: no cover
+    """SQLSpec CLI."""
+    build_cli_interface()()
+
+
+if __name__ == "__main__":  # pragma: no cover
+    run_cli()

sqlspec/__metadata__.py

@@ -0,0 +1,14 @@
+"""Metadata for the project."""
+
+from importlib.metadata import PackageNotFoundError, metadata, version
+
+__all__ = ("__project__", "__version__")
+
+try:
+    __version__ = version("sqlspec")
+    __project__ = metadata("sqlspec")["Name"]
+except PackageNotFoundError:  # pragma: no cover
+    __version__ = "0.0.1"
+    __project__ = "SQLSpec"
+finally:
+    del version, PackageNotFoundError, metadata

sqlspec/_serialization.py

@@ -0,0 +1,77 @@
+import datetime
+import enum
+from typing import Any
+
+from sqlspec.typing import PYDANTIC_INSTALLED, BaseModel
+
+
+def _type_to_string(value: Any) -> str:  # pragma: no cover
+    if isinstance(value, datetime.datetime):
+        return convert_datetime_to_gmt_iso(value)
+    if isinstance(value, datetime.date):
+        return convert_date_to_iso(value)
+    if isinstance(value, enum.Enum):
+        return str(value.value)
+    if PYDANTIC_INSTALLED and isinstance(value, BaseModel):
+        return value.model_dump_json()
+    try:
+        return str(value)
+    except Exception as exc:
+        raise TypeError from exc
+
+
+try:
+    from msgspec.json import Decoder, Encoder
+
+    encoder, decoder = Encoder(enc_hook=_type_to_string), Decoder()
+    decode_json = decoder.decode
+
+    def encode_json(data: Any) -> str:  # pragma: no cover
+        return encoder.encode(data).decode("utf-8")
+
+except ImportError:
+    try:
+        from orjson import (  # pyright: ignore[reportMissingImports]
+            OPT_NAIVE_UTC,  # pyright: ignore[reportUnknownVariableType]
+            OPT_SERIALIZE_NUMPY,  # pyright: ignore[reportUnknownVariableType]
+            OPT_SERIALIZE_UUID,  # pyright: ignore[reportUnknownVariableType]
+        )
+        from orjson import dumps as _encode_json  # pyright: ignore[reportUnknownVariableType,reportMissingImports]
+        from orjson import loads as decode_json  # type: ignore[no-redef,assignment,unused-ignore]
+
+        def encode_json(data: Any) -> str:  # pragma: no cover
+            return _encode_json(
+                data, default=_type_to_string, option=OPT_SERIALIZE_NUMPY | OPT_NAIVE_UTC | OPT_SERIALIZE_UUID
+            ).decode("utf-8")
+
+    except ImportError:
+        from json import dumps as encode_json  # type: ignore[assignment]
+        from json import loads as decode_json  # type: ignore[assignment]
+
+__all__ = ("convert_date_to_iso", "convert_datetime_to_gmt_iso", "decode_json", "encode_json")
+
+
+def convert_datetime_to_gmt_iso(dt: datetime.datetime) -> str:  # pragma: no cover
+    """Handle datetime serialization for nested timestamps.
+
+    Args:
+        dt: The datetime to convert.
+
+    Returns:
+        The ISO formatted datetime string.
+    """
+    if not dt.tzinfo:
+        dt = dt.replace(tzinfo=datetime.timezone.utc)
+    return dt.isoformat().replace("+00:00", "Z")
+
+
+def convert_date_to_iso(dt: datetime.date) -> str:  # pragma: no cover
+    """Handle datetime serialization for nested timestamps.
+
+    Args:
+        dt: The date to convert.
+
+    Returns:
+        The ISO formatted date string.
+    """
+    return dt.isoformat()