Merge pull request #206 from Mab879/fix_174

Add CAT-X to severity

Author: Mab879

stigaview_static/html_output.py

@@ -12,9 +12,16 @@
 from stigaview_static.utils import get_config, get_git_revision_short_hash
 
 
+def _severity_to_cat(severity: str) -> str:
+    """Convert severity level to DISA CAT level."""
+    mapping = {"high": "CAT I", "medium": "CAT II", "low": "CAT III"}
+    return mapping.get(severity.lower(), severity)
+
+
 def render_template(template: str, out_path: str, **kwargs):
     file_loader = FileSystemLoader("templates")
     env = Environment(loader=file_loader)
+    env.filters["severity_to_cat"] = _severity_to_cat
     template = env.get_template(template)
     config = get_config()
     context = kwargs | config

templates/control.html

@@ -6,7 +6,7 @@ <h2>{{ control.title }}</h2>
   <p>
   <b>STIG ID:</b> {{ control.disa_stig_id }} &nbsp;|&nbsp;
   <b>SRG: </b><a href="{{ control.srg.url }}">{{ control.srg.srg_id }}</a>&nbsp;|&nbsp;
-  <b>Severity: </b>{{ control.severity }}&nbsp;|&nbsp;
+  <b>Severity: </b>{{ control.severity }} ({{ control.severity | severity_to_cat }}) &nbsp;|&nbsp;
   <b>CCI: </b>{{ ",".join(control.cci) }}&nbsp;|&nbsp;
   <b>Vulnerability Id:</b> V-{{ control.vulnerability_id }}
   </p>

templates/one_page_stig.html

@@ -11,7 +11,7 @@ <h2 id="{{ control.disa_stig_id }}">{{ control.title }}</h2>
     <b><a href="#{{ control.disa_stig_id }}">STIG ID</a>:</b>
       <a href="/products/{{ product.short_name }}/{{ stig.short_version.lower() }}/{{ control.disa_stig_id }}">{{ control.disa_stig_id }}</a>&nbsp;|&nbsp;
       <b>SRG: </b><a href="{{ control.srg.url }}">{{ control.srg.srg_id }}</a>&nbsp;|&nbsp;
-      <b>Severity: </b>{{ control.severity }} &nbsp;|&nbsp;
+      <b>Severity: </b>{{ control.severity }} ({{ control.severity | severity_to_cat }}) &nbsp;|&nbsp;
       <b>CCI: </b>{{ ",".join(control.cci) }}&nbsp;|&nbsp;
       <b>Vulnerability Id:</b> V-{{ control.vulnerability_id }}
     </p>