🚸 fix(alembic): better check diff output

Author: ProgramRipper

nonebot_plugin_orm/exception.py

@@ -0,0 +1,10 @@
+from alembic import util
+from click import ClickException
+
+
+class CommandError(ClickException, util.CommandError):
+    pass
+
+
+class AutogenerateDiffsDetected(CommandError, util.AutogenerateDiffsDetected):
+    pass

nonebot_plugin_orm/migrate.py

@@ -23,7 +23,6 @@
 from sqlalchemy import MetaData, Connection
 from alembic.util.editor import open_in_editor
 from alembic.script import Script, ScriptDirectory
-from alembic.util import AutogenerateDiffsDetected
 from alembic.util.langhelpers import rev_id as _rev_id
 from alembic.operations.ops import UpgradeOps, DowngradeOps
 from alembic.migration import StampStep, RevisionStep, MigrationContext
@@ -34,6 +33,7 @@
     render_python_code,
 )
 
+from .exception import AutogenerateDiffsDetected
 from .utils import is_editable, get_parent_plugins, return_progressbar
 
 if sys.version_info >= (3, 11):