aboutsummaryrefslogtreecommitdiffstats
path: root/djangorestframework/mixins.py
diff options
context:
space:
mode:
authorCraig Blaszczyk2011-07-08 18:04:22 +0100
committerCraig Blaszczyk2011-07-08 18:04:22 +0100
commit344db0d733eed8d3be84222a549a651acd6f3ee8 (patch)
tree8814e7f1b5dc5cc927e3c37b76f85d2ea3cf841e /djangorestframework/mixins.py
parent6e6b35b5c03d66f612e55c29e11d3096cd2703ac (diff)
downloaddjango-rest-framework-344db0d733eed8d3be84222a549a651acd6f3ee8.tar.bz2
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
Diffstat (limited to 'djangorestframework/mixins.py')
-rw-r--r--djangorestframework/mixins.py17
1 files changed, 15 insertions, 2 deletions
diff --git a/djangorestframework/mixins.py b/djangorestframework/mixins.py
index 64e541f1..9f65625e 100644
--- a/djangorestframework/mixins.py
+++ b/djangorestframework/mixins.py
@@ -523,7 +523,10 @@ class CreateModelMixin(object):
for field in model._meta.many_to_many:
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]
all_kw_args = dict(content.items() + kwargs.items())
@@ -535,7 +538,17 @@ class CreateModelMixin(object):
instance.save()
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 = {}
if hasattr(instance, 'get_absolute_url'):