Customize automatic API return types

jacebrowning · jacebrowning · commit 30ccdcd52827 · 2017-04-11T22:51:54.000-04:00
diff --git a/docs/about/release-notes.md b/docs/about/release-notes.md
@@ -2,6 +2,10 @@
 
 This project is currently in beta.  It is functional and well tested but you are advised to pay close attention to the release notes when upgrading to future versions.
 
+## Version 0.7.1 (unreleased)
+
+* Added customization of automatic API return types (`app.response_class.api_return_types`).
+
 ## Version 0.7
 
 * Disabled rendering of text responses as API views.
diff --git a/flask_api/__init__.py b/flask_api/__init__.py
@@ -1,3 +1,3 @@
 from flask_api.app import FlaskAPI
 
-__version__ = '0.7'
+__version__ = '0.7.1b2'
diff --git a/flask_api/app.py b/flask_api/app.py
@@ -56,7 +56,8 @@ def make_response(self, rv):
 
         if not isinstance(rv, self.response_class):
             if isinstance(rv, (text_type, bytes, bytearray, list, dict)):
-                rv = self.response_class(rv, headers=headers, status=status_or_headers)
+                status = status_or_headers
+                rv = self.response_class(rv, headers=headers, status=status)
                 headers = status_or_headers = None
             else:
                 rv = self.response_class.force_type(rv, request.environ)
diff --git a/flask_api/response.py b/flask_api/response.py
@@ -5,11 +5,14 @@
 
 
 class APIResponse(Response):
+
+    api_return_types = (list, dict)
+
     def __init__(self, content=None, *args, **kwargs):
         super(APIResponse, self).__init__(None, *args, **kwargs)
 
         media_type = None
-        if isinstance(content, (list, dict)) or content == '':
+        if isinstance(content, self.api_return_types) or content == '':
             renderer = request.accepted_renderer
             if content != '' or renderer.handles_empty_responses:
                 media_type = request.accepted_media_type

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,3 @@`
`1`	`1`	`from flask_api.app import FlaskAPI`
`2`	`2`
`3`		`-__version__ = '0.7'`
	`3`	`+__version__ = '0.7.1b2'`