Skip to content

Commit d10abab

Browse files
committed
add management of Flask response
1 parent b4cb557 commit d10abab

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

flask_rest_jsonapi/resource.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99
from werkzeug.wrappers import Response
1010
from flask import request, url_for, make_response
11+
from flask.wrappers import Response as FlaskResponse
1112
from flask.views import MethodView, MethodViewType
1213
from marshmallow_jsonapi.exceptions import IncorrectTypeError
1314
from marshmallow import ValidationError
@@ -93,7 +94,11 @@ def dispatch_request(self, *args, **kwargs):
9394
if isinstance(data, dict):
9495
data.update({'jsonapi': {'version': '1.0'}})
9596

96-
if isinstance(data, str):
97+
if isinstance(data, FlaskResponse):
98+
data.headers.add('Content-Type', 'application/vnd.api+json')
99+
data.status_code = status_code
100+
return data
101+
elif isinstance(data, str):
97102
json_reponse = data
98103
else:
99104
json_reponse = json.dumps(data, cls=JSONEncoder)

0 commit comments

Comments
 (0)