encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-02-02 20:54:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixes to save_object

Browse Source
This commit is contained in:
Tom Christie 2013-03-16 07:35:27 +00:00
parent ccf551201f
commit 3ff103ad04
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
rest_framework/serializers.py
View File

@ -691,21 +691,23 @@ class ModelSerializer(Serializer):
if getattr(obj, '_m2m_data', None): if getattr(obj, '_m2m_data', None):
for accessor_name, object_list in obj._m2m_data.items(): for accessor_name, object_list in obj._m2m_data.items():
setattr(self.object, accessor_name, object_list) setattr(obj, accessor_name, object_list)
obj._m2m_data = {} del(obj._m2m_data)
if getattr(obj, '_related_data', None): if getattr(obj, '_related_data', None):
for accessor_name, related in obj._related_data.items(): for accessor_name, related in obj._related_data.items():
if related is None: if related is None:
previous = getattr(self.object, accessor_name, related) previous = getattr(obj, accessor_name, related)
previous.delete() if previous:
previous.delete()
elif isinstance(related, models.Model): elif isinstance(related, models.Model):
fk_field = obj._meta.get_field_by_name(accessor_name)[0].field.name fk_field = obj._meta.get_field_by_name(accessor_name)[0].field.name
setattr(related, fk_field, obj) setattr(related, fk_field, obj)
self.save_object(related) self.save_object(related)
else: else:
setattr(self.object, accessor_name, related) setattr(self.object, accessor_name, related)
obj._related_data = {} setattr(obj, accessor_name, related)
del(obj._related_data)
class HyperlinkedModelSerializerOptions(ModelSerializerOptions): class HyperlinkedModelSerializerOptions(ModelSerializerOptions):