#85 use headers.extend instead of adding headers one by one

spec-first · Nov 6, 2015 · 53ac59f · 53ac59f
1 parent b6c9457
commit 53ac59f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/connexion/decorators/produces.py b/connexion/decorators/produces.py
@@ -61,8 +61,7 @@ def process_headers(response, headers):
         :rtype flask.Response
         """
         if headers:
-            for header, value in headers.items():
-                response.headers[header] = value
+            response.headers.extend(headers)
         return response
 
     def __repr__(self):

diff --git a/connexion/problem.py b/connexion/problem.py
@@ -46,9 +46,7 @@ def problem(status, title, detail, type='about:blank', instance=None, headers=No
     body = json.dumps(problem_response)
     response = flask.current_app.response_class(body, mimetype='application/problem+json',
                                                 status=status)  # type: flask.Response
-
     if headers:
-        for key, value in headers.items():
-            response.headers[key] = value
+        response.headers.extend(headers)
 
     return response