sqlakit 0.22.11__tar.gz → 0.22.12__tar.gz

{sqlakit-0.22.11 → sqlakit-0.22.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlakit
-Version: 0.22.11
+Version: 0.22.12
 Summary: A toolkit for SQLAlchemy applications.
 Keywords: sqlalchemy,database,orm,sql,asyncio
 Author: Anton Ruhlov
@@ -15,6 +15,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
 Classifier: Topic :: Database
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
@@ -387,7 +388,7 @@ there is nothing to await.
 ## `FastAPI` integration
 
 ```python
-from collections.abc import AsyncIterator
+from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
@@ -400,7 +401,7 @@ db = Database("postgresql+psycopg://localhost/app")
 
 
 @asynccontextmanager
-async def lifespan(app: FastAPI) -> AsyncIterator[None]:
+async def lifespan(app: FastAPI) -> AsyncGenerator[None]:
     yield
     await db.dispose()  # close the pool on shutdown
 

{sqlakit-0.22.11 → sqlakit-0.22.12}/README.md

@@ -356,7 +356,7 @@ there is nothing to await.
 ## `FastAPI` integration
 
 ```python
-from collections.abc import AsyncIterator
+from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
@@ -369,7 +369,7 @@ db = Database("postgresql+psycopg://localhost/app")
 
 
 @asynccontextmanager
-async def lifespan(app: FastAPI) -> AsyncIterator[None]:
+async def lifespan(app: FastAPI) -> AsyncGenerator[None]:
     yield
     await db.dispose()  # close the pool on shutdown
 

{sqlakit-0.22.11 → sqlakit-0.22.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sqlakit"
-version = "0.22.11"
+version = "0.22.12"
 description = "A toolkit for SQLAlchemy applications."
 readme = "README.md"
 license = "MIT"
@@ -21,6 +21,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
     "Programming Language :: Python :: 3.14",
+    "Programming Language :: Python :: 3.15",
     "Topic :: Database",
     "Topic :: Software Development :: Libraries :: Python Modules",
     "Typing :: Typed",
@@ -178,7 +179,7 @@ skip = "./.venv,./site,./uv.lock,./.git"
 [dependency-groups]
 dev = [
     "pytest>=9.1.1",
-    "ty>=0.0.84",
+    "ty>=0.0.87",
     "ruff>=0.16.6",
     "poethepoet>=0.48.0",
     "anyio>=4.14.2",

{sqlakit-0.22.11 → sqlakit-0.22.12}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "sqlakit"
-version = "0.22.11"
+version = "0.22.12"
 description = "A toolkit for SQLAlchemy applications."
 readme = "README.md"
 license = "MIT"
@@ -16,6 +16,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
     "Programming Language :: Python :: 3.14",
+    "Programming Language :: Python :: 3.15",
     "Topic :: Database",
     "Topic :: Software Development :: Libraries :: Python Modules",
     "Typing :: Typed",
@@ -54,7 +55,7 @@ default-groups = ["dev", "docs"]
 [dependency-groups]
 dev = [
     "pytest>=9.1.1",
-    "ty>=0.0.84",
+    "ty>=0.0.87",
     "ruff>=0.16.6",
     "poethepoet>=0.48.0",
     "anyio>=4.14.2",

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/_base.py

@@ -5,7 +5,7 @@ import inspect
 import random
 import threading
 import time
-from collections.abc import Callable, Mapping
+from collections.abc import Callable, Generator, Mapping
 from contextlib import ExitStack, contextmanager
 from contextvars import ContextVar
 from dataclasses import dataclass, replace
@@ -56,7 +56,7 @@ from .exceptions import (
 
 if TYPE_CHECKING:
     import logging
-    from collections.abc import Iterator, Sequence
+    from collections.abc import Sequence
     from os import PathLike
 
     from sqlalchemy.engine import Engine
@@ -364,7 +364,7 @@ class BaseDatabase(Generic[ConnectionT, SessionT]):
         skip_queries_from: Sequence[str | PathLike[str]] = (),
         into: Recording | None = None,
         send_to: Callable[[Recording], object] | None = None,
-    ) -> Iterator[Recording]:
+    ) -> Generator[Recording]:
         """Record the statements of this block, and what they add up to.
 
         ```python
@@ -430,7 +430,7 @@ class BaseDatabase(Generic[ConnectionT, SessionT]):
         *,
         at_most: int | None = None,
         duplicates: bool = True,
-    ) -> Iterator[Recording]:
+    ) -> Generator[Recording]:
         """Assert what the block asks of this database.
 
         ```python
@@ -568,7 +568,7 @@ class BaseDatabase(Generic[ConnectionT, SessionT]):
         return scope is not None and scope.session is not None
 
     @contextmanager
-    def unbound(self) -> Iterator[None]:
+    def unbound(self) -> Generator[None]:
         """Hide the block open around this one, for the code inside to open its own.
 
         A test opens a transaction for the whole test, so code that reaches for
@@ -644,7 +644,7 @@ class BaseDatabase(Generic[ConnectionT, SessionT]):
         return outer, savepoint or rollback or bool(outer and outer.savepoint)
 
     @contextmanager
-    def override(self, db: Self) -> Iterator[Self]:
+    def override(self, db: Self) -> Generator[Self]:
         """Use another database in place of this one until the block ends.
 
         ```python
@@ -944,7 +944,7 @@ class _DatabaseRegistryMixin(BaseDatabase[Any, Any], Generic[DatabaseT]):
         into: Recording | None = None,
         send_to: Callable[[Recording], object] | None = None,
         using: str | DatabaseT | Sequence[str | DatabaseT] | None = None,
-    ) -> Iterator[Recording]:
+    ) -> Generator[Recording]:
         """Record every database this registry has, not the default one alone.
 
         ```python
@@ -996,7 +996,7 @@ class _DatabaseRegistryMixin(BaseDatabase[Any, Any], Generic[DatabaseT]):
                     send_to(together)
 
     @contextmanager
-    def unbound(self) -> Iterator[None]:
+    def unbound(self) -> Generator[None]:
         """Hide the block open on every database this registry holds."""
         with ExitStack() as stack:
             for alias in self.aliases:
@@ -1024,7 +1024,7 @@ class _DatabaseRegistryMixin(BaseDatabase[Any, Any], Generic[DatabaseT]):
         db: DatabaseT,
         *,
         alias: str = DEFAULT_ALIAS,
-    ) -> Iterator[DatabaseT]:
+    ) -> Generator[DatabaseT]:
         """Use another database under an alias until the block ends.
 
         ```python

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/_db.py

@@ -29,7 +29,7 @@ from ._base import (
 from .exceptions import TransactionRolledBackError
 
 if TYPE_CHECKING:
-    from collections.abc import Callable, Iterator, Sequence
+    from collections.abc import Callable, Generator, Sequence
     from types import TracebackType
 
     from sqlalchemy.engine import Engine
@@ -151,7 +151,7 @@ class Database(BaseDatabase[sa.Connection, Session]):
         return cast("sa.Connection", scope.connection)
 
     @contextmanager
-    def connect(self) -> Iterator[sa.Connection]:
+    def connect(self) -> Generator[sa.Connection]:
         """Open a connection and bind it, or reuse the one already bound."""
         elsewhere = self._override
         if elsewhere is not None:
@@ -198,7 +198,7 @@ class Database(BaseDatabase[sa.Connection, Session]):
         return autocommit
 
     @contextmanager
-    def _autocommit(self) -> Iterator[sa.Connection]:
+    def _autocommit(self) -> Generator[sa.Connection]:
         """Open a connection in ``AUTOCOMMIT`` and bind it, or join the outer one."""
         elsewhere = self._override
         if elsewhere is not None:
@@ -328,7 +328,7 @@ class Database(BaseDatabase[sa.Connection, Session]):
         return transaction
 
     @contextmanager
-    def session_factory(self) -> Iterator[Session]:
+    def session_factory(self) -> Generator[Session]:
         """Open a session for the block, and bind it.
 
         The session arrives at once, the connection on its first query or
@@ -365,7 +365,7 @@ class Database(BaseDatabase[sa.Connection, Session]):
         *,
         commit: bool = False,
         autocommit: bool = False,
-    ) -> Iterator[sa.Connection]:
+    ) -> Generator[sa.Connection]:
         """Bind ``connection`` for the block, ending the session it opened.
 
         ``commit`` keeps that session's work. Closing a session that isolates
@@ -389,7 +389,7 @@ class Database(BaseDatabase[sa.Connection, Session]):
         metadata: sa.MetaData,
         *,
         tables: Sequence[sa.Table] | None = None,
-    ) -> Iterator[None]:
+    ) -> Generator[None]:
         """Create these tables here, and drop them when the block ends.
 
         A test session opens this once, around everything that needs a schema:

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/_model.py

@@ -53,7 +53,7 @@ DatabaseT = TypeVar("DatabaseT", bound="BaseDatabase[Any, Any]")
 class DatabaseSource(Protocol):
     """The registry an alias is looked up in: the importable one, or a stand-in."""
 
-    def __getitem__(self, alias: str) -> BaseDatabase[Any, Any]: ...
+    def __getitem__(self, alias: str, /) -> BaseDatabase[Any, Any]: ...
 
 
 class DatabaseDescriptor(Generic[DatabaseT]):

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/_registry.py

@@ -8,7 +8,7 @@ from ._base import _DatabaseRegistryMixin, default_backoff, late_bound
 from ._db import Database, RetryingTransaction
 
 if TYPE_CHECKING:
-    from collections.abc import Callable, Iterator
+    from collections.abc import Callable, Generator
 
     from ._base import RetryOn
     from ._db import Transaction
@@ -40,7 +40,7 @@ class Databases(_DatabaseRegistryMixin[Database], Database):
     def transactions(
         self,
         **arguments: Any,  # noqa: ANN401
-    ) -> Iterator[None]:
+    ) -> Generator[None]:
         """Open a transaction on every database, not the default one alone.
 
         A single database needs `transaction(rollback=True)`. A test harness

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/_sql.py

@@ -6,7 +6,7 @@ import importlib.util
 import inspect
 import itertools
 import re
-from collections.abc import Mapping, Sized
+from collections.abc import Generator, Mapping, Sized
 from contextlib import contextmanager
 from contextvars import ContextVar
 from dataclasses import dataclass, is_dataclass
@@ -49,7 +49,7 @@ from .exceptions import (
 )
 
 if TYPE_CHECKING:
-    from collections.abc import Callable, Iterable, Iterator, Sequence
+    from collections.abc import Callable, Iterable, Sequence
 
     from pydantic import BaseModel, TypeAdapter
     from sqlalchemy.sql import Executable
@@ -86,7 +86,7 @@ _calls_kept: ContextVar[bool] = ContextVar("sqlakit.calls_kept", default=False)
 
 
 @contextmanager
-def calls_kept() -> Iterator[None]:
+def calls_kept() -> Generator[None]:
     """Write a call that cannot be made with the values at hand as the call itself.
 
     An editor shows a template rendered with made-up values, or with none: a

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/asyncio/_db.py

@@ -37,7 +37,7 @@ from sqlakit._base import (
 from sqlakit.exceptions import MissingConnectionError, TransactionRolledBackError
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncIterator, Callable, Coroutine, Sequence
+    from collections.abc import AsyncGenerator, Callable, Coroutine, Sequence
     from types import TracebackType
 
     from sqlalchemy.ext.asyncio import AsyncTransaction
@@ -187,7 +187,7 @@ class Database(BaseDatabase[AsyncConnection, AsyncSession]):
         return cast("AsyncConnection", scope.connection)
 
     @asynccontextmanager
-    async def connect(self) -> AsyncIterator[AsyncConnection]:
+    async def connect(self) -> AsyncGenerator[AsyncConnection]:
         """Open a connection and bind it, or reuse the one already bound."""
         elsewhere = self._override
         if elsewhere is not None:
@@ -234,7 +234,7 @@ class Database(BaseDatabase[AsyncConnection, AsyncSession]):
         return autocommit
 
     @asynccontextmanager
-    async def _autocommit(self) -> AsyncIterator[AsyncConnection]:
+    async def _autocommit(self) -> AsyncGenerator[AsyncConnection]:
         """Open a connection in ``AUTOCOMMIT`` and bind it, or join the outer one."""
         elsewhere = self._override
         if elsewhere is not None:
@@ -364,7 +364,7 @@ class Database(BaseDatabase[AsyncConnection, AsyncSession]):
         return transaction
 
     @asynccontextmanager
-    async def session_factory(self) -> AsyncIterator[AsyncSession]:
+    async def session_factory(self) -> AsyncGenerator[AsyncSession]:
         """Open a session for the block, and bind it.
 
         The session arrives at once, the connection on its first query or
@@ -401,7 +401,7 @@ class Database(BaseDatabase[AsyncConnection, AsyncSession]):
         *,
         commit: bool = False,
         autocommit: bool = False,
-    ) -> AsyncIterator[AsyncConnection]:
+    ) -> AsyncGenerator[AsyncConnection]:
         """Bind ``connection`` for the block, ending the session it opened.
 
         ``commit`` keeps that session's work. Closing a session that isolates
@@ -425,7 +425,7 @@ class Database(BaseDatabase[AsyncConnection, AsyncSession]):
         metadata: sa.MetaData,
         *,
         tables: Sequence[sa.Table] | None = None,
-    ) -> AsyncIterator[None]:
+    ) -> AsyncGenerator[None]:
         """Create these tables here, and drop them when the block ends.
 
         A test session opens this once, around everything that needs a schema:

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/asyncio/_registry.py

@@ -9,7 +9,7 @@ from sqlakit._base import _DatabaseRegistryMixin, default_backoff, late_bound
 from ._db import Database, RetryingTransaction
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncIterator, Callable
+    from collections.abc import AsyncGenerator, Callable
 
     from sqlakit._base import RetryOn
 
@@ -41,7 +41,7 @@ class Databases(_DatabaseRegistryMixin[Database], Database):
     async def transactions(
         self,
         **arguments: Any,  # noqa: ANN401
-    ) -> AsyncIterator[None]:
+    ) -> AsyncGenerator[None]:
         """Open a transaction on every database, not the default one alone.
 
         A test harness with more than one database needs this, where a single

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/asyncio/orm.py

@@ -42,10 +42,16 @@ from sqlakit._query import (
 from sqlakit.exceptions import InstanceNotFoundError
 
 from ._db import Database
-from ._registry import Databases, db
+from ._registry import db
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncIterator, Iterable, Mapping, Sequence
+    from collections.abc import (
+        AsyncGenerator,
+        AsyncIterator,
+        Iterable,
+        Mapping,
+        Sequence,
+    )
 
     from sqlalchemy.engine import CursorResult, Result, ScalarResult
     from sqlalchemy.orm import InstrumentedAttribute
@@ -611,8 +617,8 @@ class QueryDescriptor(Generic[QueryT]):
     def __init__(self, query_class: type[QueryT]) -> None:
         self.query_class = query_class
 
-    # `ClassVar` cannot hold a type variable, so the base declares the query
-    # every model has and a read narrows it to the model it is read from.
+    # `QueryDescriptor(Query)` reads as a query of the model it is read from,
+    # any other query class as itself.
     @overload
     def __get__(
         self: QueryDescriptor[Query[Any]],
@@ -644,16 +650,18 @@ class ModelMixin(BaseModel[Database]):
     [`set_db`][sqlakit.asyncio.orm.ModelMixin.set_db].
     """
 
-    __db__: ClassVar[str | Database] = DEFAULT_ALIAS
-    __dbs__: ClassVar[Databases] = db
+    __db__ = DEFAULT_ALIAS
+    __dbs__ = db
 
-    query: ClassVar[QueryDescriptor[Query[Any]]] = QueryDescriptor(Query)
+    # Not annotated: a model's `query = UserQuery.as_descriptor()` would take
+    # the declared type and read as `Query[User]`.
+    query = QueryDescriptor(Query)
 
     db: ClassVar[DatabaseDescriptor[Database]] = DatabaseDescriptor()
 
     @classmethod
     @asynccontextmanager
-    async def provisioned_tables(cls, alias: str | None = None) -> AsyncIterator[None]:
+    async def provisioned_tables(cls, alias: str | None = None) -> AsyncGenerator[None]:
         """Create the tables that belong on this model's database, and drop them after.
 
         A test session opens this once, around everything that needs a schema:
@@ -668,7 +676,7 @@ class ModelMixin(BaseModel[Database]):
         With models on more than one database, name the alias, and each database
         gets the tables of the models pointed at it.
         """
-        db = cls.db if alias is None else cls.__dbs__[alias]
+        db = cls.db if alias is None else cls.dbs[alias]
         async with db.provisioned_tables(cls.metadata, tables=tables_for(cls, db)):
             yield
 

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/asyncio/sql.py

@@ -256,9 +256,7 @@ class SQLQuery(SQLRows[sa.Row[Any]]):
 
         """
         require_pydantic()
-        return cast(
-            "SQLRows[OtherT]", self._as(SQLRows, type_=type_, validation=validation)
-        )
+        return self._as(SQLRows, type_=type_, validation=validation)
 
     def scalars(self) -> SQLRows[Any]:
         """Read the first column of each row instead of whole rows.

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/orm.py

@@ -40,11 +40,11 @@ from ._query import (
     orderable,
     ordered,
 )
-from ._registry import Databases, db
+from ._registry import db
 from .exceptions import InstanceNotFoundError
 
 if TYPE_CHECKING:
-    from collections.abc import Iterable, Iterator, Mapping, Sequence
+    from collections.abc import Generator, Iterable, Iterator, Mapping, Sequence
 
     from sqlalchemy.engine import CursorResult, Result, ScalarResult
     from sqlalchemy.orm import InstrumentedAttribute
@@ -593,8 +593,8 @@ class QueryDescriptor(Generic[QueryT]):
     def __init__(self, query_class: type[QueryT]) -> None:
         self.query_class = query_class
 
-    # `ClassVar` cannot hold a type variable, so the base declares the query
-    # every model has and a read narrows it to the model it is read from.
+    # `QueryDescriptor(Query)` reads as a query of the model it is read from,
+    # any other query class as itself.
     @overload
     def __get__(
         self: QueryDescriptor[Query[Any]],
@@ -625,16 +625,18 @@ class ModelMixin(BaseModel[Database]):
     importable registry, or one set with [`set_db`][sqlakit.orm.ModelMixin.set_db].
     """
 
-    __db__: ClassVar[str | Database] = DEFAULT_ALIAS
-    __dbs__: ClassVar[Databases] = db
+    __db__ = DEFAULT_ALIAS
+    __dbs__ = db
 
-    query: ClassVar[QueryDescriptor[Query[Any]]] = QueryDescriptor(Query)
+    # Not annotated: a model's `query = UserQuery.as_descriptor()` would take
+    # the declared type and read as `Query[User]`.
+    query = QueryDescriptor(Query)
 
     db: ClassVar[DatabaseDescriptor[Database]] = DatabaseDescriptor()
 
     @classmethod
     @contextmanager
-    def provisioned_tables(cls, alias: str | None = None) -> Iterator[None]:
+    def provisioned_tables(cls, alias: str | None = None) -> Generator[None]:
         """Create the tables that belong on this model's database, and drop them after.
 
         A test session opens this once, around everything that needs a schema:
@@ -649,7 +651,7 @@ class ModelMixin(BaseModel[Database]):
         With models on more than one database, name the alias, and each database
         gets the tables of the models pointed at it.
         """
-        db = cls.db if alias is None else cls.__dbs__[alias]
+        db = cls.db if alias is None else cls.dbs[alias]
         with db.provisioned_tables(cls.metadata, tables=tables_for(cls, db)):
             yield
 

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/pytest_plugin.py

@@ -55,7 +55,7 @@ from ._registry import db as importable_db
 from .exceptions import UnknownDatabaseError
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncIterator, Iterator, Sequence
+    from collections.abc import AsyncIterator, Generator, Iterator, Sequence
 
     import sqlalchemy as sa
 
@@ -346,7 +346,7 @@ def assert_queries(sqlakit_db: Any) -> AssertQueries:  # noqa: ANN401
         at_most: int | None = None,
         duplicates: bool = True,
         using: Any = None,  # noqa: ANN401
-    ) -> Iterator[Recording]:
+    ) -> Generator[Recording]:
         require_expectation(count, at_most, duplicates)
         recording = Recording()
         with ExitStack() as stack:
@@ -402,7 +402,7 @@ async def _sqlakit_async_transaction(
 
 
 @contextmanager
-def _reported(request: pytest.FixtureRequest, db: Any) -> Iterator[None]:  # noqa: ANN401
+def _reported(request: pytest.FixtureRequest, db: Any) -> Generator[None]:  # noqa: ANN401
     """Record what the test runs, for the report, when one was asked for.
 
     The recording is named after the test and filed under the file it lives in,
@@ -435,7 +435,7 @@ def _named(node: pytest.Item) -> str:
 
 
 @contextmanager
-def _entered(blocks: list[Any]) -> Iterator[None]:
+def _entered(blocks: list[Any]) -> Generator[None]:
     """Hold these blocks open, awaiting the ones that need it.
 
     An `asyncio` database builds its schema in a coroutine, and this fixture is

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/sql.py

@@ -246,9 +246,7 @@ class SQLQuery(SQLRows[sa.Row[Any]]):
 
         """
         require_pydantic()
-        return cast(
-            "SQLRows[OtherT]", self._as(SQLRows, type_=type_, validation=validation)
-        )
+        return self._as(SQLRows, type_=type_, validation=validation)
 
     def scalars(self) -> SQLRows[Any]:
         """Read the first column of each row instead of whole rows.

{sqlakit-0.22.11 → sqlakit-0.22.12}/sqlakit/testing.py

@@ -12,7 +12,7 @@ from .exceptions import UnknownDatabaseError
 from .types import AssertQueries
 
 if TYPE_CHECKING:
-    from collections.abc import Iterator
+    from collections.abc import Generator
 
 __all__ = ["AssertQueries", "assert_queries"]
 
@@ -24,7 +24,7 @@ def assert_queries(
     at_most: int | None = None,
     duplicates: bool = True,
     using: str | Any = None,  # noqa: ANN401
-) -> Iterator[Recording]:
+) -> Generator[Recording]:
     """Assert what the block asks of the database.
 
     ```python