Mishandled lists with primitive types

Author: Mike Hearing

pyrestorm/models.py

@@ -5,6 +5,8 @@
 from pyrestorm.exceptions import orm as orm_exceptions
 from pyrestorm.manager import RestOrmManager
 
+primitives = [int, str, unicode, bool]
+
 
 class RestModelBase(type):
     # Called when class is imported got guarantee proper setup of child classes to parent
@@ -116,12 +118,15 @@ def _serialize_data(self, obj):
                 continue
 
             value_type = type(value)
-            if value_type not in [int, str, unicode, bool]:
+            if value_type not in primitives:
                 # Nested structure
                 if value_type == list:
                     local_diff[key] = copy.deepcopy(value)
                     for idx, inner_value in enumerate(value):
-                        local_diff[key][idx] = self._serialize_data(inner_value)
+                        if type(inner_value) not in primitives:
+                            local_diff[key][idx] = self._serialize_data(inner_value)
+                        else:
+                            local_diff[key][idx] = inner_value
                 # Object/Dictionary
                 else:
                     self._serialize_data(value)

tests/test_models.py

@@ -71,6 +71,7 @@ def test_restmodel_nestedstructures(self):
         post = Post(title='Hello', body='World', userId=1)
         post.author = Author(name='Michael Hearing')
         post.comments = [Comment(body='Are we having fun yet?'), Comment(body='Hoe about now?')]
+        post.fun_numbers = [1, 2, 3, 5, 7]
         post.meta = {'datetime': 'September 23, 2016'}
         post.save()
         self.assertEqual(post.id, 101)