WSGI

Signed-off-by: Anuraag Agrawal <anuraaga@gmail.com>

Author: anuraaga

pyproject.toml

@@ -28,7 +28,7 @@ classifiers = [
   "Topic :: Software Development :: Libraries :: Python Modules",
   "Typing :: Typed",
 ]
-dependencies = ["protobuf>=5.28", "pyqwest>=0.4.1"]
+dependencies = ["protobuf>=5.28", "pyqwest>=0.5.1"]
 
 [project.urls]
 Documentation = "https://connectrpc.com/docs/python/getting-started/"

src/connectrpc/_server_sync.py

@@ -2,6 +2,7 @@
 
 import base64
 import functools
+import traceback
 from abc import ABC, abstractmethod
 from dataclasses import replace
 from http import HTTPStatus
@@ -46,9 +47,9 @@
     from .compression import Compression
 
     if sys.version_info >= (3, 11):
-        from wsgiref.types import StartResponse, WSGIEnvironment
+        from wsgiref.types import ErrorStream, StartResponse, WSGIEnvironment
     else:
-        from _typeshed.wsgi import StartResponse, WSGIEnvironment
+        from _typeshed.wsgi import ErrorStream, StartResponse, WSGIEnvironment
 else:
     StartResponse = "wsgiref.types.StartResponse"
     WSGIEnvironment = "wsgiref.types.WSGIEnvironment"
@@ -251,6 +252,7 @@ def __call__(
 
         except Exception as e:
             _drain_request_body(environ)
+            _maybe_log_exception(environ, e)
             return self._handle_error(e, ctx, start_response)
 
     def _handle_unary(
@@ -502,6 +504,7 @@ def _handle_stream(
             # response message will be handled by _response_stream, so here we have a
             # full error-only response.
             _drain_request_body(environ)
+            _maybe_log_exception(environ, e)
             _send_stream_response_headers(
                 start_response, protocol, codec, resp_compression.name(), ctx
             )
@@ -668,3 +671,12 @@ def _drain_request_body(environ: WSGIEnvironment) -> None:
         # server that doesn't do so, so we go ahead and do it ourselves.
         for _ in _read_body(environ):
             pass
+
+
+def _maybe_log_exception(environ: WSGIEnvironment, exc: Exception) -> None:
+    if isinstance(exc, (ConnectError, HTTPException)):
+        return
+    errors: ErrorStream = environ["wsgi.errors"]
+    errors.write(
+        f"Exception in WSGI application\n{''.join(traceback.format_exception(type(exc), exc, exc.__traceback__))}"
+    )

test/test_errors.py

@@ -491,9 +491,7 @@ async def make_hat(self, request, ctx) -> NoReturn:
         with pytest.raises(ConnectError, match="We're broken"):
             await client.make_hat(request=Size(inches=10))
 
-    # Workaround https://github.com/curioswitch/pyqwest/pull/148
-    # TODO: Remove after fix is released
-    assert getattr(transport, "_app_exception", None) is None
+    assert transport.app_exception is None
 
 
 @pytest.mark.asyncio
@@ -515,9 +513,7 @@ def make_similar_hats(
             async for _ in client.make_similar_hats(request=Size(inches=10)):
                 pass
 
-    # Workaround https://github.com/curioswitch/pyqwest/pull/148
-    # TODO: Remove after fix is released
-    assert getattr(transport, "_app_exception", None) is None
+    assert transport.app_exception is None
 
 
 @pytest.mark.asyncio
@@ -536,6 +532,50 @@ async def make_hat(self, request, ctx) -> NoReturn:
         with pytest.raises(ConnectError, match="Internal Server Error"):
             await client.make_hat(request=Size(inches=10))
 
-    # Workaround https://github.com/curioswitch/pyqwest/pull/148
-    # TODO: Remove after fix is released
-    assert getattr(transport, "_app_exception", None) is None
+    assert transport.app_exception is None
+
+
+def test_sync_unhandled_exception_logged() -> None:
+    class RaisingHaberdasher(HaberdasherSync):
+        def make_hat(self, request, ctx) -> NoReturn:
+            raise TypeError("Something went wrong")
+
+    app = HaberdasherWSGIApplication(RaisingHaberdasher())
+    transport = WSGITransport(app)
+    http_client = SyncClient(transport)
+
+    with (
+        HaberdasherClientSync(
+            "http://localhost", timeout_ms=200, http_client=http_client
+        ) as client,
+        pytest.raises(ConnectError, match="Something went wrong"),
+    ):
+        client.make_hat(request=Size(inches=10))
+
+    logged_error = transport.error_stream.getvalue()
+    assert "Exception in WSGI application" in logged_error
+    assert "TypeError: Something went wrong" in logged_error
+    assert "Traceback" in logged_error
+
+
+def test_sync_unhandled_exception_logged_stream() -> None:
+    class RaisingHaberdasher(HaberdasherSync):
+        def make_similar_hats(self, request, ctx) -> NoReturn:
+            raise TypeError("Something went wrong")
+
+    app = HaberdasherWSGIApplication(RaisingHaberdasher())
+    transport = WSGITransport(app)
+    http_client = SyncClient(transport)
+
+    with (
+        HaberdasherClientSync(
+            "http://localhost", timeout_ms=200, http_client=http_client
+        ) as client,
+        pytest.raises(ConnectError, match="Something went wrong"),
+    ):
+        next(client.make_similar_hats(request=Size(inches=10)))
+
+    logged_error = transport.error_stream.getvalue()
+    assert "Exception in WSGI application" in logged_error
+    assert "TypeError: Something went wrong" in logged_error
+    assert "Traceback" in logged_error