diff --git a/passbook/api/pagination.py b/passbook/api/pagination.py new file mode 100644 index 000000000..9c9ff2759 --- /dev/null +++ b/passbook/api/pagination.py @@ -0,0 +1,29 @@ +"""Pagination which includes total pages and current page""" +from rest_framework import pagination +from rest_framework.response import Response + + +class Pagination(pagination.PageNumberPagination): + """Pagination which includes total pages and current page""" + + def get_paginated_response(self, data): + previous_page_number = 0 + if self.page.has_previous(): + previous_page_number = self.page.previous_page_number() + next_page_number = 0 + if self.page.has_next(): + next_page_number = self.page.next_page_number() + return Response( + { + "pagination": { + "next": next_page_number, + "previous": previous_page_number, + "count": self.page.paginator.count, + "current": self.page.number, + "total_pages": self.page.paginator.num_pages, + "start_index": self.page.start_index(), + "end_index": self.page.end_index(), + }, + "results": data, + } + ) diff --git a/passbook/policies/api.py b/passbook/policies/api.py index f48540bd4..636ff9530 100644 --- a/passbook/policies/api.py +++ b/passbook/policies/api.py @@ -97,3 +97,4 @@ class PolicyBindingViewSet(ModelViewSet): queryset = PolicyBinding.objects.all() serializer_class = PolicyBindingSerializer filterset_fields = ["policy", "target", "enabled", "order", "timeout"] + search_fields = ["policy__name"] diff --git a/passbook/root/settings.py b/passbook/root/settings.py index 3039f371e..81effe556 100644 --- a/passbook/root/settings.py +++ b/passbook/root/settings.py @@ -140,7 +140,7 @@ SWAGGER_SETTINGS = { } REST_FRAMEWORK = { - "DEFAULT_PAGINATION_CLASS": "rest_framework.pagination.LimitOffsetPagination", + "DEFAULT_PAGINATION_CLASS": "passbook.api.pagination.Pagination", "PAGE_SIZE": 100, "DEFAULT_FILTER_BACKENDS": [ "rest_framework_guardian.filters.ObjectPermissionsFilter",