plain.postgres 0.96.0__tar.gz → 0.97.0__tar.gz

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plain.postgres
-Version: 0.96.0
+Version: 0.97.0
 Summary: Model your data and store it in a database.
 Author-email: Dave Gaeddert <dave.gaeddert@dropseed.dev>
 License-Expression: BSD-3-Clause
@@ -82,26 +82,63 @@ admin_users = User.query.filter(is_admin=True)
 
 ## Database connection
 
-To connect to a database, you can provide a `DATABASE_URL` environment variable:
+Configure the database with a single URL. The canonical Plain setting is `POSTGRES_URL`:
+
+```python
+# app/settings.py
+POSTGRES_URL = "postgresql://user:password@localhost:5432/dbname"
+```
+
+Or via environment variable:
+
+```sh
+PLAIN_POSTGRES_URL=postgresql://user:password@localhost:5432/dbname
+```
+
+Plain also reads the `DATABASE_URL` environment variable as a fallback — it's the widely-used convention for Postgres connection strings, so most hosting setups work without extra configuration:
 
 ```sh
 DATABASE_URL=postgresql://user:password@localhost:5432/dbname
 ```
 
-Or you can set the individual `POSTGRES_*` settings (via `PLAIN_POSTGRES_*` environment variables or in `app/settings.py`):
+Precedence (highest to lowest): `PLAIN_POSTGRES_URL` → `POSTGRES_URL` in `settings.py` → `DATABASE_URL` environment variable.
 
-```python
-# app/settings.py
-POSTGRES_HOST = "localhost"
-POSTGRES_PORT = 5432
-POSTGRES_DATABASE = "dbname"
-POSTGRES_USER = "user"
-POSTGRES_PASSWORD = "password"
+The URL supports any libpq connection parameter as a query string — for example `?sslmode=require&application_name=web&connect_timeout=10`. These are parsed and passed through to the driver.
+
+To explicitly disable the database (e.g. during Docker builds where no database is available), set the URL to the string `none`:
+
+```sh
+PLAIN_POSTGRES_URL=none
 ```
 
-If `DATABASE_URL` is set, it takes priority and the individual connection settings are parsed from it.
+### Bypassing a connection pooler for management operations
+
+Transaction-mode poolers (PlanetScale, Supabase's pooler, Neon's pooler, standalone pgbouncer in transaction mode) can't run DDL, long transactions, or `pg_dump`. To work around this, set a second URL that management commands use to reach Postgres directly:
+
+```sh
+PLAIN_POSTGRES_URL=postgresql://app@pooler:6432/myapp
+PLAIN_POSTGRES_MANAGEMENT_URL=postgresql://app@postgres:5432/myapp
+```
+
+When `POSTGRES_MANAGEMENT_URL` is set, these commands connect through it instead of `POSTGRES_URL`:
+
+- `plain migrations create`, `plain migrations apply`, `plain migrations list`, `plain migrations prune`, `plain migrations squash`
+- `plain postgres sync`, `plain postgres converge`, `plain postgres schema`
+- `plain postgres diagnose`, `plain postgres drop-unknown-tables`, `plain postgres shell`
+
+When it's unset, all commands use `POSTGRES_URL` — there's no behavior change for existing apps.
+
+To route custom code through the management connection, use the `use_management_connection()` context manager:
+
+```python
+from plain.postgres import use_management_connection
+
+with use_management_connection():
+    # Any get_connection() / ORM calls inside this block use POSTGRES_MANAGEMENT_URL.
+    run_custom_schema_change()
+```
 
-To explicitly disable the database (e.g. during Docker builds where no database is available), set `DATABASE_URL=none`.
+You _can_ point the two URLs at different Postgres roles — e.g. a least-privilege DML role for runtime and a DDL-capable role for management. Plain does not currently automate the grant/ownership plumbing that split requires (default privileges for newly-created tables, ownership reassignment, preflight checks that the runtime role can see the schema). If you adopt that pattern, you're responsible for wiring those up yourself.
 
 **PostgreSQL is the only supported database.** You need to install a PostgreSQL driver separately — [psycopg](https://www.psycopg.org/) is recommended:
 
@@ -1206,27 +1243,18 @@ These are static, code-level checks that catch issues before you deploy. The `di
 
 ## Settings
 
-Connection settings are configured via `DATABASE_URL` or individual `POSTGRES_*` settings.
-
-When `DATABASE_URL` is set, it is parsed into the individual connection settings automatically. When `DATABASE_URL` is not set, the connection settings are required individually.
-
-Set `DATABASE_URL=none` to explicitly disable the database (e.g. during Docker image builds).
-
-| Setting                                  | Type          | Default | Env var                                        |
-| ---------------------------------------- | ------------- | ------- | ---------------------------------------------- |
-| `POSTGRES_HOST`                          | `str`         | —       | `PLAIN_POSTGRES_HOST`                          |
-| `POSTGRES_PORT`                          | `int \| None` | `None`  | `PLAIN_POSTGRES_PORT`                          |
-| `POSTGRES_DATABASE`                      | `str`         | —       | `PLAIN_POSTGRES_DATABASE`                      |
-| `POSTGRES_USER`                          | `str`         | —       | `PLAIN_POSTGRES_USER`                          |
-| `POSTGRES_PASSWORD`                      | `Secret[str]` | —       | `PLAIN_POSTGRES_PASSWORD`                      |
-| `POSTGRES_CONN_MAX_AGE`                  | `int`         | `600`   | `PLAIN_POSTGRES_CONN_MAX_AGE`                  |
-| `POSTGRES_CONN_HEALTH_CHECKS`            | `bool`        | `True`  | `PLAIN_POSTGRES_CONN_HEALTH_CHECKS`            |
-| `POSTGRES_OPTIONS`                       | `dict`        | `{}`    | —                                              |
-| `POSTGRES_TIME_ZONE`                     | `str \| None` | `None`  | `PLAIN_POSTGRES_TIME_ZONE`                     |
-| `POSTGRES_MIGRATION_LOCK_TIMEOUT`        | `str`         | `"3s"`  | `PLAIN_POSTGRES_MIGRATION_LOCK_TIMEOUT`        |
-| `POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   | `str`         | `"3s"`  | `PLAIN_POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   |
-| `POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      | `str`         | `"3s"`  | `PLAIN_POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      |
-| `POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` | `str`         | `"3s"`  | `PLAIN_POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` |
+The connection is configured with a single URL (`POSTGRES_URL`). `DATABASE_URL` is read as a platform-compat fallback. Set the URL to `none` to explicitly disable the database (e.g. during Docker image builds).
+
+| Setting                                  | Type          | Default                 | Env var                                        |
+| ---------------------------------------- | ------------- | ----------------------- | ---------------------------------------------- |
+| `POSTGRES_URL`                           | `Secret[str]` | `$DATABASE_URL` or `""` | `PLAIN_POSTGRES_URL`                           |
+| `POSTGRES_MANAGEMENT_URL`                | `Secret[str]` | `""`                    | `PLAIN_POSTGRES_MANAGEMENT_URL`                |
+| `POSTGRES_CONN_MAX_AGE`                  | `int`         | `600`                   | `PLAIN_POSTGRES_CONN_MAX_AGE`                  |
+| `POSTGRES_CONN_HEALTH_CHECKS`            | `bool`        | `True`                  | `PLAIN_POSTGRES_CONN_HEALTH_CHECKS`            |
+| `POSTGRES_MIGRATION_LOCK_TIMEOUT`        | `str`         | `"3s"`                  | `PLAIN_POSTGRES_MIGRATION_LOCK_TIMEOUT`        |
+| `POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   | `str`         | `"3s"`                  | `PLAIN_POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   |
+| `POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      | `str`         | `"3s"`                  | `PLAIN_POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      |
+| `POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` | `str`         | `"3s"`                  | `PLAIN_POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` |
 
 See [`default_settings.py`](./default_settings.py) for more details.
 

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/CHANGELOG.md

@@ -1,5 +1,35 @@
 # plain-postgres changelog
 
+## [0.97.0](https://github.com/dropseed/plain/releases/plain-postgres@0.97.0) (2026-04-21)
+
+### What's changed
+
+- **Replaced individual `POSTGRES_*` connection fields with a single `POSTGRES_URL` setting.** `POSTGRES_HOST`, `POSTGRES_PORT`, `POSTGRES_DATABASE`, `POSTGRES_USER`, `POSTGRES_PASSWORD`, `POSTGRES_OPTIONS`, and `POSTGRES_TIME_ZONE` are gone — configure the connection with one URL (e.g. `postgresql://user:pass@host:5432/db?sslmode=require`). `DATABASE_URL` is still read as a fallback. Set the URL to `none` to explicitly disable the database (e.g. during Docker image builds). ([770a74606463](https://github.com/dropseed/plain/commit/770a74606463))
+- **Added `POSTGRES_MANAGEMENT_URL` for routing DDL through a separate connection.** When set, `plain migrations create|apply|list|prune|squash`, `plain postgres sync|converge|schema|diagnose|drop-unknown-tables|shell` connect through this URL instead of `POSTGRES_URL`. Use it to bypass transaction-mode poolers (PlanetScale, Supabase's pooler, Neon's pooler, pgbouncer) for schema changes, long transactions, and `pg_dump`. A new `use_management_connection()` context manager routes custom code through the same connection. When unset, all commands use `POSTGRES_URL` — no behavior change for existing apps. ([d1cc9630d049](https://github.com/dropseed/plain/commit/d1cc9630d049))
+- **Extracted the test-database lifecycle off `DatabaseConnection`.** Test setup/teardown now lives in `plain.postgres.test` instead of coupling it to the runtime connection class. ([ea67f82c746c](https://github.com/dropseed/plain/commit/ea67f82c746c))
+- **Removed thin psycopg re-export wrappers.** Internal code now imports directly from `psycopg` rather than the redundant Plain-level passthroughs. ([d1cb74100e0d](https://github.com/dropseed/plain/commit/d1cb74100e0d))
+
+### Upgrade instructions
+
+- **Replace individual `POSTGRES_*` settings with `POSTGRES_URL`** in `app/settings.py` (or `PLAIN_POSTGRES_URL` in the environment). For example:
+
+    ```python
+    # Before
+    POSTGRES_HOST = "localhost"
+    POSTGRES_PORT = 5432
+    POSTGRES_DATABASE = "myapp"
+    POSTGRES_USER = "app"
+    POSTGRES_PASSWORD = "secret"
+
+    # After
+    POSTGRES_URL = "postgresql://app:secret@localhost:5432/myapp"
+    ```
+
+    Apps that already set `DATABASE_URL` in the environment don't need any change.
+
+- **If `POSTGRES_OPTIONS` or `POSTGRES_TIME_ZONE` were set**, move them into the URL as query parameters (e.g. `?application_name=web&timezone=UTC`).
+- **If you run behind a transaction-mode pooler**, consider setting `POSTGRES_MANAGEMENT_URL` to a direct-to-Postgres connection string so `plain migrations` and `plain postgres sync` can issue DDL.
+
 ## [0.96.0](https://github.com/dropseed/plain/releases/plain-postgres@0.96.0) (2026-04-17)
 
 ### What's changed

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/README.md

@@ -70,26 +70,63 @@ admin_users = User.query.filter(is_admin=True)
 
 ## Database connection
 
-To connect to a database, you can provide a `DATABASE_URL` environment variable:
+Configure the database with a single URL. The canonical Plain setting is `POSTGRES_URL`:
+
+```python
+# app/settings.py
+POSTGRES_URL = "postgresql://user:password@localhost:5432/dbname"
+```
+
+Or via environment variable:
+
+```sh
+PLAIN_POSTGRES_URL=postgresql://user:password@localhost:5432/dbname
+```
+
+Plain also reads the `DATABASE_URL` environment variable as a fallback — it's the widely-used convention for Postgres connection strings, so most hosting setups work without extra configuration:
 
 ```sh
 DATABASE_URL=postgresql://user:password@localhost:5432/dbname
 ```
 
-Or you can set the individual `POSTGRES_*` settings (via `PLAIN_POSTGRES_*` environment variables or in `app/settings.py`):
+Precedence (highest to lowest): `PLAIN_POSTGRES_URL` → `POSTGRES_URL` in `settings.py` → `DATABASE_URL` environment variable.
 
-```python
-# app/settings.py
-POSTGRES_HOST = "localhost"
-POSTGRES_PORT = 5432
-POSTGRES_DATABASE = "dbname"
-POSTGRES_USER = "user"
-POSTGRES_PASSWORD = "password"
+The URL supports any libpq connection parameter as a query string — for example `?sslmode=require&application_name=web&connect_timeout=10`. These are parsed and passed through to the driver.
+
+To explicitly disable the database (e.g. during Docker builds where no database is available), set the URL to the string `none`:
+
+```sh
+PLAIN_POSTGRES_URL=none
 ```
 
-If `DATABASE_URL` is set, it takes priority and the individual connection settings are parsed from it.
+### Bypassing a connection pooler for management operations
+
+Transaction-mode poolers (PlanetScale, Supabase's pooler, Neon's pooler, standalone pgbouncer in transaction mode) can't run DDL, long transactions, or `pg_dump`. To work around this, set a second URL that management commands use to reach Postgres directly:
+
+```sh
+PLAIN_POSTGRES_URL=postgresql://app@pooler:6432/myapp
+PLAIN_POSTGRES_MANAGEMENT_URL=postgresql://app@postgres:5432/myapp
+```
+
+When `POSTGRES_MANAGEMENT_URL` is set, these commands connect through it instead of `POSTGRES_URL`:
+
+- `plain migrations create`, `plain migrations apply`, `plain migrations list`, `plain migrations prune`, `plain migrations squash`
+- `plain postgres sync`, `plain postgres converge`, `plain postgres schema`
+- `plain postgres diagnose`, `plain postgres drop-unknown-tables`, `plain postgres shell`
+
+When it's unset, all commands use `POSTGRES_URL` — there's no behavior change for existing apps.
+
+To route custom code through the management connection, use the `use_management_connection()` context manager:
+
+```python
+from plain.postgres import use_management_connection
+
+with use_management_connection():
+    # Any get_connection() / ORM calls inside this block use POSTGRES_MANAGEMENT_URL.
+    run_custom_schema_change()
+```
 
-To explicitly disable the database (e.g. during Docker builds where no database is available), set `DATABASE_URL=none`.
+You _can_ point the two URLs at different Postgres roles — e.g. a least-privilege DML role for runtime and a DDL-capable role for management. Plain does not currently automate the grant/ownership plumbing that split requires (default privileges for newly-created tables, ownership reassignment, preflight checks that the runtime role can see the schema). If you adopt that pattern, you're responsible for wiring those up yourself.
 
 **PostgreSQL is the only supported database.** You need to install a PostgreSQL driver separately — [psycopg](https://www.psycopg.org/) is recommended:
 
@@ -1194,27 +1231,18 @@ These are static, code-level checks that catch issues before you deploy. The `di
 
 ## Settings
 
-Connection settings are configured via `DATABASE_URL` or individual `POSTGRES_*` settings.
-
-When `DATABASE_URL` is set, it is parsed into the individual connection settings automatically. When `DATABASE_URL` is not set, the connection settings are required individually.
-
-Set `DATABASE_URL=none` to explicitly disable the database (e.g. during Docker image builds).
-
-| Setting                                  | Type          | Default | Env var                                        |
-| ---------------------------------------- | ------------- | ------- | ---------------------------------------------- |
-| `POSTGRES_HOST`                          | `str`         | —       | `PLAIN_POSTGRES_HOST`                          |
-| `POSTGRES_PORT`                          | `int \| None` | `None`  | `PLAIN_POSTGRES_PORT`                          |
-| `POSTGRES_DATABASE`                      | `str`         | —       | `PLAIN_POSTGRES_DATABASE`                      |
-| `POSTGRES_USER`                          | `str`         | —       | `PLAIN_POSTGRES_USER`                          |
-| `POSTGRES_PASSWORD`                      | `Secret[str]` | —       | `PLAIN_POSTGRES_PASSWORD`                      |
-| `POSTGRES_CONN_MAX_AGE`                  | `int`         | `600`   | `PLAIN_POSTGRES_CONN_MAX_AGE`                  |
-| `POSTGRES_CONN_HEALTH_CHECKS`            | `bool`        | `True`  | `PLAIN_POSTGRES_CONN_HEALTH_CHECKS`            |
-| `POSTGRES_OPTIONS`                       | `dict`        | `{}`    | —                                              |
-| `POSTGRES_TIME_ZONE`                     | `str \| None` | `None`  | `PLAIN_POSTGRES_TIME_ZONE`                     |
-| `POSTGRES_MIGRATION_LOCK_TIMEOUT`        | `str`         | `"3s"`  | `PLAIN_POSTGRES_MIGRATION_LOCK_TIMEOUT`        |
-| `POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   | `str`         | `"3s"`  | `PLAIN_POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   |
-| `POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      | `str`         | `"3s"`  | `PLAIN_POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      |
-| `POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` | `str`         | `"3s"`  | `PLAIN_POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` |
+The connection is configured with a single URL (`POSTGRES_URL`). `DATABASE_URL` is read as a platform-compat fallback. Set the URL to `none` to explicitly disable the database (e.g. during Docker image builds).
+
+| Setting                                  | Type          | Default                 | Env var                                        |
+| ---------------------------------------- | ------------- | ----------------------- | ---------------------------------------------- |
+| `POSTGRES_URL`                           | `Secret[str]` | `$DATABASE_URL` or `""` | `PLAIN_POSTGRES_URL`                           |
+| `POSTGRES_MANAGEMENT_URL`                | `Secret[str]` | `""`                    | `PLAIN_POSTGRES_MANAGEMENT_URL`                |
+| `POSTGRES_CONN_MAX_AGE`                  | `int`         | `600`                   | `PLAIN_POSTGRES_CONN_MAX_AGE`                  |
+| `POSTGRES_CONN_HEALTH_CHECKS`            | `bool`        | `True`                  | `PLAIN_POSTGRES_CONN_HEALTH_CHECKS`            |
+| `POSTGRES_MIGRATION_LOCK_TIMEOUT`        | `str`         | `"3s"`                  | `PLAIN_POSTGRES_MIGRATION_LOCK_TIMEOUT`        |
+| `POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   | `str`         | `"3s"`                  | `PLAIN_POSTGRES_MIGRATION_STATEMENT_TIMEOUT`   |
+| `POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      | `str`         | `"3s"`                  | `PLAIN_POSTGRES_CONVERGENCE_LOCK_TIMEOUT`      |
+| `POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` | `str`         | `"3s"`                  | `PLAIN_POSTGRES_CONVERGENCE_STATEMENT_TIMEOUT` |
 
 See [`default_settings.py`](./default_settings.py) for more details.
 

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/__init__.py

@@ -6,7 +6,7 @@ from . import (
 # Imports that would create circular imports if sorted
 from .base import Model
 from .constraints import CheckConstraint, UniqueConstraint
-from .db import get_connection
+from .db import get_connection, use_management_connection
 from .deletion import CASCADE, NO_ACTION, RESTRICT, SET_NULL
 from .expressions import F
 from .enums import TextChoices
@@ -104,6 +104,7 @@ __all__ = [
     "ReverseManyToMany",
     # From db
     "get_connection",
+    "use_management_connection",
     # From registry
     "register_model",
     "models_registry",

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/cli/converge.py

@@ -5,6 +5,7 @@ import sys
 import click
 
 from ..convergence import execute_plan, plan_convergence
+from .decorators import database_management_command
 
 
 @click.command()
@@ -14,6 +15,7 @@ from ..convergence import execute_plan, plan_convergence
     is_flag=True,
     help="Skip confirmation prompt.",
 )
+@database_management_command
 def converge(yes: bool) -> None:
     """Fix schema mismatches between models and the database.
 

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/cli/core.py

@@ -13,6 +13,7 @@ from plain.cli import register_cli
 from ..db import get_connection
 from ..dialect import quote_name
 from .converge import converge
+from .decorators import database_management_command
 from .diagnose import diagnose
 from .schema import schema
 from .sync import sync
@@ -32,6 +33,7 @@ cli.add_command(sync)
 
 @cli.command()
 @click.argument("parameters", nargs=-1)
+@database_management_command
 def shell(parameters: tuple[str, ...]) -> None:
     """Open an interactive database shell"""
     conn = get_connection()
@@ -67,6 +69,7 @@ def shell(parameters: tuple[str, ...]) -> None:
     is_flag=True,
     help="Skip confirmation prompt.",
 )
+@database_management_command
 def drop_unknown_tables(yes: bool) -> None:
     """Drop all tables not associated with a Plain model"""
     from ..introspection import get_unknown_tables

plain_postgres-0.97.0/plain/postgres/cli/decorators.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+import functools
+from collections.abc import Callable
+from typing import Any
+
+from ..db import use_management_connection
+
+
+def database_management_command[F: Callable[..., Any]](f: F) -> F:
+    """Run a click command's body through `use_management_connection()`.
+
+    Apply to CLI commands that perform schema changes, migrations, or other
+    database management operations. Inside the command, `get_connection()`
+    returns a connection opened against `POSTGRES_MANAGEMENT_URL` (falling
+    back to `POSTGRES_URL` when unset).
+    """
+
+    @functools.wraps(f)
+    def wrapper(*args: Any, **kwargs: Any) -> Any:
+        with use_management_connection():
+            return f(*args, **kwargs)
+
+    return wrapper  # ty: ignore[invalid-return-type]

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/cli/diagnose.py

@@ -8,6 +8,7 @@ import click
 
 from ..db import get_connection
 from ..introspection import CheckItem, CheckResult, build_table_owners, run_all_checks
+from .decorators import database_management_command
 
 STATUS_SYMBOLS = {
     "ok": ("✓", "green"),
@@ -182,6 +183,7 @@ def format_human(
 @click.option(
     "--all", "show_all", is_flag=True, help="Include package issues in detail"
 )
+@database_management_command
 def diagnose(output_json: bool, show_all: bool) -> None:
     """Run health checks against the database"""
     conn = get_connection()

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/cli/migrations.py

@@ -27,6 +27,7 @@ from ..migrations.recorder import MigrationRecorder
 from ..migrations.state import ModelState, ProjectState
 from ..migrations.writer import MigrationWriter
 from ..registry import models_registry
+from .decorators import database_management_command
 
 if TYPE_CHECKING:
     from ..connection import DatabaseConnection
@@ -68,6 +69,7 @@ def cli() -> None:
     default=1,
     help="Verbosity level; 0=minimal output, 1=normal output, 2=verbose output, 3=very verbose output",
 )
+@database_management_command
 def create(
     package_labels: tuple[str, ...],
     dry_run: bool,
@@ -345,6 +347,7 @@ def create(
     is_flag=True,
     help="Suppress migration output (used for test database creation).",
 )
+@database_management_command
 def apply(
     package_label: str | None,
     migration_name: str | None,
@@ -638,6 +641,7 @@ def apply(
     default=1,
     help="Verbosity level; 0=minimal output, 1=normal output, 2=verbose output, 3=very verbose output",
 )
+@database_management_command
 def list_migrations(
     package_labels: tuple[str, ...], format: str, verbosity: int
 ) -> None:
@@ -777,6 +781,7 @@ def list_migrations(
     is_flag=True,
     help="Skip confirmation prompt.",
 )
+@database_management_command
 def prune(yes: bool) -> None:
     """Remove stale migration records from the database"""
     # Load migrations from disk and database
@@ -899,6 +904,7 @@ def prune(yes: bool) -> None:
     default=1,
     help="Verbosity level; 0=minimal output, 1=normal output, 2=verbose output, 3=very verbose output",
 )
+@database_management_command
 def squash(
     package_label: str,
     start_migration_name: str | None,

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/cli/schema.py

@@ -10,6 +10,7 @@ from ..convergence.planning import can_auto_fix
 from ..db import get_connection
 from ..introspection import MANAGED_CONSTRAINT_TYPES, get_unknown_tables
 from ..registry import models_registry
+from .decorators import database_management_command
 
 
 def _ok() -> None:
@@ -108,6 +109,7 @@ def _render_model(analysis: ModelAnalysis) -> None:
 @click.command()
 @click.argument("model_label", required=False)
 @click.option("--json", "output_json", is_flag=True, help="Output as JSON")
+@database_management_command
 def schema(model_label: str | None, output_json: bool) -> None:
     """Show database schema from models, compared against the actual database"""
     models = models_registry.get_models()

{plain_postgres-0.96.0 → plain_postgres-0.97.0}/plain/postgres/cli/sync.py

@@ -7,6 +7,7 @@ import click
 from plain.runtime import settings
 
 from ..convergence import execute_plan, plan_convergence
+from .decorators import database_management_command
 
 
 @click.command()
@@ -15,6 +16,7 @@ from ..convergence import execute_plan, plan_convergence
     is_flag=True,
     help="Exit with non-zero status if sync would make any database changes.",
 )
+@database_management_command
 def sync(check: bool) -> None:
     """Sync the database schema with models.