muffin 0.93.4__py3-none-any.whl → 0.94.2__py3-none-any.whl

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):
@@ -125,8 +131,8 @@ class Application(BaseApp):
         package_name = parent_frame.f_locals["__name__"]
         return import_submodules(package_name, *submodules)
 
-    def run_background(self, task: Awaitable):
-        """Await the given awaitable after the request is completed.
+    def run_after_response(self, task: Awaitable):
+        """Await the given awaitable after the response is completed.
 
         .. code-block:: python
 
@@ -143,7 +149,7 @@ class Application(BaseApp):
             async def send(request):
 
             # Schedule any awaitable for later execution
-            app.run_background(send_email('user@email.com', 'Hello from Muffin!'))
+            app.run_after_response(send_email('user@email.com', 'Hello from Muffin!'))
 
             # Return response to a client immediately
             # The task will be executed after the response is sent
@@ -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.4.dist-info → muffin-0.94.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: muffin
-Version: 0.93.4
+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.4.dist-info → muffin-0.94.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 muffin/__init__.py,sha256=F8Qp4ETX5FqI1832dj2moRAQ5hEb0TO9u4aRjSKXNjI,982
-muffin/app.py,sha256=-Gw9G_xL4WBhiB05aIvVPZSRNlXmORN0xX9o8C9CJPE,4774
+muffin/app.py,sha256=tfmm9Dc-ZbUM9UwFUAYDIGPMcNJJU3hz5Wt8uwIylYM,5033
 muffin/constants.py,sha256=Q9Nm3Q4P3llL_irsvNQAkkD3CqlD4zKkGAM4nabzwrg,38
 muffin/errors.py,sha256=I-vKbMMBiMU07zPdKvoJKqA7s4xYAUA-4oZXrRMRzcM,701
 muffin/handler.py,sha256=c0b4pwsIF8jmqYwsWn3iuXR5BMwq2e6ASatGcPu9-N4,3667
@@ -9,9 +9,9 @@ muffin/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 muffin/pytest.py,sha256=vnSEsQP8dJCFeS8nZklzAh7Stc3H-ng3jm6KlnAOMN0,2169
 muffin/types.py,sha256=wsUj5oAfqSZMoEf-wyFJLBlWa8Mc-eJGqKLr02HxuXE,153
 muffin/utils.py,sha256=4tdGNAmdEy6LPaf-5rLYx2YMZ5EOe6tXSqAC1GtG_TY,2422
-muffin-0.93.4.dist-info/LICENSE,sha256=xHPkOZhjyKBMOwXpWn9IB_BVLjrrMxv2M9slKkHj2hM,1082
-muffin-0.93.4.dist-info/METADATA,sha256=O3gdr3jLwZyfZT0w9Tp9fhnVP6ubXCvHUvmwipYmldg,12021
-muffin-0.93.4.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-muffin-0.93.4.dist-info/entry_points.txt,sha256=Q3DAZGQk-y5n7HmzJ4FrkNizqTRE2OVXsniPQXzS0Bg,87
-muffin-0.93.4.dist-info/top_level.txt,sha256=b_xNEFyizTmSqZWCTHwcvbycGyADewdsnaLXo2Pq1Ck,7
-muffin-0.93.4.dist-info/RECORD,,
+muffin-0.94.2.dist-info/LICENSE,sha256=xHPkOZhjyKBMOwXpWn9IB_BVLjrrMxv2M9slKkHj2hM,1082
+muffin-0.94.2.dist-info/METADATA,sha256=205DzwT2ZUtnfwD7ki1HeFSkBQ5M3leSOX5cUU_RuSA,12021
+muffin-0.94.2.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+muffin-0.94.2.dist-info/entry_points.txt,sha256=Q3DAZGQk-y5n7HmzJ4FrkNizqTRE2OVXsniPQXzS0Bg,87
+muffin-0.94.2.dist-info/top_level.txt,sha256=b_xNEFyizTmSqZWCTHwcvbycGyADewdsnaLXo2Pq1Ck,7
+muffin-0.94.2.dist-info/RECORD,,