Commit 338e7783 by Chris Pickett

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.
parent 0a5ca000
...@@ -675,9 +675,7 @@ class ListModelMixin(ModelMixin): ...@@ -675,9 +675,7 @@ class ListModelMixin(ModelMixin):
queryset = None queryset = None
def get(self, request, *args, **kwargs): def get(self, request, *args, **kwargs):
model = self.resource.model queryset = self.get_queryset()
queryset = self.queryset if self.queryset is not None else model.objects.all()
if hasattr(self, 'resource'): if hasattr(self, 'resource'):
ordering = getattr(self.resource, 'ordering', None) ordering = getattr(self.resource, 'ordering', None)
...@@ -689,6 +687,10 @@ class ListModelMixin(ModelMixin): ...@@ -689,6 +687,10 @@ class ListModelMixin(ModelMixin):
queryset = queryset.order_by(*args) queryset = queryset.order_by(*args)
return queryset.filter(self.build_query(**kwargs)) 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 ########## ########## Pagination Mixins ##########
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment