From 106362b437f45e04faaea759df57a66a8a2d7cfd Mon Sep 17 00:00:00 2001
From: Tom Christie
Date: Thu, 18 Sep 2014 14:58:08 +0100
Subject: ModelSerializer.create() to handle many to many by default

---
 rest_framework/serializers.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

(limited to 'rest_framework/serializers.py')

diff --git a/rest_framework/serializers.py b/rest_framework/serializers.py
index 9f3e53fd..03e20df8 100644
--- a/rest_framework/serializers.py
+++ b/rest_framework/serializers.py
@@ -344,7 +344,25 @@ class ModelSerializer(Serializer):
 
     def create(self, attrs):
         ModelClass = self.Meta.model
-        return ModelClass.objects.create(**attrs)
+
+        # Remove many-to-many relationships from attrs.
+        # They are not valid arguments to the default `.create()` method,
+        # as they require that the instance has already been saved.
+        info = model_meta.get_field_info(ModelClass)
+        many_to_many = {}
+        for key, relation_info in info.relations.items():
+            if relation_info.to_many and (key in attrs):
+                many_to_many[key] = attrs.pop(key)
+
+        instance = ModelClass.objects.create(**attrs)
+
+        # Save many to many relationships after the instance is created.
+        if many_to_many:
+            for key, value in many_to_many.items():
+                setattr(instance, key, value)
+            instance.save()
+
+        return instance
 
     def update(self, obj, attrs):
         for attr, value in attrs.items():
-- 
cgit v1.2.3