Fix risk score pipeline

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

Author: TG1999

vulnerabilities/models.py

@@ -2930,6 +2930,12 @@ def latest_for_avid(self, avid: str):
 
     def latest_per_avid(self):
         return self.filter(is_latest=True)
+    
+    def latest_for_avid_completely_imported_advisories(self, avid):
+        return self.get(avid=avid, is_latest=True, _all_impacts_unfurled=True)
+
+    def latest_completely_imported_advisories_per_avid(self):
+        return self.filter(is_latest=True, _all_impacts_unfurled=True)
 
     def latest_for_avids(self, avids):
         return self.filter(avid__in=avids).latest_per_avid()
@@ -2941,7 +2947,7 @@ def latest_affecting_advisories_for_purl(self, purl):
             advisory___all_impacts_unfurled=True,
         ).values_list("advisory_id", flat=True)
 
-        return self.filter(id__in=Subquery(adv_ids)).latest_per_avid()
+        return self.filter(id__in=Subquery(adv_ids))
 
     def latest_affecting_advisories_for_purls(self, purls):
         adv_ids = ImpactedPackage.objects.filter(
@@ -2952,7 +2958,7 @@ def latest_affecting_advisories_for_purls(self, purls):
             "advisory_id",
             flat=True,
         )
-        return self.filter(id__in=Subquery(adv_ids)).latest_per_avid()
+        return self.filter(id__in=Subquery(adv_ids))
 
     def latest_affecting_advisories_for_packages(self, packages):
         adv_ids = ImpactedPackage.objects.filter(
@@ -2963,7 +2969,7 @@ def latest_affecting_advisories_for_packages(self, packages):
             "advisory_id",
             flat=True,
         )
-        return self.filter(id__in=Subquery(adv_ids)).latest_per_avid()
+        return self.filter(id__in=Subquery(adv_ids))
 
     def latest_fixed_by_advisories_for_purl(self, purl):
         adv_ids = ImpactedPackage.objects.filter(
@@ -2974,7 +2980,7 @@ def latest_fixed_by_advisories_for_purl(self, purl):
             "advisory_id",
             flat=True,
         )
-        return self.filter(id__in=Subquery(adv_ids)).latest_per_avid()
+        return self.filter(id__in=Subquery(adv_ids))
 
     def latest_fixed_by_advisories_for_purls(self, purls):
         adv_ids = ImpactedPackage.objects.filter(
@@ -2986,7 +2992,7 @@ def latest_fixed_by_advisories_for_purls(self, purls):
             flat=True,
         )
 
-        return self.filter(id__in=Subquery(adv_ids)).latest_per_avid()
+        return self.filter(id__in=Subquery(adv_ids))
 
     def latest_advisories_for_purls(self, purls):
         adv_ids = (
@@ -3012,7 +3018,7 @@ def latest_advisories_for_purls(self, purls):
         )
 
         qs = self.filter(id__in=Subquery(adv_ids))
-        return qs.latest_per_avid()
+        return qs
 
     def latest_advisories_for_purl(self, purl):
         adv_ids = (
@@ -3038,7 +3044,7 @@ def latest_advisories_for_purl(self, purl):
         )
 
         qs = self.filter(id__in=Subquery(adv_ids))
-        return qs.latest_per_avid()
+        return qs
 
     def todo_excluded(self):
         """Exclude advisory ineligible for ToDo computation."""

vulnerabilities/pipelines/v2_improvers/compute_package_risk.py

@@ -132,7 +132,7 @@ def compute_and_store_vulnerability_risk_score(self):
 
     def compute_and_store_package_risk_score(self):
 
-        latest_advisories = AdvisoryV2.objects.latest_per_avid()
+        latest_advisories = AdvisoryV2.objects.latest_completely_imported_advisories_per_avid()
 
         qs = (
             PackageV2.objects.filter(