Skip to content

Commit

Permalink
Merge pull request encode#179 from shawnlewis/master
Browse files Browse the repository at this point in the history
Okay, that looks good to me!
  • Loading branch information
tomchristie committed Feb 24, 2012
2 parents 7183815 + 9c92f96 commit f041ee0
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 22 deletions.
29 changes: 7 additions & 22 deletions djangorestframework/serializer.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,16 +25,9 @@ def _field_to_tuple(field):

def _fields_to_list(fields):
"""
Return a list of field names.
Return a list of field tuples.
"""
return [_field_to_tuple(field)[0] for field in fields or ()]


def _fields_to_dict(fields):
"""
Return a `dict` of field name -> None, or tuple of fields, or Serializer class
"""
return dict([_field_to_tuple(field) for field in fields or ()])
return [_field_to_tuple(field) for field in fields or ()]


class _SkipField(Exception):
Expand Down Expand Up @@ -110,9 +103,6 @@ def __init__(self, depth=None, stack=[], **kwargs):
self.stack = stack

def get_fields(self, obj):
"""
Return the set of field names/keys to use for a model instance/dict.
"""
fields = self.fields

# If `fields` is not set, we use the default fields and modify
Expand All @@ -123,9 +113,6 @@ def get_fields(self, obj):
exclude = self.exclude or ()
fields = set(default + list(include)) - set(exclude)

else:
fields = _fields_to_list(self.fields)

return fields

def get_default_fields(self, obj):
Expand All @@ -139,9 +126,7 @@ def get_default_fields(self, obj):
else:
return obj.keys()

def get_related_serializer(self, key):
info = _fields_to_dict(self.fields).get(key, None)

def get_related_serializer(self, info):
# If an element in `fields` is a 2-tuple of (str, tuple)
# then the second element of the tuple is the fields to
# set on the related serializer
Expand Down Expand Up @@ -175,11 +160,11 @@ def serialize_key(self, key):
"""
return self.rename.get(smart_str(key), smart_str(key))

def serialize_val(self, key, obj):
def serialize_val(self, key, obj, related_info):
"""
Convert a model field or dict value into a serializable representation.
"""
related_serializer = self.get_related_serializer(key)
related_serializer = self.get_related_serializer(related_info)

if self.depth is None:
depth = None
Expand Down Expand Up @@ -219,7 +204,7 @@ def serialize_model(self, instance):
fields = self.get_fields(instance)

# serialize each required field
for fname in fields:
for fname, related_info in _fields_to_list(fields):
try:
# we first check for a method 'fname' on self,
# 'fname's signature must be 'def fname(self, instance)'
Expand All @@ -237,7 +222,7 @@ def serialize_model(self, instance):
continue

key = self.serialize_key(fname)
val = self.serialize_val(fname, obj)
val = self.serialize_val(fname, obj, related_info)
data[key] = val
except _SkipField:
pass
Expand Down
21 changes: 21 additions & 0 deletions djangorestframework/tests/serializer.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,6 +104,27 @@ class SerializerM3(Serializer):
self.assertEqual(SerializerM2().serialize(self.m2), {'field': {'field1': u'foo'}})
self.assertEqual(SerializerM3().serialize(self.m3), {'field': {'field2': u'bar'}})

def test_serializer_no_fields(self):
"""
Test related serializer works when the fields attr isn't present. Fix for
#178.
"""
class NestedM2(Serializer):
fields = ('field1', )

class NestedM3(Serializer):
fields = ('field2', )

class SerializerM2(Serializer):
include = [('field', NestedM2)]
exclude = ('id', )

class SerializerM3(Serializer):
fields = [('field', NestedM3)]

self.assertEqual(SerializerM2().serialize(self.m2), {'field': {'field1': u'foo'}})
self.assertEqual(SerializerM3().serialize(self.m3), {'field': {'field2': u'bar'}})

def test_serializer_classname_nesting(self):
"""
Test related model serialization
Expand Down

0 comments on commit f041ee0

Please sign in to comment.