muffin 0.93.5__tar.gz → 0.94.2__tar.gz

{muffin-0.93.5/muffin.egg-info → muffin-0.94.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: muffin
-Version: 0.93.5
+Version: 0.94.2
 Summary: Muffin is a fast, simple and asyncronous web-framework for Python 3 (asyncio, trio, curio)
 Author-email: Kirill Klenov <horneds@gmail.com>
 License: MIT License

{muffin-0.93.5 → muffin-0.94.2}/muffin/app.py

@@ -2,12 +2,13 @@
 from __future__ import annotations
 
 import logging
+from contextvars import ContextVar
 from inspect import isawaitable, stack
 from logging.config import dictConfig
-from typing import TYPE_CHECKING, Any, Dict, Mapping, Union
+from typing import TYPE_CHECKING, Any, Dict, Final, Mapping, Union
 
 from asgi_tools import App as BaseApp
-from asgi_tools.middleware import BACKGROUND_TASK
+from asgi_tools._compat import aio_wait
 from modconfig import Config
 
 from muffin.constants import CONFIG_ENV_VARIABLE
@@ -21,6 +22,11 @@ if TYPE_CHECKING:
 
     from muffin.plugins import BasePlugin
 
+BACKGROUND_TASK: Final["ContextVar[set[Awaitable] | None]"] = ContextVar(
+    "background_tasks",
+    default=None,
+)
+
 
 class Application(BaseApp):
     """The Muffin Application."""
@@ -103,9 +109,9 @@ class Application(BaseApp):
     ):
         """Support background tasks."""
         await self.lifespan(scope, receive, send)
-        bgtask = BACKGROUND_TASK.get()
-        if bgtask is not None:
-            await bgtask
+        bgtasks = BACKGROUND_TASK.get()
+        if bgtasks is not None:
+            await aio_wait(*bgtasks)
             BACKGROUND_TASK.set(None)
 
     def import_submodules(self, *submodules: str):
@@ -153,4 +159,7 @@ class Application(BaseApp):
         if not isawaitable(task):
             raise TypeError("Task must be awaitable")  # noqa: TRY003
 
-        BACKGROUND_TASK.set(task)
+        scheduled = BACKGROUND_TASK.get() or set()
+        scheduled.add(task)
+
+        BACKGROUND_TASK.set(scheduled)

{muffin-0.93.5 → muffin-0.94.2/muffin.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: muffin
-Version: 0.93.5
+Version: 0.94.2
 Summary: Muffin is a fast, simple and asyncronous web-framework for Python 3 (asyncio, trio, curio)
 Author-email: Kirill Klenov <horneds@gmail.com>
 License: MIT License

{muffin-0.93.5 → muffin-0.94.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "muffin"
-version = "0.93.5"
+version = "0.94.2"
 description = "Muffin is a fast, simple and asyncronous web-framework for Python 3 (asyncio, trio, curio)"
 readme = "README.rst"
 requires-python = ">=3.8"