nested save for deserialized objects

This commit is contained in:
Dan Stephenson 2013-09-10 16:21:34 +01:00
parent 5cfea7b933
commit 5214615bff

View File

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