modern-di-faststream 1.1.0__tar.gz → 2.0.0a0__tar.gz

{modern_di_faststream-1.1.0 → modern_di_faststream-2.0.0a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modern-di-faststream
-Version: 1.1.0
+Version: 2.0.0a0
 Summary: Modern-DI integration for FastStream
 Project-URL: repository, https://github.com/modern-python/modern-di
 Project-URL: docs, https://modern-di.readthedocs.io

{modern_di_faststream-1.1.0 → modern_di_faststream-2.0.0a0}/modern_di_faststream/main.py

@@ -7,7 +7,7 @@ import faststream
 import modern_di
 from faststream.asgi import AsgiFastStream
 from faststream.types import DecodedMessage
-from modern_di import AsyncContainer, providers
+from modern_di import Container, providers
 
 
 T_co = typing.TypeVar("T_co", covariant=True)
@@ -20,7 +20,7 @@ _OLD_MIDDLEWARES = int(_major) == 0 and int(_minor) < 6  # noqa: PLR2004
 class _DIMiddlewareFactory:
     __slots__ = ("di_container",)
 
-    def __init__(self, di_container: AsyncContainer) -> None:
+    def __init__(self, di_container: Container) -> None:
         self.di_container = di_container
 
     def __call__(self, *args: P.args, **kwargs: P.kwargs) -> "_DiMiddleware[P]":
@@ -28,7 +28,7 @@ class _DIMiddlewareFactory:
 
 
 class _DiMiddleware(faststream.BaseMiddleware, typing.Generic[P]):
-    def __init__(self, di_container: AsyncContainer, *args: P.args, **kwargs: P.kwargs) -> None:
+    def __init__(self, di_container: Container, *args: P.args, **kwargs: P.kwargs) -> None:
         self.di_container = di_container
         super().__init__(*args, **kwargs)  # type: ignore[arg-type]
 
@@ -37,14 +37,14 @@ class _DiMiddleware(faststream.BaseMiddleware, typing.Generic[P]):
         call_next: Callable[[typing.Any], Awaitable[typing.Any]],
         msg: faststream.StreamMessage[typing.Any],
     ) -> typing.AsyncIterator[DecodedMessage]:
-        async with self.di_container.build_child_container(
+        request_container = self.di_container.build_child_container(
             scope=modern_di.Scope.REQUEST, context={faststream.StreamMessage: msg}
-        ) as request_container:
-            with self.faststream_context.scope("request_container", request_container):
-                return typing.cast(
-                    typing.AsyncIterator[DecodedMessage],
-                    await call_next(msg),
-                )
+        )
+        with self.faststream_context.scope("request_container", request_container):
+            return typing.cast(
+                typing.AsyncIterator[DecodedMessage],
+                await call_next(msg),
+            )
 
     if _OLD_MIDDLEWARES:  # pragma: no cover
 
@@ -59,21 +59,19 @@ class _DiMiddleware(faststream.BaseMiddleware, typing.Generic[P]):
             return self.context
 
 
-def fetch_di_container(app_: faststream.FastStream | AsgiFastStream) -> AsyncContainer:
-    return typing.cast(AsyncContainer, app_.context.get("di_container"))
+def fetch_di_container(app_: faststream.FastStream | AsgiFastStream) -> Container:
+    return typing.cast(Container, app_.context.get("di_container"))
 
 
 def setup_di(
     app: faststream.FastStream | AsgiFastStream,
-    container: AsyncContainer,
-) -> AsyncContainer:
+    container: Container,
+) -> Container:
     if not app.broker:
         msg = "Broker must be defined to setup DI"
         raise RuntimeError(msg)
 
     app.context.set_global("di_container", container)
-    app.on_startup(container.enter)
-    app.after_shutdown(container.close)
     app.broker.add_middleware(_DIMiddlewareFactory(container))
     return container
 
@@ -83,10 +81,10 @@ class Dependency(typing.Generic[T_co]):
     dependency: providers.AbstractProvider[T_co] | type[T_co]
 
     async def __call__(self, context: faststream.ContextRepo) -> T_co:
-        request_container: modern_di.AsyncContainer = context.get("request_container")
+        request_container: Container = context.get("request_container")
         if isinstance(self.dependency, providers.AbstractProvider):
-            return await request_container.resolve_provider(self.dependency)
-        return await request_container.resolve(dependency_type=self.dependency)
+            return request_container.resolve_provider(self.dependency)
+        return request_container.resolve(dependency_type=self.dependency)
 
 
 def FromDI(  # noqa: N802

{modern_di_faststream-1.1.0 → modern_di_faststream-2.0.0a0}/pyproject.toml

@@ -16,7 +16,7 @@ classifiers = [
     "Topic :: Software Development :: Libraries",
 ]
 dependencies = ["faststream>=0.5,<1", "modern-di>=1.0.0alpha"]
-version = "1.1.0"
+version = "2.0.0a0"
 
 [project.urls]
 repository = "https://github.com/modern-python/modern-di"