refactor: report info

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)