darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-09 23:02:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix content-type conditionally

Browse Source
This commit is contained in:
Kenneth Kehl
2025-02-03 12:06:21 -08:00
parent 0de1dd1fd5
commit d61f96d916
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
notifications_utils/request_helper.py
View File

@@ -81,17 +81,21 @@ class ResponseHeaderMiddleware(object):
for key, value in headers
if key.lower() not in ["server", "last-modified"]
]
found_a_text_yaml = False
old_headers_len = len(headers)
headers = [
(key, value)
for key, value in headers
if "werkzeug" not in value.lower()
if "text/yaml" not in value.lower()
]
new_headers_len = len(headers)
if new_headers_len < old_headers_len:
found_a_text_yaml = True
for key, value in headers:
if key.lower() == "content-type" and "text/yaml" in value.lower():
headers.pop("Content-Type")
headers.append("Content-Type", "application/yaml")
headers.append(("Server", "SecureServer"))
if found_a_text_yaml:
headers.append(("Content-Type", "application/yaml"))
print(headers)
return start_response(status, headers, exc_info)
return self._app(environ, rewrite_response_headers)