File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 83
83
84
84
85
85
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 :
87
87
reader = csv .reader (infile )
88
88
return {rows [0 ]:[rows [1 ],rows [2 ]] for rows in reader }
89
89
90
90
def remediation_is_valid (vuln , remediation_data ):
91
91
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
+
94
95
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
95
99
return remediation_data [vulnerability_name ]
96
100
else :
97
101
return None
You can’t perform that action at this time.
0 commit comments