From 338e77837c9df6f36c77db67a65a3a485b1064a4 Mon Sep 17 00:00:00 2001 From: Chris Pickett Date: Thu, 19 Jan 2012 12:56:09 -0500 Subject: [PATCH] Added get_queryset method to ListModelMixin. This can be easily re-implemented in a view, to allow the user to take such things as request.user into account when creating the queryset. --- djangorestframework/mixins.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/djangorestframework/mixins.py b/djangorestframework/mixins.py index 15d7faf0e..c7f7b10cd 100644 --- a/djangorestframework/mixins.py +++ b/djangorestframework/mixins.py @@ -675,9 +675,7 @@ class ListModelMixin(ModelMixin): queryset = None def get(self, request, *args, **kwargs): - model = self.resource.model - - queryset = self.queryset if self.queryset is not None else model.objects.all() + queryset = self.get_queryset() if hasattr(self, 'resource'): ordering = getattr(self.resource, 'ordering', None) @@ -689,6 +687,10 @@ class ListModelMixin(ModelMixin): queryset = queryset.order_by(*args) return queryset.filter(self.build_query(**kwargs)) + def get_queryset(self): + model = self.resource.model + return model.objects.all() if self.queryset is None else self.queryset + ########## Pagination Mixins ##########