add json_request decorator with test-cases

Author: nitinbhojwani

jsonview/decorators.py

@@ -159,3 +159,31 @@ def _wrapped(request, *a, **kw):
         return decorator(args[0])
     else:
         return decorator
+
+
+def json_request(*args, **kwargs):
+    assume_json = bool(kwargs.get('assume_json', True))
+
+    def decorator(f):
+        @wraps(f)
+        def _wrapped(request, *a, **kw):
+            request.data = {}
+            if assume_json or request.META.get('CONTENT_TYPE') == JSON:
+                try:
+                    if isinstance(request.body, bytes):
+                        request.data = json.loads(request.body.decode('ascii'))
+                    else:
+                        request.data = json.loads(request.body)
+                except:
+                    pass
+            elif request.method == 'GET':
+                request.data = request.GET.dict()
+            elif request.method == 'POST':
+                request.data = request.POST.dict()
+            return f(request, *a, **kw)
+        return _wrapped
+
+    if len(args) == 1 and callable(args[0]):
+        return decorator(args[0])
+    else:
+        return decorator

jsonview/tests.py

@@ -17,7 +17,7 @@
 
 import mock
 
-from .decorators import json_view
+from .decorators import json_view, json_request
 from .exceptions import BadRequest
 
 
@@ -29,6 +29,7 @@ def eq_(a, b, msg=None):
     """From nose.tools.eq_."""
     assert a == b, msg or '%r != %r' % (a, b)
 
+
 if sys.version < '3':
     def b(x):
         return x
@@ -326,3 +327,101 @@ def get(self, request):
         eq_(JSON, res['content-type'])
         data = json.loads(res.content.decode('utf-8'))
         eq_('bar', data['foo'])
+
+
+class JsonRequestTests(TestCase):
+    def test_application_json(self):
+        data = {
+            'foo': 'bar',
+            'baz': 'qux',
+            'quz': [{'foo': 'bar'}],
+        }
+
+        @json_request
+        def temp(req):
+            return req.data
+
+        res = temp(rf.post(
+                '/',
+                data=json.dumps(data),
+                content_type='application/json'
+            ))
+        eq_(res, data)
+
+    def test_get_requests(self):
+        data = {
+            'foo': 'bar',
+            'baz': '0'
+        }
+
+        @json_request(assume_json=False)
+        def temp(req):
+            return req.data
+
+        res = temp(rf.get('/?foo=bar&baz=0'))
+        eq_(res, data)
+
+    def test_post_requests(self):
+        data = {
+            'foo': 'bar',
+            'baz': '0'
+        }
+
+        @json_request(assume_json=False)
+        def temp(req):
+            return req.data
+
+        # test application/x-www-form-urlencoded
+        res = temp(rf.post(
+                '/',
+                data='foo=bar&baz=0',
+                content_type='application/x-www-form-urlencoded'
+            ))
+        eq_(res, data)
+
+        # test multipart/form-data
+        res = temp(rf.post('/', data=data, files=None))
+        eq_(res, data)
+
+    def test_assume_json(self):
+        data = {
+            'foo': 'bar',
+            'baz': '0'
+        }
+
+        @json_request(assume_json=True)
+        def temp(req):
+            return req.data
+
+        @json_request(assume_json=False)
+        def temp_2(req):
+            return req.data
+
+        # test get request
+        res = temp(rf.get('/?foo=bar&baz=0'))
+        eq_(res, {})
+
+        res = temp_2(rf.get('/?foo=bar&baz=0'))
+        eq_(res, data)
+
+        # test application/x-www-form-urlencoded
+        res = temp(rf.post(
+                '/',
+                data='foo=bar&baz=0',
+                content_type='application/x-www-form-urlencoded'
+            ))
+        eq_(res, {})
+
+        res = temp_2(rf.post(
+                '/',
+                data='foo=bar&baz=0',
+                content_type='application/x-www-form-urlencoded'
+            ))
+        eq_(res, data)
+
+        # test multipart/form-data
+        res = temp(rf.post('/', data=data, files=None))
+        eq_(res, {})
+
+        res = temp_2(rf.post('/', data=data, files=None))
+        eq_(res, data)