Skip to content

Commit 0bb9b62

Browse files
author
Jakob Maier
committed
only actual remediations count as such
1 parent 0da7b46 commit 0bb9b62

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

examples/vuln_batch_remediation.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -83,15 +83,19 @@
8383

8484

8585
def load_remediation_input(remediation_file):
86-
with open(remediation_file, mode='r') as infile:
86+
with open(remediation_file, mode='r', encoding="utf-8") as infile:
8787
reader = csv.reader(infile)
8888
return {rows[0]:[rows[1],rows[2]] for rows in reader}
8989

9090
def remediation_is_valid(vuln, remediation_data):
9191
vulnerability_name = vuln['vulnerabilityWithRemediation']['vulnerabilityName']
92-
# remediation_status = vuln['vulnerabilityWithRemediation']['remediationStatus']
93-
# remediation_comment = vuln['vulnerabilityWithRemediation'].get('remediationComment','')
92+
remediation_status = vuln['vulnerabilityWithRemediation']['remediationStatus']
93+
remediation_comment = vuln['vulnerabilityWithRemediation'].get('remediationComment','')
94+
9495
if vulnerability_name in remediation_data.keys():
96+
remediation = remediation_data[vulnerability_name]
97+
if (remediation_status == remediation[0] and remediation_comment == remediation[1]):
98+
return None
9599
return remediation_data[vulnerability_name]
96100
else:
97101
return None

0 commit comments

Comments
 (0)