diff --git a/contrib/report_builders/html_report_builder.py b/contrib/report_builders/html_report_builder.py index f3d2f389274e8097c94c11fffb7277fdc6080b1d..b6fee5238e02778e8c577f6bc550b1b535b815a8 100644 --- a/contrib/report_builders/html_report_builder.py +++ b/contrib/report_builders/html_report_builder.py @@ -8,19 +8,18 @@ from contrib.descriptions import VulnDescriptionProvider from contrib.report_builders import JsonReportBuilder -class Severity: - def __init__(self, severity): +class ReportInfo: + def __init__(self, severity='Low', is_empty=True): self.level = severity self.num_level = 0 - if severity == 'Low': - self.num_level = 1 + self.empty = is_empty if severity == 'Medium': - self.num_level = 2 + self.num_level = 1 if severity == 'High': - self.num_level = 3 + self.num_level = 2 def toJSON(self): return json.dumps(self, default=lambda o: o.__dict__, @@ -40,15 +39,15 @@ class JinjaHtmlReportBuilder(JsonReportBuilder): def build(self) -> Any: json_data = self._buffer - max_severity = Severity('Empty') + max_severity = ReportInfo() for vv in json_data['vulnerable'].values(): for v in vv['vulnerabilities']: - s = Severity(v['severity_str']) + s = ReportInfo(v['severity_str'], False) if s > max_severity: max_severity = s - with open(f'{json_data["output_path"]}.severity', 'w+') as severity: + with open(f'{json_data["output_path"]}.info', 'w+') as severity: severity.write(max_severity.toJSON()) return self._template.render(data=json_data)