diff options
Diffstat (limited to 'djangorestframework/modelresource.py')
| -rw-r--r-- | djangorestframework/modelresource.py | 15 | 
1 files changed, 12 insertions, 3 deletions
| diff --git a/djangorestframework/modelresource.py b/djangorestframework/modelresource.py index 0123e77a..0b06dacb 100644 --- a/djangorestframework/modelresource.py +++ b/djangorestframework/modelresource.py @@ -1,6 +1,7 @@  from django.forms import ModelForm -from django.db.models.query import QuerySet  from django.db.models import Model +from django.db.models.query import QuerySet +from django.db.models.fields.related import RelatedField  from djangorestframework.response import Response, ResponseException  from djangorestframework.resource import Resource @@ -340,6 +341,13 @@ class ModelResource(Resource, ModelFormValidatorMixin):      def post(self, request, auth, content, *args, **kwargs):          # TODO: test creation on a non-existing resource url +         +        # translated related_field into related_field_id +        for related_name in [field.name for field in self.model._meta.fields if isinstance(field, RelatedField)]: +            if kwargs.has_key(related_name): +                kwargs[related_name + '_id'] = kwargs[related_name] +                del kwargs[related_name] +          all_kw_args = dict(content.items() + kwargs.items())          if args:              instance = self.model(pk=args[-1], **all_kw_args) @@ -373,6 +381,7 @@ class ModelResource(Resource, ModelFormValidatorMixin):              else:                  # Otherwise assume the kwargs uniquely identify the model                  instance = self.model.objects.get(**kwargs) +              for (key, val) in content.items():                  setattr(instance, key, val)          except self.model.DoesNotExist: @@ -404,7 +413,7 @@ class RootModelResource(ModelResource):      def get(self, request, auth, *args, **kwargs):          queryset = self.queryset if self.queryset else self.model.objects.all() -        return queryset +        return queryset.filter(**kwargs)  class QueryModelResource(ModelResource): @@ -418,5 +427,5 @@ class QueryModelResource(ModelResource):      def get(self, request, auth, *args, **kwargs):          queryset = self.queryset if self.queryset else self.model.objects.all() -        return queryset +        return queryset.filer(**kwargs) | 
