Skip to content

Commit 9f4d4d0

Browse files
committed
Update CodeFix pipeline to process impacted packages
Signed-off-by: Keshav Priyadarshi <git@keshav.space>
1 parent 79a7e7f commit 9f4d4d0

File tree

1 file changed

+11
-13
lines changed

1 file changed

+11
-13
lines changed

vulnerabilities/pipelines/v2_improvers/collect_commits.py

Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ def steps(cls):
3737

3838
def collect_and_store_fix_commits(self):
3939
affected_advisories = (
40-
AdvisoryV2.objects.filter(affecting_packages__isnull=False)
41-
.prefetch_related("affecting_packages")
40+
AdvisoryV2.objects.filter(impacted_packages__affecting_packages__isnull=False)
41+
.prefetch_related("impacted_packages__affecting_packages", "references")
4242
.distinct()
4343
)
4444

@@ -66,19 +66,17 @@ def collect_and_store_fix_commits(self):
6666
)
6767
continue
6868
# check if vcs_url has commit
69-
for package in adv.affecting_packages.all():
70-
code_fix, created = CodeFixV2.objects.get_or_create(
71-
commits=[vcs_url],
72-
advisory=adv,
73-
affected_package=package,
74-
)
75-
76-
if created:
77-
created_fix_count += 1
78-
self.log(
79-
f"Created CodeFix entry for reference: {reference.url} with VCS URL {vcs_url}"
69+
for impact in adv.impacted_packages.all():
70+
for package in impact.affecting_packages.all():
71+
code_fix, created = CodeFixV2.objects.get_or_create(
72+
commits=[vcs_url],
73+
advisory=adv,
74+
affected_package=package,
8075
)
8176

77+
if created:
78+
created_fix_count += 1
79+
8280
self.log(f"Successfully created {created_fix_count:,d} CodeFix entries.")
8381

8482

0 commit comments

Comments
 (0)