bazaar-contrib-storage 0.3.0rc1__tar.gz

bazaar_contrib_storage-0.3.0rc1/LICENSE

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

bazaar_contrib_storage-0.3.0rc1/PKG-INFO

@@ -0,0 +1,34 @@
+Metadata-Version: 2.4
+Name: bazaar-contrib-storage
+Version: 0.3.0rc1
+Summary: Shared explicit storage contracts and backend selection
+License-Expression: Apache-2.0
+License-File: LICENSE
+Requires-Dist: bazaar-core~=0.3.0rc1
+Requires-Dist: pydantic>=2.12.0
+Requires-Python: >=3.15
+Project-URL: Repository, https://github.com/yuchanns/bazaar-compute-node
+Description-Content-Type: text/markdown
+
+# bazaar-contrib-storage
+
+Enable backend registration and select the database backend. Install the selected provider as well.
+
+```sh
+bcn contrib add storage --enable
+```
+
+The corresponding entry in `config.toml`:
+
+```toml
+version = "5"
+
+[[contribs]]
+package = "bazaar-contrib-storage~=0.3.0rc1"
+[[contribs.enabled]]
+entry = "storage"
+[[contribs.enabled]]
+entry = "database"
+[contribs.enabled.config]
+backend = "sqlite"
+```

bazaar_contrib_storage-0.3.0rc1/README.md

@@ -0,0 +1,22 @@
+# bazaar-contrib-storage
+
+Enable backend registration and select the database backend. Install the selected provider as well.
+
+```sh
+bcn contrib add storage --enable
+```
+
+The corresponding entry in `config.toml`:
+
+```toml
+version = "5"
+
+[[contribs]]
+package = "bazaar-contrib-storage~=0.3.0rc1"
+[[contribs.enabled]]
+entry = "storage"
+[[contribs.enabled]]
+entry = "database"
+[contribs.enabled.config]
+backend = "sqlite"
+```

bazaar_contrib_storage-0.3.0rc1/pyproject.toml

@@ -0,0 +1,33 @@
+[project]
+name = "bazaar-contrib-storage"
+version = "0.3.0rc1"
+description = "Shared explicit storage contracts and backend selection"
+readme = "README.md"
+license = "Apache-2.0"
+license-files = ["LICENSE"]
+requires-python = ">=3.15"
+dependencies = [
+    "bazaar-core~=0.3.0rc1",
+    "pydantic>=2.12.0",
+]
+
+[project.urls]
+Repository = "https://github.com/yuchanns/bazaar-compute-node"
+
+[project.entry-points."bazaar_compute_node.contribs"]
+storage = "bazaar_contrib_storage:storage"
+database = "bazaar_contrib_storage:database"
+
+[build-system]
+requires = ["uv_build>=0.13.0,<0.14.0"]
+build-backend = "uv_build"
+
+[tool.bcn.ci]
+skip = true
+
+[tool.pyright]
+extends = "../../pyrightconfig.json"
+include = ["src"]
+
+[tool.ruff]
+extend = "../../pyproject.toml"

bazaar_contrib_storage-0.3.0rc1/pyproject.toml.orig

@@ -0,0 +1,30 @@
+[project]
+name = "bazaar-contrib-storage"
+version = "0.3.0rc1"
+description = "Shared explicit storage contracts and backend selection"
+readme = "README.md"
+license = "Apache-2.0"
+license-files = ["LICENSE"]
+requires-python = ">=3.15"
+dependencies = ["bazaar-core~=0.3.0rc1", "pydantic>=2.12.0"]
+
+[project.urls]
+Repository = "https://github.com/yuchanns/bazaar-compute-node"
+
+[project.entry-points."bazaar_compute_node.contribs"]
+storage = "bazaar_contrib_storage:storage"
+database = "bazaar_contrib_storage:database"
+
+[build-system]
+requires = ["uv_build>=0.13.0,<0.14.0"]
+build-backend = "uv_build"
+
+[tool.bcn.ci]
+skip = true
+
+[tool.pyright]
+extends = "../../pyrightconfig.json"
+include = ["src"]
+
+[tool.ruff]
+extend = "../../pyproject.toml"

bazaar_contrib_storage-0.3.0rc1/src/bazaar_contrib_storage/__init__.py

@@ -0,0 +1,135 @@
+"""Explicit queries, schema operations, and interchangeable storage services."""
+
+from .contracts import (
+    DB,
+    STORAGE,
+    BackendName,
+    BackendRegistry,
+    Database,
+    DatabaseConfig,
+    MigrationError,
+    Session,
+    StorageClosedError,
+    StorageError,
+    StorageIntegrityError,
+    StorageQueryError,
+    StorageQueryTimeoutError,
+    StorageUnavailableError,
+    backend_key,
+    database,
+    storage,
+)
+from .models import (
+    AddColumn,
+    All,
+    AnyOf,
+    Binary,
+    Call,
+    Case,
+    Column,
+    ColumnDefinition,
+    Compare,
+    Conflict,
+    Contains,
+    Copy,
+    CreateIndex,
+    CreateTable,
+    CreateTrigger,
+    Delete,
+    DropColumn,
+    DropIndex,
+    DropTable,
+    DropTrigger,
+    Expression,
+    ForeignKey,
+    Identifier,
+    In,
+    Index,
+    Insert,
+    IsNull,
+    Join,
+    Mutation,
+    Not,
+    Predicate,
+    Record,
+    Reference,
+    RenameColumn,
+    RenameTable,
+    RowId,
+    RowNumber,
+    Scalar,
+    SchemaOperation,
+    Search,
+    Select,
+    Sort,
+    Subquery,
+    Table,
+    Update,
+    Value,
+)
+
+__all__ = [
+    "DB",
+    "STORAGE",
+    "AddColumn",
+    "All",
+    "AnyOf",
+    "BackendName",
+    "BackendRegistry",
+    "Binary",
+    "Call",
+    "Case",
+    "Column",
+    "ColumnDefinition",
+    "Compare",
+    "Conflict",
+    "Contains",
+    "Copy",
+    "CreateIndex",
+    "CreateTable",
+    "CreateTrigger",
+    "Database",
+    "DatabaseConfig",
+    "Delete",
+    "DropColumn",
+    "DropIndex",
+    "DropTable",
+    "DropTrigger",
+    "Expression",
+    "ForeignKey",
+    "Identifier",
+    "In",
+    "Index",
+    "Insert",
+    "IsNull",
+    "Join",
+    "MigrationError",
+    "Mutation",
+    "Not",
+    "Predicate",
+    "Record",
+    "Reference",
+    "RenameColumn",
+    "RenameTable",
+    "RowId",
+    "RowNumber",
+    "Scalar",
+    "SchemaOperation",
+    "Search",
+    "Select",
+    "Session",
+    "Sort",
+    "StorageClosedError",
+    "StorageError",
+    "StorageIntegrityError",
+    "StorageQueryError",
+    "StorageQueryTimeoutError",
+    "StorageUnavailableError",
+    "Subquery",
+    "Table",
+    "Update",
+    "Value",
+    "backend_key",
+    "database",
+    "storage",
+]

bazaar_contrib_storage-0.3.0rc1/src/bazaar_contrib_storage/contracts.py

@@ -0,0 +1,154 @@
+"""Storage services and lifespan-bound backend registration."""
+
+from __future__ import annotations
+
+from contextlib import asynccontextmanager, contextmanager
+from functools import cache
+from typing import Annotated, Protocol
+lazy from collections.abc import Awaitable, Callable, Iterator
+
+from bazaar_core import Context, Contrib, ServiceKey
+from pydantic import BaseModel, ConfigDict, Field, TypeAdapter
+
+lazy from .models import (
+    Copy,
+    Delete,
+    Insert,
+    Mutation,
+    Record,
+    SchemaOperation,
+    Select,
+    Update,
+)
+
+type BackendName = Annotated[str, Field(pattern=r"^[A-Za-z_][A-Za-z0-9_-]*$")]
+_NAME = TypeAdapter(BackendName)
+
+
+class StorageError(RuntimeError):
+    """A backend could not complete a public storage operation."""
+
+
+class StorageClosedError(StorageError):
+    """The backend has finished its lifecycle."""
+
+
+class StorageUnavailableError(StorageError):
+    """The underlying data source is unavailable or busy."""
+
+
+class StorageIntegrityError(StorageError):
+    """A write violates the declared data constraints."""
+
+
+class StorageQueryError(StorageError):
+    """The data source could not execute the declared operation."""
+
+
+class StorageQueryTimeoutError(TimeoutError, StorageError):
+    """The data source interrupted a query after its execution budget."""
+
+
+class MigrationError(StorageError):
+    """The stored migration history cannot be safely advanced."""
+
+
+class Session(Protocol):
+    async def table_exists(self, name: str) -> bool: ...
+
+    async def select(self, request: Select) -> tuple[Record, ...]: ...
+
+    async def copy(self, request: Copy) -> Mutation: ...
+
+    async def insert(self, request: Insert) -> Mutation: ...
+
+    async def update(self, request: Update) -> Mutation: ...
+
+    async def delete(self, request: Delete) -> Mutation: ...
+
+    async def apply_schema(self, operations: tuple[SchemaOperation, ...]) -> None: ...
+
+
+class Database(Session, Protocol):
+    async def read[T](
+        self,
+        operation: Callable[[Session], Awaitable[T]],
+        *,
+        snapshot: bool = False,
+        query_budget: float | None = None,
+    ) -> T:
+        """Run on one reader, optionally holding its original snapshot and budget."""
+        ...
+
+    async def write[T](
+        self,
+        operation: Callable[[Session], Awaitable[T]],
+        *,
+        transactional: bool = False,
+    ) -> T:
+        """Run one writer operation with its declared transaction boundary."""
+        ...
+
+    async def transaction[T](self, operation: Callable[[Session], Awaitable[T]]) -> T:
+        """Commit the operation together, rolling it back on failure or cancellation."""
+        ...
+
+
+class BackendRegistry:
+    def __init__(self) -> None:
+        self._backends: dict[str, Database] = {}
+
+    @property
+    def names(self) -> tuple[str, ...]:
+        return tuple(self._backends)
+
+    def get(self, name: str) -> Database:
+        return self._backends[name]
+
+    @contextmanager
+    def register(self, name: str, database: Database) -> Iterator[None]:
+        name = _NAME.validate_python(name)
+        if name in self._backends:
+            raise StorageError(
+                f"storage backend {name!r} is already registered; choose a unique backend name"
+            )
+        self._backends[name] = database
+        try:
+            yield
+        finally:
+            del self._backends[name]
+
+
+STORAGE = ServiceKey[BackendRegistry]("storage.backends")
+DB = ServiceKey[Database]("storage.database")
+
+
+@cache
+def backend_key(name: str) -> ServiceKey[Database]:
+    return ServiceKey[Database](f"storage.backend.{_NAME.validate_python(name)}")
+
+
+class DatabaseConfig(BaseModel):
+    model_config = ConfigDict(extra="forbid", frozen=True)
+    backend: BackendName
+
+
+@asynccontextmanager
+async def _storage(ctx: Context):
+    ctx.offer(STORAGE, BackendRegistry())
+    yield
+
+
+@asynccontextmanager
+async def _database(ctx: Context, config: DatabaseConfig):
+    ctx.offer(DB, ctx.accept(STORAGE).get(config.backend))
+    yield
+
+
+storage = Contrib(offers=[STORAGE], lifespan=_storage)
+database = Contrib(
+    config=DatabaseConfig,
+    accepts=lambda config: [STORAGE, backend_key(config.backend)],
+    offers=[DB],
+    lifespan=_database,
+)

bazaar_contrib_storage-0.3.0rc1/src/bazaar_contrib_storage/models.py

@@ -0,0 +1,410 @@
+"""Backend-independent data, query, and schema descriptions."""
+
+from __future__ import annotations
+
+from typing import Annotated, Literal, Self
+
+from pydantic import (
+    BaseModel,
+    ConfigDict,
+    Field,
+    StrictBool,
+    StrictBytes,
+    StrictFloat,
+    StrictInt,
+    StrictStr,
+    model_validator,
+)
+
+type Identifier = Annotated[
+    str, Field(pattern=r"^[A-Za-z_][A-Za-z0-9_]*$", max_length=63)
+]
+type Scalar = StrictStr | StrictInt | StrictFloat | StrictBool | StrictBytes | None
+type Record = dict[str, Scalar]
+
+
+class Model(BaseModel):
+    model_config = ConfigDict(
+        extra="forbid", frozen=True, allow_inf_nan=False, ser_json_bytes="base64"
+    )
+
+
+class Column(Model):
+    kind: Literal["column"] = "column"
+    name: Identifier
+    table: Identifier | None = None
+
+
+class RowId(Model):
+    """The provider's physical row identity within the current table."""
+
+    kind: Literal["row_id"] = "row_id"
+    table: Identifier | None = None
+
+
+class Value(Model):
+    kind: Literal["value"] = "value"
+    value: Scalar
+
+
+class Binary(Model):
+    kind: Literal["binary"] = "binary"
+    operator: Literal["+", "-", "*", "/", "%", "||"]
+    left: Expression
+    right: Expression
+
+
+class Call(Model):
+    kind: Literal["call"] = "call"
+    name: Literal[
+        "count", "sum", "min", "max", "avg", "coalesce", "lower", "length", "replace"
+    ]
+    arguments: tuple[Expression, ...] = ()
+
+
+class RowNumber(Model):
+    kind: Literal["row_number"] = "row_number"
+    order_by: Annotated[tuple[Sort, ...], Field(min_length=1)]
+    partition_by: tuple[Expression, ...] = ()
+
+
+class Case(Model):
+    kind: Literal["case"] = "case"
+    when: Predicate
+    then: Expression
+    otherwise: Expression
+
+
+class Subquery(Model):
+    kind: Literal["subquery"] = "subquery"
+    query: Select
+
+
+type Expression = Annotated[
+    Column | RowId | Value | Binary | Call | RowNumber | Case | Subquery,
+    Field(discriminator="kind"),
+]
+
+
+class Compare(Model):
+    kind: Literal["compare"] = "compare"
+    left: Expression
+    operator: Literal["=", "!=", "<", "<=", ">", ">="]
+    right: Expression
+
+
+class In(Model):
+    kind: Literal["in"] = "in"
+    expression: Expression
+    values: tuple[Scalar, ...]
+    exclude: bool = False
+
+
+class IsNull(Model):
+    kind: Literal["null"] = "null"
+    expression: Expression
+    exclude: bool = False
+
+
+class Contains(Model):
+    kind: Literal["contains"] = "contains"
+    expression: Expression
+    text: str
+
+
+class All(Model):
+    kind: Literal["all"] = "all"
+    conditions: tuple[Predicate, ...]
+
+
+class AnyOf(Model):
+    kind: Literal["any"] = "any"
+    conditions: tuple[Predicate, ...]
+
+
+class Not(Model):
+    kind: Literal["not"] = "not"
+    condition: Predicate
+
+
+class Search(Model):
+    """Require every literal term to occur in at least one indexed column."""
+
+    kind: Literal["search"] = "search"
+    index: Identifier
+    table: Identifier
+    columns: Annotated[tuple[Identifier, ...], Field(min_length=1)]
+    terms: Annotated[
+        tuple[Annotated[str, Field(min_length=1)], ...], Field(min_length=1)
+    ]
+    row_key: Identifier | None = None
+
+
+type Predicate = Annotated[
+    Compare | In | IsNull | Contains | All | AnyOf | Not | Search,
+    Field(discriminator="kind"),
+]
+
+
+class Sort(Model):
+    expression: Expression
+    descending: bool = False
+    nulls: Literal["first", "last"] = "last"
+
+
+class Join(Model):
+    table: Identifier
+    alias: Identifier | None = None
+    kind: Literal["inner", "left"] = "inner"
+    on: Predicate
+
+
+class Select(Model):
+    table: Identifier
+    ctes: dict[Identifier, Select] = Field(default_factory=dict)
+    alias: Identifier | None = None
+    columns: dict[Identifier, Expression] = Field(default_factory=dict)
+    where: Predicate | None = None
+    joins: tuple[Join, ...] = ()
+    group_by: tuple[Expression, ...] = ()
+    having: Predicate | None = None
+    order_by: tuple[Sort, ...] = ()
+    limit: Annotated[int, Field(ge=0)] | None = None
+    offset: Annotated[int, Field(ge=0)] = 0
+    distinct: bool = False
+
+
+class Copy(Model):
+    """Insert the explicitly named projection of a query into another table."""
+
+    table: Identifier
+    query: Select
+
+    @model_validator(mode="after")
+    def projection(self) -> Self:
+        if not self.query.columns:
+            raise ValueError("copy queries must explicitly name their target columns")
+        return self
+
+
+class Conflict(Model):
+    columns: Annotated[tuple[Identifier, ...], Field(min_length=1)]
+    update: tuple[Identifier, ...] = ()
+
+
+class Insert(Model):
+    table: Identifier
+    rows: Annotated[tuple[dict[Identifier, Scalar], ...], Field(min_length=1)]
+    conflict: Conflict | None = None
+    returning: tuple[Identifier, ...] = ()
+
+    @model_validator(mode="after")
+    def columns(self) -> Self:
+        columns = self.rows[0].keys()
+        if not columns or any(row.keys() != columns for row in self.rows):
+            raise ValueError("insert rows must contain the same non-empty column set")
+        return self
+
+
+class Update(Model):
+    table: Identifier
+    values: Annotated[dict[Identifier, Expression], Field(min_length=1)]
+    where: Predicate | None = None
+    returning: tuple[Identifier, ...] = ()
+
+
+class Delete(Model):
+    table: Identifier
+    where: Predicate | None = None
+    returning: tuple[Identifier, ...] = ()
+
+
+class Mutation(Model):
+    count: Annotated[int, Field(ge=0)]
+    rows: tuple[Record, ...] = ()
+
+
+class Reference(Model):
+    table: Identifier
+    column: Identifier
+    on_delete: Literal["restrict", "cascade", "set null"] = "restrict"
+
+
+class ColumnDefinition(Model):
+    name: Identifier
+    type: Literal["text", "integer", "real", "blob"]
+    nullable: bool = False
+    default: Scalar = None
+    generated: bool = False
+    unique: bool = False
+    check: Predicate | None = None
+    references: Reference | None = None
+
+
+class ForeignKey(Model):
+    columns: Annotated[tuple[Identifier, ...], Field(min_length=1)]
+    table: Identifier
+    references: Annotated[tuple[Identifier, ...], Field(min_length=1)]
+    on_delete: Literal["restrict", "cascade", "set null"] = "restrict"
+
+
+class Table(Model):
+    name: Identifier
+    columns: Annotated[tuple[ColumnDefinition, ...], Field(min_length=1)]
+    primary_key: tuple[Identifier, ...] = ()
+    foreign_keys: tuple[ForeignKey, ...] = ()
+    unique: tuple[tuple[Identifier, ...], ...] = ()
+    checks: tuple[Predicate, ...] = ()
+
+    @model_validator(mode="after")
+    def definitions(self) -> Self:
+        columns = {column.name for column in self.columns}
+        if len(columns) != len(self.columns) or not set(self.primary_key) <= columns:
+            raise ValueError("table columns must be unique and contain its primary key")
+        if any(not key or not set(key) <= columns for key in self.unique):
+            raise ValueError("unique keys must contain columns belonging to the table")
+        for column in self.columns:
+            if column.generated and (
+                column.type != "integer" or self.primary_key != (column.name,)
+            ):
+                raise ValueError("a generated integer must be the sole primary key")
+        for key in self.foreign_keys:
+            if not set(key.columns) <= columns or len(key.columns) != len(
+                key.references
+            ):
+                raise ValueError(
+                    "foreign key columns must belong to the table and match references"
+                )
+        return self
+
+
+class Index(Model):
+    name: Identifier
+    columns: tuple[Identifier, ...] = ()
+    order_by: tuple[Sort, ...] = ()
+    unique: bool = False
+    kind: Literal["btree", "search"] = "btree"
+    where: Predicate | None = None
+    row_key: Identifier | None = None
+
+    @model_validator(mode="after")
+    def search_options(self) -> Self:
+        if not self.columns and not self.order_by:
+            raise ValueError("an index requires columns or expressions")
+        if self.kind == "search" and (
+            self.unique or self.where is not None or self.order_by or not self.columns
+        ):
+            raise ValueError(
+                "search indexes cover their full columns without uniqueness"
+            )
+        return self
+
+
+class CreateTable(Model):
+    kind: Literal["create_table"] = "create_table"
+    table: Table
+    if_not_exists: bool = False
+    temporary: bool = False
+
+
+class AddColumn(Model):
+    kind: Literal["add_column"] = "add_column"
+    table: Identifier
+    column: ColumnDefinition
+
+
+class CreateIndex(Model):
+    kind: Literal["create_index"] = "create_index"
+    table: Identifier
+    index: Index
+    if_not_exists: bool = False
+
+
+class DropIndex(Model):
+    kind: Literal["drop_index"] = "drop_index"
+    table: Identifier
+    name: Identifier
+
+
+class DropTable(Model):
+    kind: Literal["drop_table"] = "drop_table"
+    table: Identifier
+    if_exists: bool = False
+
+
+class RenameTable(Model):
+    kind: Literal["rename_table"] = "rename_table"
+    table: Identifier
+    name: Identifier
+
+
+class RenameColumn(Model):
+    kind: Literal["rename_column"] = "rename_column"
+    table: Identifier
+    column: Identifier
+    name: Identifier
+
+
+class DropColumn(Model):
+    kind: Literal["drop_column"] = "drop_column"
+    table: Identifier
+    column: Identifier
+
+
+class CreateTrigger(Model):
+    kind: Literal["create_trigger"] = "create_trigger"
+    name: Identifier
+    table: Identifier
+    event: Literal["insert", "update", "delete"]
+    when: Predicate | None = None
+    action: Update
+
+    @model_validator(mode="after")
+    def statement(self) -> Self:
+        if self.action.table != self.table or self.action.returning:
+            raise ValueError("a trigger updates its own table without returning rows")
+        return self
+
+
+class DropTrigger(Model):
+    kind: Literal["drop_trigger"] = "drop_trigger"
+    name: Identifier
+    table: Identifier
+    if_exists: bool = False
+
+
+type SchemaOperation = Annotated[
+    CreateTable
+    | AddColumn
+    | CreateIndex
+    | DropIndex
+    | DropTable
+    | RenameTable
+    | RenameColumn
+    | DropColumn
+    | CreateTrigger
+    | DropTrigger,
+    Field(discriminator="kind"),
+]
+
+
+for model in (
+    Binary,
+    Call,
+    RowNumber,
+    Case,
+    Subquery,
+    Compare,
+    In,
+    IsNull,
+    Contains,
+    All,
+    AnyOf,
+    Not,
+    Select,
+    Join,
+    Sort,
+    Update,
+    Index,
+):
+    model.model_rebuild()