entari-plugin-database 0.2.4__tar.gz → 0.3.0__tar.gz

{entari_plugin_database-0.2.4 → entari_plugin_database-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: entari-plugin-database
-Version: 0.2.4
+Version: 0.3.0
 Summary: Entari plugin for SQLAlchemy ORM
 Author-Email: RF-Tar-Railt <rf_tar_railt@qq.com>
 License: MIT
@@ -11,7 +11,7 @@ Requires-Dist: graia-amnesia>=0.11.4
 Requires-Dist: tarina<0.8.0,>=0.7.1
 Requires-Dist: arclet-letoderea>=0.19.5
 Requires-Dist: alembic>=1.16.5
-Requires-Dist: arclet-entari<0.18.0,>=0.17.0
+Requires-Dist: arclet-entari<0.19.0,>=0.18.0rc1
 Description-Content-Type: text/markdown
 
 # entari-plugin-database

{entari_plugin_database-0.2.4 → entari_plugin_database-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "entari-plugin-database"
-version = "0.2.4"
+version = "0.3.0"
 description = "Entari plugin for SQLAlchemy ORM"
 authors = [
     { name = "RF-Tar-Railt", email = "rf_tar_railt@qq.com" },
@@ -12,7 +12,7 @@ dependencies = [
     "tarina<0.8.0,>=0.7.1",
     "arclet-letoderea>=0.19.5",
     "alembic>=1.16.5",
-    "arclet-entari<0.18.0,>=0.17.0",
+    "arclet-entari<0.19.0,>=0.18.0rc1",
 ]
 requires-python = ">=3.10"
 readme = "README.md"
@@ -29,11 +29,6 @@ build-backend = "pdm.backend"
 [tool.pdm]
 distribution = true
 
-[tool.pdm.dev-dependencies]
-dev = [
-    "arclet-entari[full]>=0.17.0",
-]
-
 [tool.ruff]
 line-length = 120
 target-version = "py310"
@@ -61,3 +56,13 @@ ignore = [
     "PYI055",
     "UP038",
 ]
+
+[tool.pyright]
+pythonVersion = "3.10"
+pythonPlatform = "All"
+typeCheckingMode = "basic"
+
+[dependency-groups]
+dev = [
+    "arclet-entari[full]>=0.17.0",
+]

{entari_plugin_database-0.2.4 → entari_plugin_database-0.3.0}/src/entari_plugin_database/__init__.py

@@ -1,8 +1,6 @@
 from sqlalchemy.ext.asyncio import create_async_engine
-from arclet.letoderea.provider import global_providers
-from arclet.letoderea.scope import global_propagators
-from arclet.letoderea.core import add_task
-from arclet.entari import plugin
+from arclet.letoderea.utils import add_task
+from arclet.entari import Plugin, plugin
 from arclet.entari.config import config_model_validate
 from arclet.entari.event.config import ConfigReload
 from graia.amnesia.builtins.sqla import SqlalchemyService
@@ -14,7 +12,6 @@ from sqlalchemy.ext import asyncio as sa_async
 from sqlalchemy.orm import Mapped as Mapped, instrumentation
 from sqlalchemy.orm import mapped_column as mapped_column
 
-from .param import db_supplier, sess_provider, orm_factory
 from .param import SQLDepends as SQLDepends
 from .utils import logger
 from .migration import run_migration, register_custom_migration
@@ -25,7 +22,7 @@ plugin.declare_static()
 plugin.metadata(
     "Database 服务",
     [{"name": "RF-Tar-Railt", "email": "rf_tar_railt@qq.com"}],
-    "0.2.4",
+    "0.3.0",
     description="基于 SQLAlchemy 的数据库服务插件",
     urls={
         "homepage": "https://github.com/ArcletProject/entari-plugin-database",
@@ -33,11 +30,6 @@ plugin.metadata(
     config=Config,
     readme="README.md",
 )
-plugin.collect_disposes(
-    lambda: global_propagators.remove(db_supplier),
-    lambda: global_providers.remove(sess_provider),
-    lambda: global_providers.remove(orm_factory),
-)
 
 _config = plugin.get_config(Config)
 
@@ -133,12 +125,17 @@ def migration_callback(cls: type[Base], kwargs: dict):
     task.add_done_callback(_PENDING_TASKS.discard)
 
 
-register_callback(_setup_tablename)
-register_callback(_clean_exist)
-register_callback(migration_callback, after=True)
-plugin.collect_disposes(lambda: remove_callback(_clean_exist))
-plugin.collect_disposes(lambda: remove_callback(_setup_tablename))
-plugin.collect_disposes(lambda: remove_callback(migration_callback))
+def _register_callbacks():
+    register_callback(_setup_tablename)
+    register_callback(_clean_exist)
+    register_callback(migration_callback, after=True)
+    yield lambda: remove_callback(_clean_exist)
+    yield lambda: remove_callback(_setup_tablename)
+    yield lambda: remove_callback(migration_callback)
+
+
+plg = Plugin.current()
+plg.effect(_register_callbacks, "database model callbacks")
 
 
 BaseOrm = Base

{entari_plugin_database-0.2.4 → entari_plugin_database-0.3.0}/src/entari_plugin_database/param.py

@@ -20,22 +20,44 @@ from arclet.letoderea import Propagator, Contexts, STACK, Provider, ProviderFact
 from arclet.letoderea.ref import Deref, generate
 from arclet.letoderea.provider import global_providers
 from arclet.letoderea.scope import global_propagators
+from arclet.entari.plugin.model import Plugin
 from sqlalchemy.ext import asyncio as sa_async
 
 
+class SessionProvider(Provider[sa_async.AsyncSession]):
+    priority = 10
+
+    async def __call__(self, context: Contexts):
+        if "$db_session" in context:
+            return context["$db_session"]
+        try:
+            db = it(Launart).get_component(SqlalchemyService)
+            stack = context[STACK]
+            sess = await stack.enter_async_context(db.get_session())
+            context["$db_session"] = sess
+            return sess
+        except ValueError:
+            return
+
+
 class DatabasePropagator(Propagator):
     def validate(self, subscriber: Subscriber):
         params = subscriber.params
         if any((p.depend and isinstance(p.depend, SQLDepend)) for p in params):
             return True
+        if any(isinstance(p.annotation, sa_async.AsyncSession) for p in params):
+            return True
         if any(
-            isinstance(prod, (SessionProvider, ORMProviderFactory._ModelProvider))
+            isinstance(prod, ORMProviderFactory._ModelProvider)
             for p in params
             for prod in p.providers
         ):  # noqa: E501,UP038
             return True
         return False
 
+    def providers(self):
+        return [SessionProvider()]
+
     async def supply(self, ctx: Contexts, serv: SqlalchemyService | None = None):
         if serv is None:
             return
@@ -48,22 +70,6 @@ class DatabasePropagator(Propagator):
         yield self.supply, True, 20
 
 
-class SessionProvider(Provider[sa_async.AsyncSession]):
-    priority = 10
-
-    async def __call__(self, context: Contexts):
-        if "$db_session" in context:
-            return context["$db_session"]
-        try:
-            db = it(Launart).get_component(SqlalchemyService)
-            stack = context[STACK]
-            sess = await stack.enter_async_context(db.get_session())
-            context["$db_session"] = sess
-            return sess
-        except ValueError:
-            return
-
-
 @dataclass(unsafe_hash=True)
 class Option:
     stream: bool = True
@@ -254,6 +260,14 @@ class ORMProviderFactory(ProviderFactory):
         return self._ModelProvider(statement, option)
 
 
-global_propagators.append(db_supplier := DatabasePropagator())
-global_providers.append(sess_provider := SessionProvider())
-global_providers.append(orm_factory := ORMProviderFactory())
+plg = Plugin.current()
+
+
+def _provides():
+    global_propagators.append(db_supplier := DatabasePropagator())
+    global_providers.append(orm_factory := ORMProviderFactory())
+    yield lambda: global_propagators.remove(db_supplier)
+    yield lambda: global_providers.remove(orm_factory)
+
+
+plg.effect(_provides, "database providers")