LimitOffsetPagination: Changed the check of data type to try-catch in get_count method to isinstance method

This commit is contained in:
RokHyungSon 2021-03-26 02:33:08 +09:00
parent dffa612134
commit 9d980a9b8d

View File

@ -8,6 +8,7 @@ from urllib import parse
from django.core.paginator import InvalidPage from django.core.paginator import InvalidPage
from django.core.paginator import Paginator as DjangoPaginator from django.core.paginator import Paginator as DjangoPaginator
from django.db.models.query import QuerySet
from django.template import loader from django.template import loader
from django.utils.encoding import force_str from django.utils.encoding import force_str
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
@ -522,9 +523,9 @@ class LimitOffsetPagination(BasePagination):
""" """
Determine an object count, supporting either querysets or regular lists. Determine an object count, supporting either querysets or regular lists.
""" """
try: if isinstance(queryset, QuerySet):
return queryset.count() return queryset.count()
except (AttributeError, TypeError): else:
return len(queryset) return len(queryset)
def get_schema_fields(self, view): def get_schema_fields(self, view):