From: Michael Bayer <mike_mp@zzzcomputing.com>
Date: Wed, 20 Dec 2023 13:53:58 +0000 (+0000)
Subject: Merge "Improve `Rewriter` implementation" into main
X-Git-Tag: rel_1_13_1~2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=cc0e4dbf2d7f7c5c35fb3b8a80feff50a588122e;p=thirdparty%2Fsqlalchemy%2Falembic.git

Merge "Improve `Rewriter` implementation" into main
---

cc0e4dbf2d7f7c5c35fb3b8a80feff50a588122e
diff --cc alembic/autogenerate/rewriter.py
index 02ff431c,3efb499b..8994dcf8
--- a/alembic/autogenerate/rewriter.py
+++ b/alembic/autogenerate/rewriter.py
@@@ -21,10 -20,13 +21,14 @@@ if TYPE_CHECKING
      from ..operations.ops import MigrationScript
      from ..operations.ops import ModifyTableOps
      from ..operations.ops import OpContainer
 -    from ..runtime.environment import _GetRevArg
 +    from ..operations.ops import UpgradeOps
      from ..runtime.migration import MigrationContext
 +    from ..script.revision import _GetRevArg
  
+ ProcessRevisionDirectiveFn = Callable[
+     ["MigrationContext", "_GetRevArg", List["MigrationScript"]], None
+ ]
+ 
  
  class Rewriter:
      """A helper object that allows easy 'rewriting' of ops streams.