diff options
| author | Tom Christie | 2014-09-18 14:58:08 +0100 |
|---|---|---|
| committer | Tom Christie | 2014-09-18 14:58:08 +0100 |
| commit | 106362b437f45e04faaea759df57a66a8a2d7cfd (patch) | |
| tree | 22410f7d4d0e8ccec5c4891505a1b4b6974d68af /rest_framework/serializers.py | |
| parent | 9fdb2280d11db126771686d626aa8a0247b8a46c (diff) | |
| download | django-rest-framework-106362b437f45e04faaea759df57a66a8a2d7cfd.tar.bz2 | |
ModelSerializer.create() to handle many to many by default
Diffstat (limited to 'rest_framework/serializers.py')
| -rw-r--r-- | rest_framework/serializers.py | 20 |
1 files changed, 19 insertions, 1 deletions
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(): |
