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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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)