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: try:
response = await handler(request) response = await handler(request)
except HTTPException as err: except HTTPException as err:
for key, value in added_headers.items(): err.headers.update(added_headers)
err.headers[key] = value
raise raise
for key, value in added_headers.items(): response.headers.update(added_headers)
response.headers[key] = value
return response return response
app.middlewares.append(headers_middleware) app.middlewares.append(headers_middleware)