Switch to using update for headers middleware (#122952)

This commit is contained in:
J. Nick Koston
2024-07-31 11:17:45 -05:00
committed by GitHub
parent 69f54656c4
commit 172e2125f6
+2 -5
View File
@@ -31,13 +31,10 @@ def setup_headers(app: Application, use_x_frame_options: bool) -> None:
try:
response = await handler(request)
except HTTPException as err:
for key, value in added_headers.items():
err.headers[key] = value
err.headers.update(added_headers)
raise
for key, value in added_headers.items():
response.headers[key] = value
response.headers.update(added_headers)
return response
app.middlewares.append(headers_middleware)