auditry 0.2.1__tar.gz → 0.2.2__tar.gz

{auditry-0.2.1/src/auditry.egg-info → auditry-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auditry
-Version: 0.2.1
+Version: 0.2.2
 Summary: Observability middleware for FastAPI and Quart with request/response logging, correlation IDs, and sensitive data redaction
 Author-email: Liv Stark <livstark.work@gmail.com>
 License: MIT

{auditry-0.2.1 → auditry-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "auditry"
-version = "0.2.1"
+version = "0.2.2"
 description = "Observability middleware for FastAPI and Quart with request/response logging, correlation IDs, and sensitive data redaction"
 authors = [
     {name = "Liv Stark", email = "livstark.work@gmail.com"}

{auditry-0.2.1 → auditry-0.2.2}/src/auditry/__init__.py

@@ -31,7 +31,7 @@ from .logging_config import configure_logging, get_logger
 from .correlation import get_correlation_id
 from .models import ObservabilityConfig, BusinessEventConfig
 
-__version__ = "0.2.0"
+__version__ = "0.2.2"
 
 __all__ = [
     # Configuration

{auditry-0.2.1 → auditry-0.2.2}/src/auditry/fastapi/adapters.py

@@ -5,7 +5,8 @@ These adapters implement the abstract base classes to extract data from
 FastAPI/Starlette request and response objects.
 """
 
-from typing import Any, Dict, Optional
+import inspect
+from typing import Dict, Optional
 
 from fastapi import Request, Response
 from starlette.responses import StreamingResponse
@@ -104,8 +105,14 @@ class FastAPIResponseAdapter(BaseResponseAdapter):
         if isinstance(response, StreamingResponse):
             return None
 
+        # Check if response has a body_iterator (generator/async generator)
+        if hasattr(response, "body_iterator"):
+            # Check if it's a generator or async generator
+            if inspect.isasyncgen(response.body_iterator) or inspect.isgenerator(response.body_iterator):
+                return None
+
         # Get body if available
         if hasattr(response, "body"):
             return response.body
 
-        return None
+        return None

{auditry-0.2.1 → auditry-0.2.2}/src/auditry/quart/adapters.py

@@ -5,7 +5,7 @@ These adapters implement the abstract base classes to extract data from
 Quart request and response objects.
 """
 
-from typing import Any, Dict, Optional
+import inspect
 import json
 
 from quart import Request, Response, g
@@ -133,17 +133,44 @@ class QuartResponseAdapter(BaseResponseAdapter):
         """
         Extract body from Quart response.
 
+        Returns None for streaming responses to avoid consuming the stream.
         Handles both Response objects and tuple returns.
         """
         try:
             # For Response objects
+            if hasattr(response, "response"):
+                # Check if the response is an async generator (streaming)
+                if inspect.isasyncgen(response.response):
+                    # This is a streaming response, return None to avoid consuming it
+                    return None
+
+                # Check if it's a regular generator (also streaming)
+                if inspect.isgenerator(response.response):
+                    return None
+
+            # For Response objects with get_data method (non-streaming)
             if hasattr(response, "get_data"):
-                return await response.get_data()
+                # Only call get_data if we're sure it's not streaming
+                # Double-check for streaming response attribute
+                if hasattr(response, "response"):
+                    resp_attr = response.response
+                    if inspect.isasyncgen(resp_attr) or inspect.isgenerator(resp_attr):
+                        return None
+
+                # Safe to get data for non-streaming responses
+                data = await response.get_data()
+                if isinstance(data, str):
+                    return data.encode("utf-8")
+                return data
 
             # For tuple responses (body, status_code, headers)
             if isinstance(response, tuple) and len(response) >= 1:
                 body = response[0]
 
+                # Check if body is a generator (streaming)
+                if inspect.isasyncgen(body) or inspect.isgenerator(body):
+                    return None
+
                 # Convert various body types to bytes
                 if isinstance(body, bytes):
                     return body
@@ -154,4 +181,4 @@ class QuartResponseAdapter(BaseResponseAdapter):
 
             return None
         except Exception:
-            return None
+            return None

{auditry-0.2.1 → auditry-0.2.2}/src/auditry/quart/middleware.py

@@ -89,6 +89,41 @@ class QuartMiddleware(BaseMiddleware):
             if not hasattr(request, "observability_start_time"):
                 return response
 
+            # Check if this is a streaming response - if so, handle it separately
+            if hasattr(response, "response"):
+                if inspect.isasyncgen(response.response) or inspect.isgenerator(response.response):
+                    # For streaming responses, log without consuming the stream
+                    duration_ms = (time.time() - request.observability_start_time) * 1000
+
+                    # Get stored correlation ID from before_request
+                    correlation_id = getattr(request, "observability_correlation_id", None)
+                    request_data = getattr(request, "observability_request_data", {})
+
+                    # Re-extract user_id (might have been set during request processing)
+                    user_id = await self.request_adapter.extract_user_id(request)
+
+                    # Log with minimal response data (no body for streaming)
+                    self.logger.log_success(
+                        request_data=request_data,
+                        response_data={
+                            "status_code": response.status_code if hasattr(response, "status_code") else 200,
+                            "headers": dict(response.headers) if hasattr(response, "headers") else {},
+                            "body": None  # Don't try to extract body for streaming
+                        },
+                        duration_ms=duration_ms,
+                        correlation_id=correlation_id,
+                        user_id=user_id,
+                    )
+
+                    # Add correlation ID to response headers (same as non-streaming)
+                    if correlation_id:
+                        response.headers[self.config.correlation_id_header] = correlation_id
+
+                    # Clear cache and return immediately
+                    self.request_adapter.clear_cache(request)
+                    return response
+
+            # Non-streaming response handling
             # Calculate duration
             duration_ms = (time.time() - request.observability_start_time) * 1000
 

{auditry-0.2.1 → auditry-0.2.2/src/auditry.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auditry
-Version: 0.2.1
+Version: 0.2.2
 Summary: Observability middleware for FastAPI and Quart with request/response logging, correlation IDs, and sensitive data redaction
 Author-email: Liv Stark <livstark.work@gmail.com>
 License: MIT