added custom pagination

Author: mostafa6765

django_petra/paginate.py

@@ -1,35 +1,45 @@
-from rest_framework.response import Response
-from rest_framework.pagination import PageNumberPagination
 from collections import OrderedDict
+from math import ceil
+from django.core.paginator import Paginator
+from rest_framework.response import Response
+from urllib.parse import urlparse
+
+def paginate(request, queryset, serializer_class, per_page=10, wrap='data', additional_data=None):
+    # Get per_page from query params if present, otherwise use default
+    per_page = int(request.query_params.get('per_page', per_page))
+    page = int(request.query_params.get('page', 1))
+    total = queryset.count()
+    last_page = ceil(total / per_page)
+    offset = (page - 1) * per_page
+
+    paginated_queryset = queryset[offset:offset + per_page]
+    serializer = serializer_class(paginated_queryset, many=True)
+    data = serializer.data
 
-class LaravelStylePagination(PageNumberPagination):
-    page_size = 10
-    
-    def get_paginated_response(self, data, wrap):
-        data =  (OrderedDict([
-            ('per_page', self.page_size),
-            ('current_page', self.page.number),
-            ('last_page', self.page.paginator.num_pages),
-            ('next_page_url', self.get_next_link() or None),
-            ('prev_page_url', self.get_previous_link() or None),
-            ('total', self.page.paginator.count),
-            ('from', self.page.start_index()),
-            ('to', self.page.end_index()),
-            (wrap, data),
-        ]))
-        return data
+    full_path = str(request.build_absolute_uri())
+    parsed_url = urlparse(full_path)
+    path_without_query = f"{parsed_url.scheme}://{parsed_url.netloc}{parsed_url.path}"
 
-        
-def paginate(request, queryset, serializer_class, per_page=2, wrap='data'):
-    # Create an instance of the custom pagination class
-    paginator = LaravelStylePagination()
-    paginator.page_size = per_page
+    first_page_url = f"{path_without_query}?page=1&per_page={per_page}"
+    last_page_url = f"{path_without_query}?page={last_page}&per_page={per_page}"
+    next_page_url = f"{path_without_query}?page={page + 1}&per_page={per_page}" if page < last_page else None
+    prev_page_url = f"{path_without_query}?page={page - 1}&per_page={per_page}" if page > 1 else None
 
-    # Paginate the queryset
-    page = paginator.paginate_queryset(queryset, request)
+    # Convert all URLs to strings explicitly
+    response_data = OrderedDict([
+        (wrap, data),
+        ('total', total),
+        ('per_page', int(per_page)),  # ensure integer
+        ('current_page', int(page)),   # ensure integer
+        ('last_page', int(last_page)), # ensure integer
+        ('first_page_url', str(first_page_url)),
+        ('last_page_url', str(last_page_url)),
+        ('next_page_url', str(next_page_url) if next_page_url else None),
+        ('prev_page_url', str(prev_page_url) if prev_page_url else None),
+    ])
 
-    # Serialize the paginated data using the provided serializer class
-    serializer = serializer_class(page, many=True)
+    # Merge additional_data if provided
+    if additional_data:
+        response_data['additional_data'] = additional_data
 
-    # Return the paginated data along with pagination metadata
-    return paginator.get_paginated_response(serializer.data, wrap)
+    return response_data

django_petra/paginate_old.py

@@ -0,0 +1,35 @@
+from rest_framework.response import Response
+from rest_framework.pagination import PageNumberPagination
+from collections import OrderedDict
+
+class LaravelStylePagination(PageNumberPagination):
+    page_size = 10
+    
+    def get_paginated_response(self, data, wrap):
+        data =  (OrderedDict([
+            ('per_page', self.page_size),
+            ('current_page', self.page.number),
+            ('last_page', self.page.paginator.num_pages),
+            ('next_page_url', self.get_next_link() or None),
+            ('prev_page_url', self.get_previous_link() or None),
+            ('total', self.page.paginator.count),
+            ('from', self.page.start_index()),
+            ('to', self.page.end_index()),
+            (wrap, data),
+        ]))
+        return data
+
+        
+def paginate(request, queryset, serializer_class, per_page=2, wrap='data'):
+    # Create an instance of the custom pagination class
+    paginator = LaravelStylePagination()
+    paginator.page_size = per_page
+
+    # Paginate the queryset
+    page = paginator.paginate_queryset(queryset, request)
+
+    # Serialize the paginated data using the provided serializer class
+    serializer = serializer_class(page, many=True)
+
+    # Return the paginated data along with pagination metadata
+    return paginator.get_paginated_response(serializer.data, wrap)