♻️ refactor(alembic): support unload or uninstall used plugins

Author: ProgramRipper

nonebot_plugin_orm/__init__.py

@@ -31,6 +31,9 @@
 else:
     from typing_extensions import Annotated
 
+require("nonebot_plugin_localstore")
+from nonebot_plugin_localstore import get_data_dir
+
 __all__ = (
     # __init__
     "init_orm",
@@ -63,6 +66,7 @@
 _plugins: dict[str, Plugin]
 _session_factory: sa_async.async_sessionmaker[AsyncSession]
 
+_data_dir = get_data_dir(__plugin_meta__.name)
 _driver = get_driver()
 
 
@@ -162,17 +166,13 @@ def _init_engines():
         import aiosqlite
 
         del aiosqlite
-        require("nonebot_plugin_localstore")
-        from nonebot_plugin_localstore import get_data_file
     except (ImportError, RuntimeError):
         raise ValueError(
             '必须指定一个默认数据库 (SQLALCHEMY_DATABASE_URL 或 SQLALCHEMY_BINDS[""]). '
             "可以通过 `pip install nonebot-plugin-orm[default]` 获得开箱即用的数据库配置."
         ) from None
 
-    _engines[""] = _create_engine(
-        f"sqlite+aiosqlite:///{get_data_file(__plugin_meta__.name, 'db.sqlite3')}"
-    )
+    _engines[""] = _create_engine(f"sqlite+aiosqlite:///{_data_dir / 'db.sqlite3'}")
 
 
 def _init_table():

nonebot_plugin_orm/env.py

@@ -0,0 +1,17 @@
+from __future__ import annotations
+
+from alembic import context
+from sqlalchemy.sql.schema import SchemaItem
+
+from nonebot_plugin_orm import migrate
+
+
+def no_drop_table(
+    _, __, type_: str, reflected: bool, compare_to: SchemaItem | None
+) -> bool:
+    return not (
+        getattr(context.config.cmd_opts, "cmd", (None,))[0] == migrate.check
+        and type_ == "table"
+        and reflected
+        and compare_to is None
+    )