Typing for MigrationLoader and MigrationRecorder (not ideal)

Author: davegaeddert

plain-models/plain/models/migrations/executor.py

@@ -4,6 +4,8 @@
 from contextlib import nullcontext
 from typing import TYPE_CHECKING, Any
 
+from plain.models.connections import DatabaseConnection
+
 from ..transaction import atomic
 from .loader import MigrationLoader
 from .migration import Migration
@@ -22,7 +24,7 @@ class MigrationExecutor:
 
     def __init__(
         self,
-        connection: BaseDatabaseWrapper,
+        connection: BaseDatabaseWrapper | DatabaseConnection,
         progress_callback: Callable[..., Any] | None = None,
     ) -> None:
         self.connection = connection

plain-models/plain/models/migrations/loader.py

@@ -317,7 +317,9 @@ def build_graph(self) -> None:
             raise
         self.graph.ensure_not_cyclic()
 
-    def check_consistent_history(self, connection: BaseDatabaseWrapper) -> None:
+    def check_consistent_history(
+        self, connection: BaseDatabaseWrapper | DatabaseConnection
+    ) -> None:
         """
         Raise InconsistentMigrationHistory if any applied migrations have
         unapplied dependencies.