From 4b89ed8664835f74e108643a1cb2821b9a27171f Mon Sep 17 00:00:00 2001
From: Lukas Koenen <lukas.koenen@h-da.de>
Date: Fri, 8 Sep 2023 09:52:03 +0200
Subject: [PATCH] refactor: report info
---
contrib/report_builders/html_report_builder.py | 17 ++++++++---------
1 file changed, 8 insertions(+), 9 deletions(-)
diff --git a/contrib/report_builders/html_report_builder.py b/contrib/report_builders/html_report_builder.py
index f3d2f38..b6fee52 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)
--
GitLab