implement better support for encoders default method

RestResponse/__init__.py

@@ -1,5 +1,6 @@
 from .objects import (
-    RestEncoder, RestResponse, RestObject, RestList, NoneProp, RestResponseObj, ApiModel, ApiCollection
+    RestEncoder, RestResponse, RestObject, RestList, NoneProp, RestResponseObj, ApiModel, ApiCollection,
+    RestEncoderSimple
 )
 from . import orm
 
@@ -8,5 +9,5 @@
 
 __all__ = [
     'RestEncoder', 'RestResponse', 'RestObject', 'RestList', 'NoneProp', 'parse', 'loads', 'orm',
-    'RestResponseObj', 'ApiModel', 'ApiCollection'
+    'RestResponseObj', 'ApiModel', 'ApiCollection', 'RestEncoderSimple'
 ]

RestResponse/objects.py

@@ -13,8 +13,12 @@
 
 def _encode_hook(opts):
     def _default(o):
+        if isinstance(o, RestObject):
+            return {k: _default(v) for k, v in o.items()}
+        elif isinstance(o, RestList) or isinstance(o, ApiCollection):
+            return [_default(x) for x in o]
         if isinstance(o, ApiModel):
-            return o._data
+            return _default(o._data)
         return utils.encode_item(o, **opts)
     return _default
 

setup.py

@@ -6,7 +6,7 @@ def read(fname):
     return open(os.path.join(os.path.dirname(__file__), fname)).read()
 
 
-__version__ = '3.0.4'
+__version__ = '3.0.5'
 
 setup(
     name='RestResponse',

tests/test_objects.py

@@ -503,3 +503,29 @@ def test_api_collection():
         m.ref_collection.append(ref)
         assert m.ref_collection[-1].id == ref.id
         assert m.ref_collection[-1].string == ref.string
+
+
+def test_default():
+    obj = RestResponse.parse({
+        'callable': lambda x: x + 1
+    })
+
+    data = RestResponse.RestEncoder().default(obj)
+    assert data['callable'].startswith('__callable__')
+    assert RestResponse.utils.decode_item(data['callable'])(2) == 3
+
+    data = RestResponse.RestEncoderSimple().default(obj)
+    assert data['callable'].startswith('__callable__')
+    assert RestResponse.utils.decode_item(data['callable'])(2) == 3
+
+    obj = RestResponse.parse([
+        lambda x: x + 1
+    ])
+
+    data = RestResponse.RestEncoder().default(obj)
+    assert data[0].startswith('__callable__')
+    assert RestResponse.utils.decode_item(data[0])(2) == 3
+
+    data = RestResponse.RestEncoderSimple().default(obj)
+    assert data[0].startswith('__callable__')
+    assert RestResponse.utils.decode_item(data[0])(2) == 3