Resuable Pagination Code - SCIO+VCIO

Signed-off-by: Rishi Garg <rishigarg2503@gmail.com>

Author: Rishi-garg03

vulnerabilities/pagination_mixin.py

@@ -1,4 +1,8 @@
 class PaginatedListViewMixin:
+    """
+    A mixin that adds pagination functionality to ListView-based views.
+    """
+
     paginate_by = 20
     max_page_size = 100
 
@@ -9,6 +13,9 @@ class PaginatedListViewMixin:
     ]
 
     def get_paginate_by(self, queryset=None):
+        """
+        Get the number of items to paginate by from the request.
+        """
         try:
             page_size = int(self.request.GET.get("page_size", self.paginate_by))
             if page_size <= 0:
@@ -17,42 +24,64 @@ def get_paginate_by(self, queryset=None):
         except (ValueError, TypeError):
             return self.paginate_by
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
+    def get_pagination_context(self, paginator, page_obj):
+        """
+        Generate pagination-related context data, preserving filters.
+        """
+        if not paginator or not page_obj:
+            return {}
 
         current_page_size = self.get_paginate_by()
-        total_count = context["paginator"].count
-
-        context.update(
-            {
-                "current_page_size": current_page_size,
-                "page_size_choices": self.PAGE_SIZE_CHOICES,
-                "total_count": total_count,
-                "page_range": self._get_page_range(
-                    context["paginator"], context["page_obj"].number
-                ),
-                "search": self.request.GET.get("search", ""),
-            }
-        )
+        total_count = paginator.count
 
-        return context
+        query_params = self.request.GET.copy()
+        query_params.pop("page", None)
+
+        base_query_string = query_params.urlencode()
+        base_url = f"?{base_query_string}" if base_query_string else "?"
 
-    def _get_page_range(self, paginator, current_page, window=2):
+        pages = []
         if paginator.num_pages <= 5:
-            return range(1, paginator.num_pages + 1)
+            pages = [str(i) for i in range(1, paginator.num_pages + 1)]
+        else:
+            pages.append("1")
 
-        pages = [1]
-        if current_page > 3:
-            pages.append("...")
+            if page_obj.number > 3:
+                pages.append("...")
 
-        start_page = max(2, current_page - window)
-        end_page = min(paginator.num_pages - 1, current_page + window)
-        pages.extend(range(start_page, end_page + 1))
+            start_page = max(2, page_obj.number - 2)
+            end_page = min(paginator.num_pages - 1, page_obj.number + 2)
+            pages.extend(str(i) for i in range(start_page, end_page + 1))
 
-        if current_page < paginator.num_pages - 2:
-            pages.append("...")
+            if page_obj.number < paginator.num_pages - 2:
+                pages.append("...")
 
-        if paginator.num_pages not in pages:
-            pages.append(paginator.num_pages)
+            if str(paginator.num_pages) not in pages:
+                pages.append(str(paginator.num_pages))
 
-        return pages
+        return {
+            "current_page_size": current_page_size,
+            "page_size_choices": self.PAGE_SIZE_CHOICES,
+            "total_count": total_count,
+            "page_range": pages,
+            "search": self.request.GET.get("search", ""),
+            "base_url": base_url,
+            "previous_page_url": f"{base_url}&page={page_obj.previous_page_number}"
+            if page_obj.has_previous()
+            else None,
+            "next_page_url": f"{base_url}&page={page_obj.next_page_number}"
+            if page_obj.has_next()
+            else None,
+        }
+
+    def get_context_data(self, **kwargs):
+        """
+        Add pagination context to the existing context data.
+        """
+        context = super().get_context_data(**kwargs)
+        paginator = context.get("paginator")
+        page_obj = context.get("page_obj")
+        pagination_context = self.get_pagination_context(paginator, page_obj)
+        context.update(pagination_context)
+
+        return context

vulnerabilities/templates/includes/pagination.html

@@ -12,30 +12,52 @@
             </select>
         </div>
     </div>
+<<<<<<< Updated upstream
 
     {% if page_obj.paginator.num_pages > 1 %}
     <nav class="pagination is-centered" role="navigation" aria-label="pagination">
         {% if page_obj.has_previous %}
             <a href="?page={{ page_obj.previous_page_number }}&search={{ search|urlencode }}&page_size={{ current_page_size }}" 
+=======
+  
+    {% if page_obj.paginator.num_pages > 1 %}
+    <nav class="pagination is-centered" role="navigation" aria-label="pagination">
+        {% if page_obj.has_previous %}
+            <a href="{{ base_url }}&page={{ page_obj.previous_page_number }}" 
+>>>>>>> Stashed changes
                class="pagination-previous">Previous</a>
         {% else %}
             <button class="pagination-previous" disabled>Previous</button>
         {% endif %}
+<<<<<<< Updated upstream
 
         {% if page_obj.has_next %}
             <a href="?page={{ page_obj.next_page_number }}&search={{ search|urlencode }}&page_size={{ current_page_size }}" 
+=======
+  
+        {% if page_obj.has_next %}
+            <a href="{{ base_url }}&page={{ page_obj.next_page_number }}" 
+>>>>>>> Stashed changes
                class="pagination-next">Next</a>
         {% else %}
             <button class="pagination-next" disabled>Next</button>
         {% endif %}
+<<<<<<< Updated upstream
 
+=======
+  
+>>>>>>> Stashed changes
         <ul class="pagination-list">
             {% for page_num in page_range %}
                 {% if page_num == '...' %}
                     <li><span class="pagination-ellipsis">&hellip;</span></li>
                 {% else %}
                     <li>
+<<<<<<< Updated upstream
                         <a href="?page={{ page_num }}&search={{ search|urlencode }}&page_size={{ current_page_size }}" 
+=======
+                        <a href="{{ base_url }}&page={{ page_num }}" 
+>>>>>>> Stashed changes
                            class="pagination-link {% if page_num == page_obj.number %}is-current{% endif %}"
                            aria-label="Go to page {{ page_num }}"
                            {% if page_num == page_obj.number %}aria-current="page"{% endif %}>
@@ -47,6 +69,7 @@
         </ul>
     </nav>
     {% endif %}
+<<<<<<< Updated upstream
 </div>
 
 <style>
@@ -75,3 +98,26 @@
     window.location.href = newUrl;
 }
 </script>
+=======
+  </div>
+  
+  <style>
+  .select.is-disabled {
+    opacity: 0.7;
+    cursor: not-allowed;
+  }
+  .select.is-disabled select {
+    cursor: not-allowed;
+  }
+  </style>
+  
+  <script>
+  function handlePageSizeChange(value) {
+    const url = new URL(window.location.href);
+    const params = new URLSearchParams(url.search);
+    params.set('page_size', value);
+    params.delete('page'); 
+    window.location.href = `${url.pathname}?${params.toString()}`;
+  }
+  </script>
+>>>>>>> Stashed changes

vulnerabilities/views.py

@@ -65,10 +65,13 @@ def get_purl_version_class(purl: models.Package):
 
 
 class PackageSearch(PaginatedListViewMixin, ListView):
+<<<<<<< Updated upstream
     """
     View for searching and displaying packages with pagination.
     """
 
+=======
+>>>>>>> Stashed changes
     model = models.Package
     template_name = "packages.html"
     ordering = ["type", "namespace", "name", "version"]
@@ -89,10 +92,13 @@ def get_queryset(self, query=None):
 
 
 class VulnerabilitySearch(PaginatedListViewMixin, ListView):
+<<<<<<< Updated upstream
     """
     View for searching and displaying vulnerabilities with pagination.
     """
 
+=======
+>>>>>>> Stashed changes
     model = models.Vulnerability
     template_name = "vulnerabilities.html"
     ordering = ["vulnerability_id"]