Added inflection for proper camelization/underscore/pluralization

example/tests/test_model_viewsets.py

@@ -1,6 +1,5 @@
-
-
 import json
+
 from example.tests import TestBase
 from django.contrib.auth import get_user_model
 from django.core.urlresolvers import reverse, reverse_lazy
@@ -9,9 +8,11 @@
 
 class ModelViewSetTests(TestBase):
     """
-    Test usage with ModelViewSets
+    Test usage with ModelViewSets, also tests pluralization, camelization,
+    and underscore.
 
-    [<RegexURLPattern user-list ^user-viewsets/$>, <RegexURLPattern user-detail ^user-viewsets/(?P<pk>[^/]+)/$>]
+    [<RegexURLPattern user-list ^user-viewsets/$>,
+    <RegexURLPattern user-detail ^user-viewsets/(?P<pk>[^/]+)/$>]
     """
     list_url = reverse_lazy('user-list')
 
@@ -21,29 +22,31 @@ def setUp(self):
 
     def test_key_in_list_result(self):
         """
-        Ensure the result has a "user" key since that is the name of the model
+        Ensure the result has a 'user' key since that is the name of the model
         """
         response = self.client.get(self.list_url)
         self.assertEqual(response.status_code, 200)
 
         user = get_user_model().objects.all()[0]
-        expected = {"user": [{
-            'id': user.pk,
-            'first_name': user.first_name,
-            'last_name': user.last_name,
-            'email': user.email
-        }]}
+        expected = {
+            u'user': [{
+                u'id': user.pk,
+                u'firstName': user.first_name,
+                u'lastName': user.last_name,
+                u'email': user.email
+            }]
+        }
 
         json_content = json.loads(response.content)
-        meta = json_content.get("meta")
+        meta = json_content.get('meta')
 
         self.assertEquals(expected.get('user'), json_content.get('user'))
         self.assertEquals(meta.get('count', 0),
             get_user_model().objects.count())
-        self.assertEquals(meta.get("next"), 2)
+        self.assertEquals(meta.get('next'), 2)
         self.assertEqual('http://testserver/user-viewset/?page=2',
-            meta.get("next_link"))
-        self.assertEqual(meta.get("page"), 1)
+            meta.get('nextLink'))
+        self.assertEqual(meta.get('page'), 1)
 
     def test_page_two_in_list_result(self):
         """
@@ -53,72 +56,75 @@ def test_page_two_in_list_result(self):
         self.assertEqual(response.status_code, 200)
 
         user = get_user_model().objects.all()[1]
-        expected = {"user": [{
-            'id': user.pk,
-            'first_name': user.first_name,
-            'last_name': user.last_name,
-            'email': user.email
-        }]}
+        expected = {
+            u'user': [{
+                u'id': user.pk,
+                u'firstName': user.first_name,
+                u'lastName': user.last_name,
+                u'email': user.email
+            }]
+        }
 
         json_content = json.loads(response.content)
-        meta = json_content.get("meta")
+        meta = json_content.get('meta')
 
         self.assertEquals(expected.get('user'), json_content.get('user'))
         self.assertEquals(meta.get('count', 0),
             get_user_model().objects.count())
-        self.assertIsNone(meta.get("next"))
-        self.assertIsNone(meta.get("next_link"))
-        self.assertEqual(meta.get("previous"), 1)
+        self.assertIsNone(meta.get('next'))
+        self.assertIsNone(meta.get('next_link'))
+        self.assertEqual(meta.get('previous'), 1)
         self.assertEqual('http://testserver/user-viewset/?page=1',
-            meta.get("previous_link"))
-        self.assertEqual(meta.get("page"), 2)
+            meta.get('previousLink'))
+        self.assertEqual(meta.get('page'), 2)
 
     def test_page_range_in_list_result(self):
         """
-        Ensure that the range of a page can be changed from the client.
+        Ensure that the range of a page can be changed from the client,
+        tests pluralization as two objects means it converts ``user`` to
+        ``users``.
         """
         response = self.client.get(self.list_url, {'page_size': 2})
         self.assertEqual(response.status_code, 200)
 
         users = get_user_model().objects.all()
-        expected = {"user": [
-        {
-            'id': users[0].pk,
-            'first_name': users[0].first_name,
-            'last_name': users[0].last_name,
-            'email': users[0].email
-        },
-        {
-            'id': users[1].pk,
-            'first_name': users[1].first_name,
-            'last_name': users[1].last_name,
-            'email': users[1].email
-        }]}
+        expected = {
+            u'users': [{
+                u'id': users[0].pk,
+                u'firstName': users[0].first_name,
+                u'lastName': users[0].last_name,
+                u'email': users[0].email
+            },{
+                u'id': users[1].pk,
+                u'firstName': users[1].first_name,
+                u'lastName': users[1].last_name,
+                u'email': users[1].email
+            }]
+        }
 
         json_content = json.loads(response.content)
-        meta = json_content.get("meta")
+        meta = json_content.get('meta')
         self.assertEquals(expected.get('user'), json_content.get('user'))
         self.assertEquals(meta.get('count', 0),
             get_user_model().objects.count())
 
 
     def test_key_in_detail_result(self):
         """
-        Ensure the result has a "user" key.
+        Ensure the result has a 'user' key.
         """
         response = self.client.get(self.detail_url)
         self.assertEqual(response.status_code, 200)
 
         result = json.loads(response.content)
         expected = {
-            'user': {
-                'id': self.miles.pk,
-                'first_name': self.miles.first_name,
-                'last_name': self.miles.last_name,
-                'email': self.miles.email
+            u'user': {
+                u'id': self.miles.pk,
+                u'firstName': self.miles.first_name,
+                u'lastName': self.miles.last_name,
+                u'email': self.miles.email
             }
         }
-
         self.assertEqual(result, expected)
 
     def test_key_in_post(self):
@@ -127,11 +133,11 @@ def test_key_in_post(self):
         """
         self.client.login(username='miles', password='pw')
         data = {
-            'user': {
-                'id': self.miles.pk,
-                'first_name': self.miles.first_name,
-                'last_name': self.miles.last_name,
-                'email': 'miles@trumpet.org'
+            u'user': {
+                u'id': self.miles.pk,
+                u'firstName': self.miles.first_name,
+                u'lastName': self.miles.last_name,
+                u'email': 'miles@trumpet.org'
             }
         }
         response = self.client.put(self.detail_url, data=data, format='json')
@@ -145,4 +151,3 @@ def test_key_in_post(self):
         self.assertEqual(
             get_user_model().objects.get(pk=self.miles.pk).email,
             'miles@trumpet.org')
-

example/tests/test_multiple_id_mixin.py

@@ -1,5 +1,3 @@
-
-
 import json
 from example.tests import TestBase
 from django.contrib.auth import get_user_model
@@ -25,22 +23,22 @@ def test_single_id_in_query_params(self):
         self.assertEqual(response.status_code, 200)
 
         expected = {
-            'user': [{
-                'id': self.miles.pk,
-                'first_name': self.miles.first_name,
-                'last_name': self.miles.last_name,
-                'email': self.miles.email
+            u'user': [{
+                u'id': self.miles.pk,
+                u'firstName': self.miles.first_name,
+                u'lastName': self.miles.last_name,
+                u'email': self.miles.email
             }]
         }
 
         json_content = json.loads(response.content)
-        meta = json_content.get("meta")
+        meta = json_content.get('meta')
 
         self.assertEquals(expected.get('user'), json_content.get('user'))
         self.assertEquals(meta.get('count', 0), 1)
-        self.assertEquals(meta.get("next"), None)
-        self.assertEqual(None, meta.get("next_link"))
-        self.assertEqual(meta.get("page"), 1)
+        self.assertEquals(meta.get('next'), None)
+        self.assertEqual(None, meta.get('nextLink'))
+        self.assertEqual(meta.get('page'), 1)
 
     def test_multiple_ids_in_query_params(self):
         """
@@ -52,22 +50,22 @@ def test_multiple_ids_in_query_params(self):
         self.assertEqual(response.status_code, 200)
 
         expected = {
-            'user': [{
-                'id': self.john.pk,
-                'first_name': self.john.first_name,
-                'last_name': self.john.last_name,
-                'email': self.john.email
+            u'user': [{
+                u'id': self.john.pk,
+                u'firstName': self.john.first_name,
+                u'lastName': self.john.last_name,
+                u'email': self.john.email
             }]
         }
 
         json_content = json.loads(response.content)
-        meta = json_content.get("meta")
+        meta = json_content.get('meta')
 
         self.assertEquals(expected.get('user'), json_content.get('user'))
         self.assertEquals(meta.get('count', 0), 2)
-        self.assertEquals(meta.get("next"), 2)
+        self.assertEquals(meta.get('next'), 2)
         self.assertEqual(
             'http://testserver/user-mixin-viewset/?ids%5B%5D=2&ids%5B%5D=1&page=2',
-            meta.get("next_link"))
-        self.assertEqual(meta.get("page"), 1)
+            meta.get('nextLink'))
+        self.assertEqual(meta.get('page'), 1)
 

rest_framework_ember/parsers.py

@@ -1,8 +1,9 @@
 """
 Parsers
 """
-from rest_framework.parsers import JSONParser
+import inflection
 
+from rest_framework.parsers import JSONParser
 from rest_framework_ember.utils import get_resource_name
 
 
@@ -21,14 +22,14 @@ class EmberJSONParser(JSONParser):
     So we can work with the grain on both Ember and RestFramework,
     Do some tweaks to the payload so DRF gets what it expects.
     """
-
     def parse(self, stream, media_type=None, parser_context=None):
         """
         Parses the incoming bytestream as JSON and returns the resulting data
         """
-        result = super(EmberJSONParser, self).parse(
-            stream, media_type=None, parser_context=None)
-
-        resource_name = get_resource_name(parser_context.get('view', None))
-        return result.get(resource_name)
+        result = super(EmberJSONParser, self).parse(stream, media_type=None,
+                                                    parser_context=None)
+        resource = result.get(get_resource_name(parser_context.get('view', None)))
+        for item in resource:
+            resource[inflection.underscore(item)] = resource.pop(item)
+        return resource
 

rest_framework_ember/renderers.py

@@ -1,7 +1,10 @@
 import copy
-from rest_framework import renderers
+import inflection
+
+from django.conf import settings
 
-from rest_framework_ember.utils import get_resource_name
+from rest_framework import renderers
+from rest_framework_ember.utils import get_resource_name, get_key
 
 
 class JSONRenderer(renderers.JSONRenderer):
@@ -17,19 +20,42 @@ class JSONRenderer(renderers.JSONRenderer):
     }
     """
     def render(self, data, accepted_media_type=None, renderer_context=None):
-        view = renderer_context.get('view')
 
+        view = renderer_context.get('view')
         resource_name = get_resource_name(view)
 
         if resource_name == False:
             return super(JSONRenderer, self).render(
                 data, accepted_media_type, renderer_context)
 
+        if isinstance(data, dict):
+            for key, value in data.items():
+                data[inflection.camelize(key, False)] = data.pop(key)
+
+        elif isinstance(data, (list, tuple)):
+            for obj in data:
+                if hasattr(obj, 'items'):
+                    for key, value in obj.items():
+                        obj[get_key(key)] = obj.pop(key)
+
+            if getattr(settings, 'REST_FRAMEWORK_PLURALIZE_KEYS', False):
+                if len(data) > 1:
+                    resource_name = inflection.pluralize(resource_name)
+
         try:
             data_copy = copy.copy(data)
             content = data_copy.pop('results')
             data = {resource_name : content, "meta" : data_copy}
         except (TypeError, KeyError, AttributeError) as e:
+
+            # Default behavior
+            if not resource_name == 'data':
+                if isinstance(data, dict):
+                    for key, value in data.items():
+                        data[get_key(key)] = data.pop(key)
+                else:
+                    data = {resource_name: data}
+
             data = {resource_name : data}
         return super(JSONRenderer, self).render(
             data, accepted_media_type, renderer_context)