aboutsummaryrefslogtreecommitdiffstats
path: root/rest_framework
diff options
context:
space:
mode:
authorTomi Pajunen2012-11-08 16:31:16 +0200
committerTomi Pajunen2012-11-08 16:31:16 +0200
commit743224d000bbc178dafcff1e9315f1d74bee44de (patch)
treec9f2acab5c420ed109aa4edbd0c80cadd54cbdbc /rest_framework
parent066d51faa16d1cfd3c8370c6bfe46f8494bbc26a (diff)
downloaddjango-rest-framework-743224d000bbc178dafcff1e9315f1d74bee44de.tar.bz2
Fixed creation of objects with reversed M2M relations
Diffstat (limited to 'rest_framework')
-rw-r--r--rest_framework/serializers.py7
-rw-r--r--rest_framework/tests/pk_relations.py18
2 files changed, 25 insertions, 0 deletions
diff --git a/rest_framework/serializers.py b/rest_framework/serializers.py
index 4f68ada6..e4501679 100644
--- a/rest_framework/serializers.py
+++ b/rest_framework/serializers.py
@@ -447,6 +447,13 @@ class ModelSerializer(Serializer):
setattr(instance, key, val)
return instance
+ # Reversed relation
+ for (obj, model) in self.opts.model._meta.get_all_related_m2m_objects_with_model():
+ field_name = obj.field.related_query_name()
+ if field_name in attrs:
+ self.m2m_data[field_name] = attrs.pop(field_name)
+
+ # Forward relation
for field in self.opts.model._meta.many_to_many:
if field.name in attrs:
self.m2m_data[field.name] = attrs.pop(field.name)
diff --git a/rest_framework/tests/pk_relations.py b/rest_framework/tests/pk_relations.py
index 94709810..44ae4040 100644
--- a/rest_framework/tests/pk_relations.py
+++ b/rest_framework/tests/pk_relations.py
@@ -117,6 +117,24 @@ class PrimaryKeyManyToManyTests(TestCase):
]
self.assertEquals(serializer.data, expected)
+ def test_reverse_many_to_many_create(self):
+ data = {'id': 4, 'name': u'target-4', 'sources': [1, 3]}
+ serializer = ManyToManyTargetSerializer(data=data)
+ self.assertTrue(serializer.is_valid())
+ obj = serializer.save()
+ self.assertEquals(serializer.data, data)
+ self.assertEqual(obj.name, u'target-4')
+
+ # Ensure target 4 is added, and everything else is as expected
+ queryset = ManyToManyTarget.objects.all()
+ serializer = ManyToManyTargetSerializer(queryset)
+ expected = [
+ {'id': 1, 'name': u'target-1', 'sources': [1, 2, 3]},
+ {'id': 2, 'name': u'target-2', 'sources': [2, 3]},
+ {'id': 3, 'name': u'target-3', 'sources': [3]},
+ {'id': 4, 'name': u'target-4', 'sources': [1, 3]}
+ ]
+ self.assertEquals(serializer.data, expected)
class PrimaryKeyForeignKeyTests(TestCase):
def setUp(self):