Update impact_package model to include fixed and affecting commits

Signed-off-by: ziad hany <ziadhany2016@gmail.com>

Author: ziadhany

vulnerabilities/models.py

@@ -1960,19 +1960,6 @@ class CodeFixV2(CodeChangeV2):
         help_text="The affected package version to which this code fix applies.",
     )
 
-    affected_package = models.ForeignKey(
-        "PackageV2", on_delete=models.CASCADE, related_name="code_fix_v2_affected"
-    )
-
-    fixed_package = models.ForeignKey(
-        "PackageV2",
-        null=True,
-        blank=True,
-        on_delete=models.SET_NULL,
-        related_name="code_fix_v2_fixed",
-        help_text="The fixing package version with this code fix",
-    )
-
 
 class PipelineRun(models.Model):
     """The Database representation of a pipeline execution."""
@@ -2662,13 +2649,15 @@ class AdvisoryReference(models.Model):
 
     ADVISORY = "advisory"
     EXPLOIT = "exploit"
+    COMMIT = "commit"
     MAILING_LIST = "mailing_list"
     BUG = "bug"
     OTHER = "other"
 
     REFERENCE_TYPES = [
         (ADVISORY, "Advisory"),
         (EXPLOIT, "Exploit"),
+        (COMMIT, "Commit"),
         (MAILING_LIST, "Mailing List"),
         (BUG, "Bug"),
         (OTHER, "Other"),
@@ -2957,6 +2946,18 @@ class ImpactedPackage(models.Model):
         help_text="Packages vulnerable to this impact.",
     )
 
+    fixed_by_commits = models.ManyToManyField(
+        "CodeFixV2",
+        related_name="fixed_commits_impacts",
+        help_text="Commits that fix the vulnerability in this impact.",
+    )
+
+    affected_by_commits = models.ManyToManyField(
+        "CodeFixV2",
+        related_name="affected_commits_impacts",
+        help_text="Commits that are affected by this impact.",
+    )
+
     created_at = models.DateTimeField(
         auto_now_add=True,
         db_index=True,

vulnerabilities/pipelines/v2_improvers/collect_commits.py

@@ -64,15 +64,14 @@ def collect_and_store_fix_commits(self):
                     continue
                 # check if vcs_url has commit
                 for impact in adv.impacted_packages.all():
-                    for package in impact.affecting_packages.all():
-                        code_fix, created = CodeFixV2.objects.get_or_create(
-                            commits=[vcs_url],
-                            advisory=adv,
-                            affected_package=package,
-                        )
-
-                        if created:
-                            created_fix_count += 1
+                    code_fix, created = CodeFixV2.objects.get_or_create(
+                        commits=[vcs_url],
+                        advisory=adv,
+                    )
+                    impact.fixed_by_commits.add(code_fix)
+
+                    if created:
+                        created_fix_count += 1
 
         self.log(f"Successfully created {created_fix_count:,d} CodeFix entries.")
 

vulnerabilities/tests/pipelines/v2_improvers/test_collect_commits_v2.py

@@ -62,19 +62,29 @@ def test_is_vcs_url_already_processed_true():
         unique_content_id="11111",
         date_collected=datetime.now(),
     )
+
+    commit1 = CodeFixV2.objects.create(advisory=advisory)
+    commit1.commits = ["https://github.com/user/repo/commit/abc1"]
+    commit1.save()
+
+    commit2 = CodeFixV2.objects.create(advisory=advisory)
+    commit2.commits = ["https://github.com/user/repo/commit/abc2"]
+    commit2.save()
+
     package = PackageV2.objects.create(
         type="bar",
         name="foo",
         version="1.0",
     )
-    impact = ImpactedPackage.objects.create(advisory=advisory)
-    impact.affecting_packages.add(package)
-    CodeFixV2.objects.create(
-        commits=["https://github.com/user/repo/commit/abc123"],
+
+    impact = ImpactedPackage.objects.create(
         advisory=advisory,
-        affected_package=package,
     )
-    assert is_vcs_url_already_processed("https://github.com/user/repo/commit/abc123") is True
+    impact.affecting_packages.add(package)
+    impact.fixed_by_commits.add(commit1)
+    impact.affected_by_commits.add(commit2)
+
+    assert is_vcs_url_already_processed("https://github.com/user/repo/commit/abc1") is True
 
 
 @pytest.mark.django_db
@@ -93,7 +103,7 @@ def test_collect_fix_commits_pipeline_creates_entry():
         version="1.0",
     )
     reference = AdvisoryReference.objects.create(
-        url="https://github.com/test/testpkg/commit/abc123"
+        reference_type="commit", url="https://github.com/test/testpkg/commit/abc123"
     )
     impact = ImpactedPackage.objects.create(advisory=advisory)
     impact.affecting_packages.add(package)
@@ -107,7 +117,7 @@ def test_collect_fix_commits_pipeline_creates_entry():
     fix = codefixes.first()
     assert "abc123" in fix.commits[0]
     assert fix.advisory == advisory
-    assert fix.affected_package == package
+    assert list(fix.fixed_commits_impacts.all()) == [impact]
 
 
 @pytest.mark.django_db

vulnerabilities/tests/test_api_v2.py

@@ -798,13 +798,8 @@ def setUp(self):
             date_collected="2025-07-01T00:00:00Z",
         )
 
-        self.affected_package = PackageV2.objects.from_purl(purl="pkg:pypi/affected_package@1.0.0")
-        self.fixed_package = PackageV2.objects.from_purl(purl="pkg:pypi/fixed_package@1.0.1")
-
         self.codefix = CodeFixV2.objects.create(
             advisory=self.advisory,
-            affected_package=self.affected_package,
-            fixed_package=self.fixed_package,
             notes="Security patch",
             is_reviewed=True,
         )
@@ -816,14 +811,14 @@ def setUp(self):
         self.url = reverse("advisory-codefix-list")
 
     def test_list_all_codefixes(self):
-        with self.assertNumQueries(10):
+        with self.assertNumQueries(8):
             response = self.client.get(self.url)
         assert response.status_code == status.HTTP_200_OK
         assert response.data["count"] == 1
         assert response.data["results"][0]["affected_advisory_id"] == self.advisory.avid
 
     def test_filter_codefix_by_advisory_id_success(self):
-        with self.assertNumQueries(10):
+        with self.assertNumQueries(8):
             response = self.client.get(self.url, {"advisory_id": self.advisory.avid})
         assert response.status_code == status.HTTP_200_OK
         assert response.data["count"] == 1