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.
This commit is contained in:
Chris Pickett 2012-01-19 12:56:09 -05:00
parent 0a5ca000ed
commit 338e77837c

View File

@ -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 ##########