encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-06-29 18:03:16 +03:00
Code Issues Packages Projects Releases Wiki Activity

update mixin to work with m2m data using a through field, by storing the name of the field and manually creating an object in the related table

Browse Source
This commit is contained in:
Craig Blaszczyk 2011-07-08 18:04:22 +01:00
parent 6e6b35b5c0
commit 344db0d733
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
djangorestframework/mixins.py
View File

@ -523,7 +523,10 @@ class CreateModelMixin(object):
for field in model._meta.many_to_many: for field in model._meta.many_to_many:
if content.has_key(field.name): if content.has_key(field.name):
m2m_data[field.name] = content[field.name] m2m_data[field.name] = (
model._meta.many_to_many[0].m2m_reverse_field_name(),
content[field.name]
)
del content[field.name] del content[field.name]
all_kw_args = dict(content.items() + kwargs.items()) all_kw_args = dict(content.items() + kwargs.items())
@ -535,7 +538,17 @@ class CreateModelMixin(object):
instance.save() instance.save()
for fieldname in m2m_data: for fieldname in m2m_data:
getattr(instance, fieldname).add(*m2m_data[fieldname]) manager = getattr(instance, fieldname)
if hasattr(manager, 'add'):
manager.add(*m2m_data[fieldname][1])
else:
data = {}
data[manager.source_field_name] = instance
for related_item in m2m_data[fieldname][1]:
data[m2m_data[fieldname][0]] = related_item
manager.through(**data).save()
headers = {} headers = {}
if hasattr(instance, 'get_absolute_url'): if hasattr(instance, 'get_absolute_url'):