diff options
| author | Tom Christie | 2012-02-24 05:12:23 -0800 | 
|---|---|---|
| committer | Tom Christie | 2012-02-24 05:12:23 -0800 | 
| commit | f041ee023ea0c995feb361e3a8e9be321fc59413 (patch) | |
| tree | 3337db3efe9befbb8ca957c9163e036c44e50163 | |
| parent | 718381523543d226c30f374dd234a535f2a7a5f7 (diff) | |
| parent | 9c92f96ce2299667ef3393c6d63ffe7a83c89e4a (diff) | |
| download | django-rest-framework-f041ee023ea0c995feb361e3a8e9be321fc59413.tar.bz2 | |
Merge pull request #179 from shawnlewis/master
Okay, that looks good to me!
| -rw-r--r-- | djangorestframework/serializer.py | 29 | ||||
| -rw-r--r-- | djangorestframework/tests/serializer.py | 21 | 
2 files changed, 28 insertions, 22 deletions
diff --git a/djangorestframework/serializer.py b/djangorestframework/serializer.py index b0c02675..5dea37e8 100644 --- a/djangorestframework/serializer.py +++ b/djangorestframework/serializer.py @@ -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): @@ -110,9 +103,6 @@ class Serializer(object):          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 @@ -123,9 +113,6 @@ class Serializer(object):              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): @@ -139,9 +126,7 @@ class Serializer(object):          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 @@ -175,11 +160,11 @@ class Serializer(object):          """          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 @@ -219,7 +204,7 @@ class Serializer(object):          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)' @@ -237,7 +222,7 @@ class Serializer(object):                      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 diff --git a/djangorestframework/tests/serializer.py b/djangorestframework/tests/serializer.py index e8580610..834a60d0 100644 --- a/djangorestframework/tests/serializer.py +++ b/djangorestframework/tests/serializer.py @@ -104,6 +104,27 @@ class TestFieldNesting(TestCase):          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  | 
