diff options
| author | Craig Blaszczyk | 2011-07-08 18:04:22 +0100 | 
|---|---|---|
| committer | Craig Blaszczyk | 2011-07-08 18:04:22 +0100 | 
| commit | 344db0d733eed8d3be84222a549a651acd6f3ee8 (patch) | |
| tree | 8814e7f1b5dc5cc927e3c37b76f85d2ea3cf841e /djangorestframework/mixins.py | |
| parent | 6e6b35b5c03d66f612e55c29e11d3096cd2703ac (diff) | |
| download | django-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.py | 17 | 
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'):  | 
