Source code for thaliawebsite.api.pagination

"""Custom API pagination classes."""
from rest_framework.pagination import LimitOffsetPagination



[docs]class APIv2LimitOffsetPagination(LimitOffsetPagination):
    """Pagination class that uses LimitOffsetPagination and sets the default value for the pagination size to None for API v1."""


[docs]    def get_limit(self, request):
        if self.limit_query_param:
            return super().get_limit(request)

        if request.version == "v1":
            return None
        return self.default_limit