diff options
| author | Tom Christie | 2012-12-29 16:51:41 +0000 |
|---|---|---|
| committer | Tom Christie | 2012-12-29 16:51:41 +0000 |
| commit | caaf1c80f5bb869da26de2921be7f91aeeb7ae71 (patch) | |
| tree | 4d5eb8e233d6f29cc160f8458292d99bdc477e8c /rest_framework/serializers.py | |
| parent | 51e2664491577cc218dfa0643e0c20532b3eb96e (diff) | |
| parent | 008dafce178181855d66981cfacb908d013c5d1d (diff) | |
| download | django-rest-framework-caaf1c80f5bb869da26de2921be7f91aeeb7ae71.tar.bz2 | |
Support reverse FK creation
Diffstat (limited to 'rest_framework/serializers.py')
| -rw-r--r-- | rest_framework/serializers.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/rest_framework/serializers.py b/rest_framework/serializers.py index 6eb9c3e1..e8e6735a 100644 --- a/rest_framework/serializers.py +++ b/rest_framework/serializers.py @@ -496,19 +496,26 @@ class ModelSerializer(Serializer): Restore the model instance. """ self.m2m_data = {} + self.related_data = {} if instance is not None: for key, val in attrs.items(): setattr(instance, key, val) else: - # Reverse relations + # Reverse fk relations + for (obj, model) in self.opts.model._meta.get_all_related_objects_with_model(): + field_name = obj.field.related_query_name() + if field_name in attrs: + self.related_data[field_name] = attrs.pop(field_name) + + # Reverse m2m relations 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 relations + # Forward m2m relations for field in self.opts.model._meta.many_to_many: if field.name in attrs: self.m2m_data[field.name] = attrs.pop(field.name) @@ -534,6 +541,11 @@ class ModelSerializer(Serializer): setattr(self.object, accessor_name, object_list) self.m2m_data = {} + if getattr(self, 'related_data', None): + for accessor_name, object_list in self.related_data.items(): + setattr(self.object, accessor_name, object_list) + self.related_data = {} + return self.object |
