Merge pull request #45 from EOEPCA/release/v0.2

Release/v0.2
EOEPCA-archive · Nov 16, 2020 · accca00 · accca00
2 parents b9982c8 + f88a7e4
commit accca00
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main.py b/src/main.py
@@ -178,8 +178,9 @@ def proxy_request(request, new_header):
         excluded_headers = ['transfer-encoding']
         headers = [(name, value) for (name, value) in     res.raw.headers.items() if name.lower() not in excluded_headers]
         response = Response(res.content, res.status_code, headers)
-        response.autocorrect_location_header = False
-        response.headers["Location"] = g_config["proxy_endpoint"] + response.headers["Location"].replace(g_config["resource_server_endpoint"], '')
+        if "Location" in response.headers:
+            response.autocorrect_location_header = False
+            response.headers["Location"] = g_config["proxy_endpoint"] + response.headers["Location"].replace(g_config["resource_server_endpoint"], '')
         return response
     except Exception as e:
         response = Response()