🐛 fix(alembic): startup prompt ugly abort

nonebot · May 22, 2024 · 4be2fb6 · 4be2fb6
1 parent 1241daf
commit 4be2fb6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/nonebot_plugin_orm/__init__.py b/nonebot_plugin_orm/__init__.py
@@ -82,9 +82,12 @@ async def init_orm() -> None:
             cmd_opts.cmd = (migrate.check, [], [])
             try:
                 await greenlet_spawn(migrate.check, alembic_config)
-            except click.UsageError:
-                if not click.confirm("目标数据库未更新到最新迁移, 是否更新?"):
-                    raise
+            except click.UsageError as e:
+                try:
+                    click.confirm("目标数据库未更新到最新迁移, 是否更新?", abort=True)
+                except click.Abort:
+                    raise e
+
                 cmd_opts.cmd = (migrate.upgrade, [], [])
                 await greenlet_spawn(migrate.upgrade, alembic_config)
         else: