starmallow 0.6.3__py3-none-any.whl → 0.6.5__py3-none-any.whl

starmallow/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.6.3"
+__version__ = "0.6.5"
 
 from .applications import StarMallow
 from .exceptions import RequestValidationError

starmallow/background.py

@@ -0,0 +1,29 @@
+import typing
+from logging import getLogger
+
+from starlette.background import BackgroundTask as StarletteBackgroundTask
+from starlette.background import BackgroundTasks as StarletteBackgroundTasks
+from starlette.background import P
+from starlette.concurrency import run_in_threadpool
+
+logger = getLogger(__name__)
+
+
+class BackgroundTask(StarletteBackgroundTask):
+
+    async def __call__(self) -> None:
+        try:
+            if self.is_async:
+                await self.func(*self.args, **self.kwargs)
+            else:
+                await run_in_threadpool(self.func, *self.args, **self.kwargs)
+        except BaseException as e:
+            logger.exception(f'Background Task {self.func.__module__}.{self.func.__name__} failed: {e}')
+
+
+class BackgroundTasks(StarletteBackgroundTasks):
+    def add_task(
+        self, func: typing.Callable[P, typing.Any], *args: P.args, **kwargs: P.kwargs
+    ) -> None:
+        task = BackgroundTask(func, *args, **kwargs)
+        self.tasks.append(task)

starmallow/request_resolver.py

@@ -8,13 +8,14 @@ import marshmallow as ma
 import marshmallow.fields as mf
 from marshmallow.error_store import ErrorStore
 from marshmallow.utils import missing as missing_
-from starlette.background import BackgroundTasks
+from starlette.background import BackgroundTasks as StarletteBackgroundTasks
 from starlette.datastructures import FormData, Headers, QueryParams
 from starlette.exceptions import HTTPException
 from starlette.requests import HTTPConnection, Request
 from starlette.responses import Response
 from starlette.websockets import WebSocket
 
+from starmallow.background import BackgroundTasks
 from starmallow.params import Param, ParamType, ResolvedParam
 from starmallow.utils import (
     is_async_gen_callable,
@@ -113,7 +114,7 @@ def request_params_to_args(
 async def resolve_basic_args(
     request: Request | WebSocket,
     response: Response,
-    background_tasks: BackgroundTasks,
+    background_tasks: StarletteBackgroundTasks,
     params: Dict[ParamType, Dict[str, Param]],
 ):
     path_values, path_errors = request_params_to_args(
@@ -183,7 +184,7 @@ async def resolve_basic_args(
             values[param_name] = request
         elif lenient_issubclass(param_type, Response):
             values[param_name] = response
-        elif lenient_issubclass(param_type, BackgroundTasks):
+        elif lenient_issubclass(param_type, StarletteBackgroundTasks):
             values[param_name] = background_tasks
 
     return values, errors
@@ -217,7 +218,7 @@ async def call_resolver(
 async def resolve_subparams(
     request: Request | WebSocket,
     response: Response,
-    background_tasks: BackgroundTasks,
+    background_tasks: StarletteBackgroundTasks,
     params: Dict[str, ResolvedParam],
     dependency_cache: Optional[Dict[Tuple[Callable[..., Any], Tuple[str]], Any]],
 ) -> Dict[str, Any]:
@@ -250,10 +251,10 @@ async def resolve_subparams(
 async def resolve_params(
     request: Request | WebSocket,
     params: Dict[ParamType, Dict[str, Param]],
-    background_tasks: Optional[BackgroundTasks] = None,
+    background_tasks: Optional[StarletteBackgroundTasks] = None,
     response: Optional[Response] = None,
     dependency_cache: Optional[Dict[Tuple[Callable[..., Any], Tuple[str]], Any]] = None,
-) -> Tuple[Dict[str, Any], Dict[str, Union[Any, List, Dict]], BackgroundTasks, Response]:
+) -> Tuple[Dict[str, Any], Dict[str, Union[Any, List, Dict]], StarletteBackgroundTasks, Response]:
     dependency_cache = dependency_cache or {}
 
     if response is None:

starmallow/routing.py

@@ -96,7 +96,12 @@ def request_response(
                 response = await run_in_threadpool(func, request)
             await response(scope, receive, send)
 
-        await wrap_app_handling_exceptions(app, request)(scope, receive, send)
+        try:
+            await wrap_app_handling_exceptions(app, request)(scope, receive, send)
+        except RuntimeError as e:
+            # This likely means that the exception was thrown by a background task
+            # after the response has been successfully send to the client
+            logger.exception(f'Runtime error occurred: {e}')
 
     return app
 

{starmallow-0.6.3.dist-info → starmallow-0.6.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: starmallow
-Version: 0.6.3
+Version: 0.6.5
 Summary: StarMallow framework
 Project-URL: Homepage, https://github.com/mvanderlee/starmallow
 Author-email: Michiel Vanderlee <jmt.vanderlee@gmail.com>

{starmallow-0.6.3.dist-info → starmallow-0.6.5.dist-info}/RECORD

@@ -1,5 +1,6 @@
-starmallow/__init__.py,sha256=A9KiNmL4PORVsksM-GGh_4ohWvgxkq5cHhHTdiDscDA,322
+starmallow/__init__.py,sha256=BM647YLqYd8UyAgINXcC9B8aSFtZNYWdO3Gridcnn2s,322
 starmallow/applications.py,sha256=mSL4YDozP8n6v22g4NX7EAMXmGhzzhtjtZd68YHcFvw,31720
+starmallow/background.py,sha256=qxT6-9SfnkcGZzvecNOpIsCOMW0TPwDtpQ81GHI28P0,995
 starmallow/concurrency.py,sha256=MVRjo4Vqss_yqhaoeVt3xb7rLaSuAq_q9uYgTwbsojE,1375
 starmallow/constants.py,sha256=u0h8cJKhJY0oIZqzr7wpEZG2bPLrw5FroMnn3d8KBNQ,129
 starmallow/dataclasses.py,sha256=ap9DInvQjH2AyI4MAAnbDEuNnbPb94PigaNmEb7AQU8,2658
@@ -13,10 +14,10 @@ starmallow/exception_handlers.py,sha256=gr2qLYWEtsIEH28n7OreEiiLVz6Y7b6osRyS9esJ
 starmallow/exceptions.py,sha256=vabtPJkTmtCdC8_2OPBE8Osz0v0KxaSOX6IWf1jgNkc,872
 starmallow/fields.py,sha256=arrTabCYoJFZcoY69EZTBH3YUg7CUSr3-zYLiAjYLTM,1238
 starmallow/params.py,sha256=MJzUzUs6GEyrbpDZ1r8To8vR-QwpopdxDStq802o5Ug,8662
-starmallow/request_resolver.py,sha256=JoCxqdDYyniwNsI6FeeYKB7Muj9xDQ8LEA2ICba8n24,11012
+starmallow/request_resolver.py,sha256=frOxhQPxhWKnStJDTVrs5wu6Icf3dTbbv9VblegxS3A,11135
 starmallow/requests.py,sha256=o_yNhH9WJ35uAGuoXa5_gihevHeaveOvkP525dbwQSU,843
 starmallow/responses.py,sha256=k2pf_m21ykf_FECdODUz400pMucMJJf_Zm8TXFujvaU,2012
-starmallow/routing.py,sha256=Uxnlkl4eIs56TZ8VAgNmlh-K89ku1Xu_ROR5S8x-rkg,45128
+starmallow/routing.py,sha256=VSotmrEerVzuUfn20mpmSbuRVS4XiHrPtNRvBP8KJ4M,45397
 starmallow/schema_generator.py,sha256=yi368FwF9B50ZHSNOG0rvYVirVUeMFq2kXkUDeJUz4w,17961
 starmallow/serializers.py,sha256=rBEKMNgONgz_bai12uDvAEMCI_aEFGsqMSeIoWtlrOI,12514
 starmallow/types.py,sha256=8GXWjvzXQhF5NMHf14fbid6uErxVd1Xk_w2I4FoUgZ4,717
@@ -34,7 +35,7 @@ starmallow/security/http.py,sha256=cpGjM1kFDq3i_bOY96kMkf4cspBUxFkkET9lTK3NA-0,6
 starmallow/security/oauth2.py,sha256=1nv1580PY4cwgu5gzpQCf2MfMNv2Cfv05753AUHPOhQ,10005
 starmallow/security/open_id_connect_url.py,sha256=IPsL2YzWc2mPwJbrUn6oFRTi7uRAG6mR62CGwmzBs1k,1399
 starmallow/security/utils.py,sha256=bd8T0YM7UQD5ATKucr1bNtAvz_Y3__dVNAv5UebiPvc,293
-starmallow-0.6.3.dist-info/METADATA,sha256=ccZtRpeODLBXd3pJPyZ-Bsk_w0mxZvmoOtkRsLDvBm8,5615
-starmallow-0.6.3.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-starmallow-0.6.3.dist-info/licenses/LICENSE.md,sha256=QelyGgOzch8CXzy6HrYwHh7nmj0rlWkDA0YzmZ3CPaY,1084
-starmallow-0.6.3.dist-info/RECORD,,
+starmallow-0.6.5.dist-info/METADATA,sha256=cAAboTKZYPwxOZEqzjIxZx6lfcMlRLPbee8vRRAQkfo,5615
+starmallow-0.6.5.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+starmallow-0.6.5.dist-info/licenses/LICENSE.md,sha256=QelyGgOzch8CXzy6HrYwHh7nmj0rlWkDA0YzmZ3CPaY,1084
+starmallow-0.6.5.dist-info/RECORD,,