Add django admin functionality for searching and filtering objects

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

Author: sbs2001

vulnerabilities/admin.py

@@ -34,22 +34,24 @@
 
 @admin.register(Vulnerability)
 class VulnerabilityAdmin(admin.ModelAdmin):
-    pass
+    search_fields = ["cve_id"]
 
 
 @admin.register(VulnerabilityReference)
 class VulnerabilityReferenceAdmin(admin.ModelAdmin):
-    pass
+    search_fields = ["vulnerability__cve_id", "reference_id", "url"]
 
 
 @admin.register(Package)
 class PackageAdmin(admin.ModelAdmin):
-    pass
+    list_filter = ("type", "namespace")
+    search_fields = ["name"]
 
 
 @admin.register(PackageRelatedVulnerability)
 class PackageRelatedVulnerabilityAdmin(admin.ModelAdmin):
-    pass
+    list_filter = ("is_vulnerable", "package__type", "package__namespace")
+    search_fields = ["vulnerability__cve_id", "package__name"]
 
 
 @admin.register(Importer)

vulnerabilities/models.py

@@ -144,6 +144,9 @@ class PackageRelatedVulnerability(models.Model):
     vulnerability = models.ForeignKey(Vulnerability, on_delete=models.CASCADE)
     is_vulnerable = models.BooleanField()
 
+    def __str__(self):
+        return self.package.package_url + " " + self.vulnerability.cve_id
+
     class Meta:
         unique_together = ("package", "vulnerability")
         verbose_name_plural = "PackageRelatedVulnerabilities"