Skip to content

Commit 68bd400

Browse files
committed
Fixed response body stream
1 parent 10d72cc commit 68bd400

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

debug_toolbar/middleware.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from anyio import CapacityLimiter
88
from anyio.lowlevel import RunVar
99
from fastapi import APIRouter, HTTPException, Request, Response, status
10+
from fastapi.responses import StreamingResponse
1011
from fastapi.staticfiles import StaticFiles
1112
from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
1213
from starlette.routing import NoMatchFound
@@ -84,9 +85,12 @@ async def dispatch(
8485
toolbar.generate_server_timing_header(response)
8586

8687
if is_html:
87-
async for body in response.body_iterator: # type: ignore[attr-defined]
88-
if not isinstance(body, bytes):
89-
body = body.encode(response.charset)
88+
body = b""
89+
90+
async for chunk in response.body_iterator: # type: ignore[attr-defined]
91+
if not isinstance(chunk, bytes):
92+
chunk = chunk.encode(response.charset)
93+
body += chunk
9094

9195
decoded = body.decode(response.charset)
9296
pattern = re.escape(self.settings.INSERT_BEFORE)

0 commit comments

Comments
 (0)