kiba-core 0.5.3.dev94__tar.gz → 0.5.3.dev95__tar.gz

{kiba_core-0.5.3.dev94 → kiba_core-0.5.3.dev95}/CHANGELOG.md

@@ -8,6 +8,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 ## [Unreleased]
 
 ### Added
+- [PATCH] Updated `ExceptionHandlingMiddleware` to answer a client that disconnects before the response starts (starlette's `ClientDisconnect`, e.g. while reading the body) with a 499 instead of logging a traceback and a 500
 - [MAJOR] Added `priority` to `SqlMessageQueue.send_message` and `send_messages` (default 0): visible messages are claimed highest priority first, then by `visibleDate`, and a duplicate send, or a retry folded into a pending duplicate, keeps the higher priority. `tbl_queue_messages` has a new `priority` column (`INTEGER NOT NULL DEFAULT 0`), so apps must add it with a migration before upgrading
 - [PATCH] Fixed `DatabaseConnectionMiddleware`'s commit when the response starts failing ("another operation is in progress") for streaming responses behind a `BaseHTTPMiddleware`: `Database.execute` and `Database.commit_context_connection` now wait for each other on the same connection
 - [MINOR] Added `Database.commit_context_connection` to commit the context connection's writes so far; anything run on it afterwards is committed when the context connection ends

{kiba_core-0.5.3.dev94 → kiba_core-0.5.3.dev95}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiba-core
-Version: 0.5.3.dev94
+Version: 0.5.3.dev95
 Summary: Kiba Labs' python utilities
 Project-URL: repository, https://github.com/kibalabs/core-py
 Requires-Python: <4,>=3.12

{kiba_core-0.5.3.dev94 → kiba_core-0.5.3.dev95}/core/api/middleware/exception_handling_middleware.py

@@ -1,5 +1,6 @@
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.middleware.base import RequestResponseEndpoint
+from starlette.requests import ClientDisconnect
 from starlette.requests import Request
 from starlette.responses import Response
 from starlette.types import ASGIApp
@@ -29,6 +30,8 @@ class ExceptionHandlingMiddleware(BaseHTTPMiddleware):
         except RedirectException as exception:
             response = Response(status_code=exception.statusCode)
             response.headers.update(exception.outgoing_headers())
+        except ClientDisconnect:
+            response = Response(status_code=499)
         except ClientException as exception:
             if self.shouldSquashClientExceptions:
                 logging.error(f'{exception.exceptionType} occurred: {exception.message}')

{kiba_core-0.5.3.dev94 → kiba_core-0.5.3.dev95}/kiba_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiba-core
-Version: 0.5.3.dev94
+Version: 0.5.3.dev95
 Summary: Kiba Labs' python utilities
 Project-URL: repository, https://github.com/kibalabs/core-py
 Requires-Python: <4,>=3.12

{kiba_core-0.5.3.dev94 → kiba_core-0.5.3.dev95}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kiba-core"
-version = "0.5.3.dev94"
+version = "0.5.3.dev95"
 description = "Kiba Labs' python utilities"
 readme = "README.md"
 requires-python = ">=3.12, <4"

{kiba_core-0.5.3.dev94 → kiba_core-0.5.3.dev95}/tests/api/test_exception_handling_middleware.py

@@ -119,3 +119,24 @@ def test_plain_exception_has_no_extra_headers(plain_error_client):
     assert response.status_code == 400
     assert 'retry-after' not in response.headers
     assert 'location' not in response.headers
+
+
+@pytest.mark.asyncio
+async def test_client_disconnect_while_reading_body_returns_499():
+    @json_route(requestType=SimpleRequest, responseType=SimpleResponse)
+    async def endpoint(request: KibaApiRequest[SimpleRequest]) -> SimpleResponse:
+        return SimpleResponse(result=request.data.value)
+
+    app = Starlette(routes=[Route('/echo', endpoint, methods=['POST'])])
+    app.add_middleware(ExceptionHandlingMiddleware)
+    sentMessages = []
+
+    async def receive():
+        return {'type': 'http.disconnect'}
+
+    async def send(message):
+        sentMessages.append(message)
+
+    scope = {'type': 'http', 'asgi': {'version': '3.0'}, 'method': 'POST', 'path': '/echo', 'headers': [(b'content-type', b'application/json')], 'query_string': b''}
+    await app(scope, receive, send)
+    assert sentMessages[0]['status'] == 499