plain.models 0.47.0__py3-none-any.whl → 0.48.0__py3-none-any.whl

plain/models/CHANGELOG.md

@@ -1,5 +1,16 @@
 # plain-models changelog
 
+## [0.48.0](https://github.com/dropseed/plain/releases/plain-models@0.48.0) (2025-09-26)
+
+### What's changed
+
+- Migrations now run in a single transaction by default for databases that support transactional DDL, providing all-or-nothing migration batches for better safety and consistency ([6d0c105](https://github.com/dropseed/plain/commit/6d0c105fa9))
+- Added `--atomic-batch/--no-atomic-batch` options to `plain migrate` to explicitly control whether migrations are run in a single transaction ([6d0c105](https://github.com/dropseed/plain/commit/6d0c105fa9))
+
+### Upgrade instructions
+
+- No changes required
+
 ## [0.47.0](https://github.com/dropseed/plain/releases/plain-models@0.47.0) (2025-09-25)
 
 ### What's changed

plain/models/backends/base/creation.py

@@ -59,6 +59,7 @@ class BaseDatabaseCreation:
             prune=False,
             no_input=True,
             verbosity=max(verbosity - 1, 0),
+            atomic_batch=False,  # No need for atomic batch when creating test database
         )
 
         # Ensure a connection for the side effect of initializing the test database.

plain/models/cli.py

@@ -362,6 +362,11 @@ def makemigrations(package_labels, dry_run, empty, no_input, name, check, verbos
     default=1,
     help="Verbosity level; 0=minimal output, 1=normal output, 2=verbose output, 3=very verbose output",
 )
+@click.option(
+    "--atomic-batch/--no-atomic-batch",
+    default=None,
+    help="Run migrations in a single transaction (auto-detected by default)",
+)
 def migrate(
     package_label,
     migration_name,
@@ -372,6 +377,7 @@ def migrate(
     prune,
     no_input,
     verbosity,
+    atomic_batch,
 ):
     """Updates database schema. Manages both packages with migrations and those without."""
 
@@ -575,6 +581,65 @@ def migrate(
     # pprint(sql)
 
     if migration_plan:
+        # Determine whether to use atomic batch
+        use_atomic_batch = False
+        if len(migration_plan) > 1:
+            # Check database capabilities
+            can_rollback_ddl = db_connection.features.can_rollback_ddl
+
+            # Check if all migrations support atomic
+            non_atomic_migrations = [m for m in migration_plan if not m.atomic]
+
+            if atomic_batch is True:
+                # User explicitly requested atomic batch
+                if not can_rollback_ddl:
+                    raise click.UsageError(
+                        f"--atomic-batch not supported on {db_connection.vendor}. "
+                        "Remove the flag or use a database that supports transactional DDL."
+                    )
+                if non_atomic_migrations:
+                    names = ", ".join(
+                        f"{m.package_label}.{m.name}" for m in non_atomic_migrations[:3]
+                    )
+                    if len(non_atomic_migrations) > 3:
+                        names += f", and {len(non_atomic_migrations) - 3} more"
+                    raise click.UsageError(
+                        f"--atomic-batch requested but these migrations have atomic=False: {names}"
+                    )
+                use_atomic_batch = True
+                if verbosity >= 1:
+                    click.echo(
+                        f"  Running {len(migration_plan)} migrations in atomic batch (all-or-nothing)"
+                    )
+            elif atomic_batch is False:
+                # User explicitly disabled atomic batch
+                use_atomic_batch = False
+                if verbosity >= 1:
+                    click.echo(f"  Running {len(migration_plan)} migrations separately")
+            else:
+                # Auto-detect (atomic_batch is None)
+                if can_rollback_ddl and not non_atomic_migrations:
+                    use_atomic_batch = True
+                    if verbosity >= 1:
+                        click.echo(
+                            f"  Running {len(migration_plan)} migrations in atomic batch (all-or-nothing)"
+                        )
+                else:
+                    use_atomic_batch = False
+                    if verbosity >= 1:
+                        if not can_rollback_ddl:
+                            click.echo(
+                                f"  Running {len(migration_plan)} migrations separately ({db_connection.vendor} doesn't support batch transactions)"
+                            )
+                        elif non_atomic_migrations:
+                            click.echo(
+                                f"  Running {len(migration_plan)} migrations separately (some migrations have atomic=False)"
+                            )
+                        else:
+                            click.echo(
+                                f"  Running {len(migration_plan)} migrations separately"
+                            )
+
         if backup or (backup is None and settings.DEBUG):
             backup_name = f"migrate_{time.strftime('%Y%m%d_%H%M%S')}"
             click.secho(
@@ -599,6 +664,7 @@ def migrate(
             plan=migration_plan,
             state=pre_migrate_state.clone(),
             fake=fake,
+            atomic_batch=use_atomic_batch,
         )
         # post_migrate signals have access to all models. Ensure that all models
         # are reloaded in case any are delayed.

plain/models/migrations/executor.py

@@ -1,3 +1,6 @@
+from contextlib import nullcontext
+
+from ..transaction import atomic
 from .loader import MigrationLoader
 from .recorder import MigrationRecorder
 from .state import ProjectState
@@ -52,12 +55,14 @@ class MigrationExecutor:
                     migration.mutate_state(state, preserve=False)
         return state
 
-    def migrate(self, targets, plan=None, state=None, fake=False):
+    def migrate(self, targets, plan=None, state=None, fake=False, atomic_batch=False):
         """
         Migrate the database up to the given targets.
 
         Plain first needs to create all project states before a migration is
         (un)applied and in a second step run all the database operations.
+
+        atomic_batch: Whether to run all migrations in a single transaction.
         """
         # The plain_migrations table must be present to record applied
         # migrations, but don't create it if there are no migrations to apply.
@@ -82,34 +87,31 @@ class MigrationExecutor:
             if state is None:
                 # The resulting state should still include applied migrations.
                 state = self._create_project_state(with_applied_migrations=True)
-            state = self._migrate_all_forwards(state, plan, full_plan, fake=fake)
-
-        self.check_replacements()
 
-        return state
+            migrations_to_run = set(plan)
+
+            # Choose context manager based on atomic_batch
+            batch_context = atomic if (atomic_batch and len(plan) > 1) else nullcontext
+
+            with batch_context():
+                for migration in full_plan:
+                    if not migrations_to_run:
+                        # We remove every migration that we applied from these sets so
+                        # that we can bail out once the last migration has been applied
+                        # and don't always run until the very end of the migration
+                        # process.
+                        break
+                    if migration in migrations_to_run:
+                        if "models_registry" not in state.__dict__:
+                            if self.progress_callback:
+                                self.progress_callback("render_start")
+                            state.models_registry  # Render all -- performance critical
+                            if self.progress_callback:
+                                self.progress_callback("render_success")
+                        state = self.apply_migration(state, migration, fake=fake)
+                        migrations_to_run.remove(migration)
 
-    def _migrate_all_forwards(self, state, plan, full_plan, fake):
-        """
-        Take a list of 2-tuples of the form (migration instance, False) and
-        apply them in the order they occur in the full_plan.
-        """
-        migrations_to_run = set(plan)
-        for migration in full_plan:
-            if not migrations_to_run:
-                # We remove every migration that we applied from these sets so
-                # that we can bail out once the last migration has been applied
-                # and don't always run until the very end of the migration
-                # process.
-                break
-            if migration in migrations_to_run:
-                if "models_registry" not in state.__dict__:
-                    if self.progress_callback:
-                        self.progress_callback("render_start")
-                    state.models_registry  # Render all -- performance critical
-                    if self.progress_callback:
-                        self.progress_callback("render_success")
-                state = self.apply_migration(state, migration, fake=fake)
-                migrations_to_run.remove(migration)
+        self.check_replacements()
 
         return state
 

{plain_models-0.47.0.dist-info → plain_models-0.48.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plain.models
-Version: 0.47.0
+Version: 0.48.0
 Summary: Model your data and store it in a database.
 Author-email: Dave Gaeddert <dave.gaeddert@dropseed.dev>
 License-File: LICENSE

{plain_models-0.47.0.dist-info → plain_models-0.48.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
 plain/models/AGENTS.md,sha256=xQQW-z-DehnCUyjiGSBfLqUjoSUdo_W1b0JmwYmWieA,209
-plain/models/CHANGELOG.md,sha256=JMkQ7LwLIp63_HtRIwOMdIpveUrhHvIBJbvBDaNG9k4,17633
+plain/models/CHANGELOG.md,sha256=_EJXRX89opgWTq_FnrLrs2Q7BOKxbrINWsYU4qwZIqQ,18245
 plain/models/README.md,sha256=lqzWJrEIxBCHC1P8X1YoRjbsMFlu0-kG4ujP76B_ZO4,8572
 plain/models/__init__.py,sha256=aB9HhIKBh0iK3LZztInAE-rDF-yKsdfcjfMtwtN5vnI,2920
 plain/models/aggregates.py,sha256=P0mhsMl1VZt2CVHMuCHnNI8SxZ9citjDLEgioN6NOpo,7240
 plain/models/base.py,sha256=4NkK-gkcxbIWZm7c9Q51eDwLClIieWZvlRg6l7vJOxE,65692
-plain/models/cli.py,sha256=lhvlw7DVOU3CVoMUpHaj7qIbQ2d6qtSm_l9PdCeCnc0,36404
+plain/models/cli.py,sha256=rqrgG__OyqtaDaKLB6XZnS6Zv7esU9K9EAyKCO35McA,39548
 plain/models/config.py,sha256=-m15VY1ZJKWdPGt-5i9fnMvz9LBzPfSRgWmWeEV8Dao,382
 plain/models/connections.py,sha256=RBNa2FZ0x3C9un6PaYL-IYzH_OesRSpdHNGKvYHGiOM,2276
 plain/models/constants.py,sha256=ndnj9TOTKW0p4YcIPLOLEbsH6mOgFi6B1-rIzr_iwwU,210
@@ -34,7 +34,7 @@ plain/models/backends/utils.py,sha256=VN9b_hnGeLqndVAcCx00X7KhFC6jY2Tn6J3E62HqDE
 plain/models/backends/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plain/models/backends/base/base.py,sha256=I5040pUAe7-yNTb9wfMNOYjeiFQLZ5dcRI4rtmVKJ04,26457
 plain/models/backends/base/client.py,sha256=90Ffs6zZYCli3tJjwsPH8TItZ8tz1Pp-zhQa-EpsNqc,937
-plain/models/backends/base/creation.py,sha256=H-xx667pLediQ19HDmE_mP-b03tDKQriBmWPKigJ5t0,9332
+plain/models/backends/base/creation.py,sha256=JlLnboF5reo-s6RM35EpdWBqsRv63wcJSxX5igIPPUM,9420
 plain/models/backends/base/features.py,sha256=1AehdhpeC7VobhggwpeXIt7HJNY2EWY700j4gYX8Xjs,7856
 plain/models/backends/base/introspection.py,sha256=8icKf9h8y4kobmyrbo8JWTDcpQIAt4oS_4FtCnY7FqQ,6815
 plain/models/backends/base/operations.py,sha256=tzuw7AgVb3aZn2RHamReGNco7wIt4UBkAHx8HebDZ_s,25652
@@ -89,7 +89,7 @@ plain/models/functions/window.py,sha256=3S0QIZc_pIVcWpE5Qq-OxixmtATLb8rZrWkvCfVt
 plain/models/migrations/__init__.py,sha256=ZAQUGrfr_OxYMIO7vUBIHLs_M3oZ4iQSjDzCHRFUdtI,96
 plain/models/migrations/autodetector.py,sha256=MuJEVaU9IZ8c6HfcDsEANNujUrSGMN-iDd5WDPM7ht8,60957
 plain/models/migrations/exceptions.py,sha256=_bGjIMaBP2Py9ePUxUhiH0p1zXrQM4JhJO4lWfyF8-g,1044
-plain/models/migrations/executor.py,sha256=2_1bWM7Dp3s8z6PADAEN-Y0KnIiRzAqsUkn_nRQl5TA,6757
+plain/models/migrations/executor.py,sha256=4Hsqs6UV5I_Z4PesNi86YPvJyzPcJUTyrxHxGJu2zf0,6907
 plain/models/migrations/graph.py,sha256=nrztu_8dU0wAUSxKUqqFWpvZcSQxGEqE6dXWkPytmCU,12570
 plain/models/migrations/loader.py,sha256=qUTmaEYI1_mV6goQPQYZKjSz8rMbE6G1wqvrAsmuGwA,16464
 plain/models/migrations/migration.py,sha256=22YwRHnaRnCkBpW5p7K89tAU6h4QSsG5yiq-o7W-cSI,6505
@@ -115,8 +115,8 @@ plain/models/sql/where.py,sha256=ezE9Clt2BmKo-I7ARsgqZ_aVA-1UdayCwr6ULSWZL6c,126
 plain/models/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plain/models/test/pytest.py,sha256=KD5-mxonBxOYIhUh9Ql5uJOIiC9R4t-LYfb6sjA0UdE,3486
 plain/models/test/utils.py,sha256=S3d6zf3OFWDxB_kBJr0tDvwn51bjwDVWKPumv37N-p8,467
-plain_models-0.47.0.dist-info/METADATA,sha256=oBEWDFezcMc1KxtVoQIMSEr15Oy7NB43aLiAGurmYfg,8884
-plain_models-0.47.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-plain_models-0.47.0.dist-info/entry_points.txt,sha256=IYJAW9MpL3PXyXFWmKmALagAGXC_5rzBn2eEGJlcV04,112
-plain_models-0.47.0.dist-info/licenses/LICENSE,sha256=m0D5O7QoH9l5Vz_rrX_9r-C8d9UNr_ciK6Qwac7o6yo,3175
-plain_models-0.47.0.dist-info/RECORD,,
+plain_models-0.48.0.dist-info/METADATA,sha256=rE7jXG5-QLDFTyIJQnTb9QGt6Z23o_LZ16soPfLKvlI,8884
+plain_models-0.48.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+plain_models-0.48.0.dist-info/entry_points.txt,sha256=IYJAW9MpL3PXyXFWmKmALagAGXC_5rzBn2eEGJlcV04,112
+plain_models-0.48.0.dist-info/licenses/LICENSE,sha256=m0D5O7QoH9l5Vz_rrX_9r-C8d9UNr_ciK6Qwac7o6yo,3175
+plain_models-0.48.0.dist-info/RECORD,,