encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-05 13:00:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

nested save for deserialized objects

Browse Source
This commit is contained in:
Dan Stephenson 2013-09-10 16:21:34 +01:00
parent 5cfea7b933
commit 5214615bff
1 changed files with 51 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
rest_framework/serializers.py
View File

@ -887,6 +887,10 @@ class ModelSerializer(Serializer):
""" """
Save the deserialized object and return it. Save the deserialized object and return it.
""" """
def save_nested_forward_relations(obj):
"""
Save nested nested forward relations
"""
if getattr(obj, '_nested_forward_relations', None): if getattr(obj, '_nested_forward_relations', None):
# Nested relationships need to be saved before we can save the # Nested relationships need to be saved before we can save the
# parent instance. # parent instance.
@ -897,18 +901,25 @@ class ModelSerializer(Serializer):
obj.save(**kwargs) obj.save(**kwargs)
if getattr(obj, '_m2m_data', None): def save_m2m(obj):
if hasattr(obj._m2m_data, '__iter__'): """
Save nested ManyToMany relations
"""
if getattr(obj, '_m2m_data', None) and hasattr(obj._m2m_data, '__iter__'):
for accessor_name, object_list in obj._m2m_data.items(): for accessor_name, object_list in obj._m2m_data.items():
if object_list: if hasattr(object_list, '__iter__'):
for m2m_object in object_list: for m2m_object in object_list:
for field in m2m_object.__dict__.keys(): save_nested_forward_relations(m2m_object)
if not field == 'id': save_m2m(m2m_object)
attr = getattr(m2m_object, field) save_related_data(m2m_object)
setattr(m2m_object, field, m2m_object.__dict__[field]) setattr(obj, accessor_name, object_list)
m2m_object.save() m2m_object.save()
del(obj._m2m_data) del(obj._m2m_data)
def save_related_data(obj):
"""
Save nested nested related 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 isinstance(related, RelationsList): if isinstance(related, RelationsList):
@ -932,6 +943,11 @@ class ModelSerializer(Serializer):
setattr(obj, accessor_name, related) setattr(obj, accessor_name, related)
del(obj._related_data) del(obj._related_data)
# Save
save_nested_forward_relations(obj)
save_m2m(obj)
save_related_data(obj)
class HyperlinkedModelSerializerOptions(ModelSerializerOptions): class HyperlinkedModelSerializerOptions(ModelSerializerOptions):
""" """