Refactored total_count computation line

This commit is contained in:
Morgan Aubert 2017-04-06 10:40:48 -04:00
parent 9be68178e4
commit 2b09cdaef2
2 changed files with 2 additions and 4 deletions

View file

@ -36,8 +36,7 @@ class WagtailPagination(BasePagination):
stop = offset + limit
self.view = view
self.total_count = (queryset.count() if isinstance(queryset, QuerySet) or
isinstance(queryset, BaseSearchResults) else len(queryset))
self.total_count = queryset.count() if isinstance(queryset, (BaseSearchResults, QuerySet)) else len(queryset)
return queryset[start:stop]
def get_paginated_response(self, data):

View file

@ -36,8 +36,7 @@ class WagtailPagination(BasePagination):
stop = offset + limit
self.view = view
self.total_count = (queryset.count() if isinstance(queryset, QuerySet) or
isinstance(queryset, BaseSearchResults) else len(queryset))
self.total_count = queryset.count() if isinstance(queryset, (BaseSearchResults, QuerySet)) else len(queryset)
return queryset[start:stop]
def get_paginated_response(self, data):