✨ Add pagination to vulnerability search

Signed-off-by: Shivam Sandbhor <shivam.sandbhor@gmail.com>

Author: sbs2001

vulnerabilities/templates/vulnerabilities.html

@@ -1,5 +1,4 @@
 {% extends 'base.html' %}
-
 {% block title %}
 <h1 class="title">
 Explore Vulnerabilities
@@ -23,15 +22,16 @@ <h1 class="title">
 </div>
 
 {% if vulnerabilities %}
-    <section class="section is-medium  has-text-centered">
-      <div class="container ">
-        <h1 class="title" id="results">Results of your query</h1>
+    <section class="section is-medium has-text-centered">
+      <div class="container">
+        <h1 class="title" id="results">Results of your query "{{searched_for}}"</h1>
         <h3 class="subtitle" id="results">Total {{vulnerabilities|length}} results</h3>
       </div>
     </section>
 
     <section class="section has-background-dark">
-        <div class="content is-medium">
+      
+        <div class="content is-large">
           <table class="table is-bordered is-striped is-narrow is-hoverable is-fullwidth">
           <tr>
               <th>Vulnerabilities</th>
@@ -47,6 +47,17 @@ <h3 class="subtitle" id="results">Total {{vulnerabilities|length}} results</h3>
           {% endfor %}
           </table>
         </div>
+
+        <div class="is-field is-horizontal">
+          {% if vulnerabilities.has_previous %}
+            <a class="button" href="?vuln_id={{ searched_for }}&page={{ vulnerabilities.previous_page_number }}">Previous Page</a>
+          {% endif %}
+
+          {% if vulnerabilities.has_next %}
+            <a class="button" href="?vuln_id={{ searched_for }}&page={{ vulnerabilities.next_page_number}}">Next Page</a>
+          {% endif %}
+        </div>
+
     </section>
 
 {% endif %}

vulnerabilities/views.py

@@ -20,6 +20,7 @@
 #  VulnerableCode is a free software code scanning tool from nexB Inc. and others.
 #  Visit https://github.com/nexB/vulnerablecode/ for support and download.
 
+from django.core.paginator import Paginator
 from django.shortcuts import render
 from django.urls import reverse
 from django.views.generic.list import ListView
@@ -66,19 +67,21 @@ class VulnerabilitySearchView(View):
 
     def get(self, request):
         context = {"form": forms.VulnerabilitySearchForm()}
-
+        print(request.GET)
         if request.GET:
             vulnerabilities = self.request_to_queryset(request)
+            pages = Paginator(vulnerabilities, 50)
+            vulnerabilities = pages.get_page(int(self.request.GET.get('page', 1)))
             context["vulnerabilities"] = vulnerabilities
+            context["searched_for"] = request.GET.get("vuln_id")
 
         return render(request, self.template_name, context)
 
     @staticmethod
     def request_to_queryset(request):
-
         if request.GET["vuln_id"]:
             vuln_id = request.GET["vuln_id"]
-            return models.Vulnerability.objects.filter(cve_id__contains=vuln_id).select_related()
+            return models.Vulnerability.objects.filter(cve_id__contains=vuln_id)
 
 
 class PackageUpdate(UpdateView):
@@ -183,7 +186,6 @@ def get_form(self, *args, **kwargs):
         return form
 
     def form_valid(self, form):
-
         is_vulnerable = "impacted" in self.request.headers["Referer"]
         package = models.Package.objects.get(id=self.kwargs["pid"])
         form.instance.package = package