Skip to content

Commit 88b9a87

Browse files
authored
include vuln_id_from_tool in group_by (#12744)
1 parent 4255895 commit 88b9a87

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

dojo/finding/helper.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -255,6 +255,9 @@ def get_group_by_group_name(finding, finding_group_by_option):
255255
group_name = f"Filepath {finding.file_path}"
256256
elif finding_group_by_option == "finding_title":
257257
group_name = finding.title
258+
elif finding_group_by_option == "vuln_id_from_tool":
259+
if finding.vuln_id_from_tool:
260+
group_name = f"Vulnerability ID {finding.vuln_id_from_tool}" if finding.vuln_id_from_tool else "None"
258261
else:
259262
msg = f"Invalid group_by option {finding_group_by_option}"
260263
raise ValueError(msg)

dojo/models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3449,7 +3449,8 @@ class Finding_Group(TimeStampedModel):
34493449
GROUP_BY_OPTIONS = [("component_name", "Component Name"),
34503450
("component_name+component_version", "Component Name + Version"),
34513451
("file_path", "File path"),
3452-
("finding_title", "Finding Title")]
3452+
("finding_title", "Finding Title"),
3453+
("vuln_id_from_tool", "Vulnerability ID from Tool")]
34533454

34543455
name = models.CharField(max_length=255, blank=False, null=False)
34553456
test = models.ForeignKey(Test, on_delete=models.CASCADE)

0 commit comments

Comments
 (0)