refactor: add revision context to AutogenerateDiffsDetected so that w…

…rappers may make other formatting of the diff

fixes: #1597

alembic/command.py

@@ -298,7 +298,8 @@ def retrieve_migrations(rev, context):
 
     if diffs:
         raise util.AutogenerateDiffsDetected(
-            f"New upgrade operations detected: {diffs}"
+            f"New upgrade operations detected: {diffs}",
+            revision_context=revision_context,
         )
     else:
         config.print_stdout("No new upgrade operations detected.")

alembic/util/exc.py

@@ -1,6 +1,16 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from alembic.autogenerate import RevisionContext
+
+
 class CommandError(Exception):
     pass
 
 
 class AutogenerateDiffsDetected(CommandError):
-    pass
+    def __init__(
+        self, message: str, revision_context: "RevisionContext"
+    ) -> None:
+        super().__init__(message)
+        self.revision_context = revision_context

docs/build/unreleased/issue1597.rst

@@ -0,0 +1,6 @@
+ .. change::
+        :tags: check, autogenerate
+        :tickets: 1597
+
+        Add revision context to AutogenerateDiffsDetected so that command can be wrapped and diffs may be output in a different format.
+        Pull request courtesy of Louis-Amaury Chaib (@lachaib).