diff --git a/wagtail/api/v2/pagination.py b/wagtail/api/v2/pagination.py index 3bfdac7e1..089d99547 100644 --- a/wagtail/api/v2/pagination.py +++ b/wagtail/api/v2/pagination.py @@ -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): diff --git a/wagtail/contrib/wagtailapi/pagination.py b/wagtail/contrib/wagtailapi/pagination.py index b8c72b4dc..4182a3f98 100644 --- a/wagtail/contrib/wagtailapi/pagination.py +++ b/wagtail/contrib/wagtailapi/pagination.py @@ -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):