Add custom pagination to return total number of pages.

savepointradio/api/pagination.py

@@ -0,0 +1,20 @@
+from rest_framework import pagination
+from rest_framework.response import Response
+
+
+class TotalPagesPagination(pagination.PageNumberPagination):
+    '''
+    Custom pagination class to add the total numer of pages for the results.
+
+    (Thanks to https://stackoverflow.com/questions/40985248/)
+    '''
+    def get_paginated_response(self, data):
+        return Response({
+            'links': {
+                'next': self.get_next_link(),
+                'previous': self.get_previous_link()
+            },
+            'count': self.page.paginator.count,
+            'total_pages': self.page.paginator.num_pages,
+            'results': data
+        })

savepointradio/savepointradio/settings/base.py

@@ -102,7 +102,7 @@ REST_FRAMEWORK = {
         'rest_framework.authentication.SessionAuthentication',
         'rest_framework.authentication.TokenAuthentication',
     ),
-    'DEFAULT_PAGINATION_CLASS': 'rest_framework.pagination.PageNumberPagination',
+    'DEFAULT_PAGINATION_CLASS': 'api.pagination.TotalPagesPagination',
     'DEFAULT_PERMISSION_CLASSES': (
         'rest_framework.permissions.IsAuthenticated',
     ),