Add queryset

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/migrations/0107_remove_advisoryv2_date_imported.py

@@ -0,0 +1,17 @@
+# Generated by Django 4.2.25 on 2025-12-29 07:48
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("vulnerabilities", "0106_alter_advisoryreference_url_and_more"),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name="advisoryv2",
+            name="date_imported",
+        ),
+    ]

vulnerabilities/models.py

@@ -42,7 +42,9 @@
 from django.db import transaction
 from django.db.models import Count
 from django.db.models import Exists
+from django.db.models import F
 from django.db.models import OuterRef
+from django.db.models import Subquery
 from django.db.models import Prefetch
 from django.db.models import Q
 from django.db.models.functions import Length
@@ -1368,6 +1370,17 @@ def search(query):
             | Q(references__url__icontains=query)
         ).distinct()
 
+class AdvisoryV2QuerySet(BaseQuerySet):
+    def latest_for_avid(self, avid: str):
+        return (
+            self.filter(avid=avid)
+            .order_by(
+                F("date_collected").desc(nulls_last=True),
+                "-id",
+            )
+            .first()
+        )
+
 
 # FIXME: Remove when migration from Vulnerability to Advisory is completed
 class Advisory(models.Model):
@@ -2916,9 +2929,6 @@ class AdvisoryV2(models.Model):
         blank=True, null=True, help_text="UTC Date of publication of the advisory"
     )
     date_collected = models.DateTimeField(help_text="UTC Date on which the advisory was collected")
-    date_imported = models.DateTimeField(
-        blank=True, null=True, help_text="UTC Date on which the advisory was imported"
-    )
 
     original_advisory_text = models.TextField(
         blank=True,
@@ -2959,11 +2969,15 @@ def risk_score(self):
             risk_score = min(float(self.exploitability * self.weighted_severity), 10.0)
             return round(risk_score, 1)
 
-    objects = AdvisoryQuerySet.as_manager()
+    objects = AdvisoryV2QuerySet.as_manager()
 
     class Meta:
         unique_together = ["datasource_id", "advisory_id", "unique_content_id"]
         ordering = ["datasource_id", "advisory_id", "date_published", "unique_content_id"]
+        # indexes = models.Index(
+        #     fields=["avid", "-date_collected", "-id"],
+        #     name="advisory_latest_by_avid_idx",
+        # )
 
     def save(self, *args, **kwargs):
         self.full_clean()
@@ -3004,6 +3018,29 @@ def get_aliases(self):
         """
         return self.aliases.all()
 
+    def latest_for_avid(self, avid: str):
+        return (
+            self.filter(avid=avid)
+            .order_by(
+                F("date_collected").desc(nulls_last=True),
+                "-id",
+            )
+            .first()
+        )
+    
+    def latest_per_avid(self):
+        latest_ids = (
+            AdvisoryV2.objects
+            .filter(avid=OuterRef("avid"))
+            .order_by(
+                F("date_collected").desc(nulls_last=True),
+                "-id",
+            )
+            .values("id")[:1]
+        )
+
+        return self.filter(id=Subquery(latest_ids))
+
     alias = get_aliases
 
 

vulnerabilities/views.py

@@ -23,6 +23,7 @@
 from django.urls import reverse_lazy
 from django.views import View
 from django.views import generic
+from django.db.models import F
 from django.views.generic.detail import DetailView
 from django.views.generic.edit import FormMixin
 from django.views.generic.list import ListView
@@ -360,6 +361,15 @@ class AdvisoryDetails(DetailView):
     slug_url_kwarg = "avid"
     slug_field = "avid"
 
+    def get_object(self, queryset=None):
+        avid = self.kwargs.get(self.slug_url_kwarg)
+        obj = models.AdvisoryV2.objects.latest_for_avid(avid)
+
+        if not obj:
+            raise Http404("Advisory not found")
+
+        return obj
+
     def get_queryset(self):
         return (
             super()