ObjectId serialisation on ListField(ReferenceField) fix

Author: Umut Bozkurt

rest_framework_mongoengine/fields.py

@@ -5,6 +5,8 @@
 from mongoengine.base.document import BaseDocument
 from mongoengine.document import Document
 from rest_framework import serializers
+from mongoengine.fields import ObjectId
+
 import sys
 
 if sys.version_info[0] >= 3:
@@ -66,8 +68,8 @@ def transform_object(self, obj, depth):
         elif obj is None:
             return None
         else:
-            # Default to string
-            return unicode(obj)
+            print obj
+            return unicode(obj) if isinstance(obj, ObjectId) else obj
 
 
 class ReferenceField(MongoDocumentField):
@@ -101,7 +103,7 @@ def from_native(self, value):
         return self.model_field.to_python(value)
 
     def to_native(self, obj):
-        return [self.model_field.to_mongo(list_item) for list_item in obj]
+        return self.transform_object(obj, self.depth)
 
 
 class EmbeddedDocumentField(MongoDocumentField):
@@ -134,4 +136,4 @@ class DynamicField(MongoDocumentField):
     type_label = 'DynamicField'
 
     def to_native(self, obj):
-        return self.model_field.to_python(obj)
+        return self.model_field.to_python(obj)