Added possibility to specify what attributes of ForeignKeys and OneToOne fields to include via nested tuples

1 files changed, 13 insertions, 1 deletions

diff --git a/djangorestframework/resources.py b/djangorestframework/resources.py
index 36bad914..4b81bf79 100644
--- a/djangorestframework/resources.py
+++ b/djangorestframework/resources.py
@@ -62,6 +62,12 @@ def _model_to_dict(instance, resource=None):
     
     # Method fields
     for fname in extra_fields:
+        
+        if isinstance(fname, (tuple, list)):
+            fname, fields = fname
+        else:
+            fname, fields = fname, False
+
         try:
             if hasattr(resource, fname):
                 # check the resource first, to allow it to override fields
@@ -77,7 +83,13 @@ def _model_to_dict(instance, resource=None):
     
             # TODO: It would be nicer if this didn't recurse here.
             # Let's keep _model_to_dict flat, and _object_to_data recursive.
-            data[fname] = _object_to_data(obj)
+            if fields:
+                Resource = type('Resource', (object,), {'fields': fields,
+                                                        'include': (),
+                                                        'exclude': ()})
+                data[fname] = _object_to_data(obj, Resource())
+            else:
+                data[fname] = _object_to_data(obj)
 
         except NoReverseMatch:
             # Ug, bit of a hack for now