aboutsummaryrefslogtreecommitdiffstats
path: root/rest_framework/serializers.py
diff options
context:
space:
mode:
authorTom Christie2012-12-29 16:51:41 +0000
committerTom Christie2012-12-29 16:51:41 +0000
commitcaaf1c80f5bb869da26de2921be7f91aeeb7ae71 (patch)
tree4d5eb8e233d6f29cc160f8458292d99bdc477e8c /rest_framework/serializers.py
parent51e2664491577cc218dfa0643e0c20532b3eb96e (diff)
parent008dafce178181855d66981cfacb908d013c5d1d (diff)
downloaddjango-rest-framework-caaf1c80f5bb869da26de2921be7f91aeeb7ae71.tar.bz2
Support reverse FK creation
Diffstat (limited to 'rest_framework/serializers.py')
-rw-r--r--rest_framework/serializers.py16
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