From 8cba83e7e246a10d8943dd689bf90419d40ea9e2 Mon Sep 17 00:00:00 2001
From: Ahmed Abdelsalam <ahmed.abdelsalam@greenbone.net>
Date: Fri, 3 Jan 2025 17:49:35 +0100
Subject: [PATCH] Change: Update the severity ratings to match CVSS 3.1

---
 pheme/templatetags/charts/__init__.py    | 3 ++-
 pheme/transformation/scanreport/gvmd.py  | 2 +-
 pheme/transformation/scanreport/model.py | 1 +
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/pheme/templatetags/charts/__init__.py b/pheme/templatetags/charts/__init__.py
index f07e1574..49809841 100644
--- a/pheme/templatetags/charts/__init__.py
+++ b/pheme/templatetags/charts/__init__.py
@@ -21,7 +21,8 @@
 from django import template
 
 _severity_class_colors = {
-    "High": "#d4003e",
+    "Critical": "#bf0000",
+    "High": "#f51414",
     "Medium": "#fcb900",
     "Low": "#7db4d0",
 }
diff --git a/pheme/transformation/scanreport/gvmd.py b/pheme/transformation/scanreport/gvmd.py
index 1ba64e9a..c51ac162 100644
--- a/pheme/transformation/scanreport/gvmd.py
+++ b/pheme/transformation/scanreport/gvmd.py
@@ -37,7 +37,7 @@
 
 logger = logging.getLogger(__name__)
 
-__threats = ["High", "Medium", "Low"]
+__threats = ["Critical", "High", "Medium", "Low"]
 __threat_index_lookup = {v: i for i, v in enumerate(__threats)}
 
 
diff --git a/pheme/transformation/scanreport/model.py b/pheme/transformation/scanreport/model.py
index cf6bac86..84381470 100644
--- a/pheme/transformation/scanreport/model.py
+++ b/pheme/transformation/scanreport/model.py
@@ -105,6 +105,7 @@ def describe():
                 host="str; ip address of host",
                 hostname="str; main hostname of the host",
                 threats={
+                    "critical": "int; amount of critical nvts in host",
                     "high": "int; amount of high nvts in host",
                     "medium": "int; amount of medium nvts in host",
                     "low": "int; amount of low nvts in host",