From 336149339a176566a7796a8781a326c156988695 Mon Sep 17 00:00:00 2001
From: Bread Genie <63963181+BreadGenie@users.noreply.github.com>
Date: Fri, 14 May 2021 02:02:19 +0530
Subject: [PATCH] Handle UNKNOWN severity in PDF reports (#1152)

Fixes: #1151
---
 cve_bin_tool/output_engine/__init__.py   | 2 ++
 cve_bin_tool/output_engine/pdfbuilder.py | 9 +++++++++
 2 files changed, 11 insertions(+)

diff --git a/cve_bin_tool/output_engine/__init__.py b/cve_bin_tool/output_engine/__init__.py
index 3b65ebd62f..6a15919ca0 100644
--- a/cve_bin_tool/output_engine/__init__.py
+++ b/cve_bin_tool/output_engine/__init__.py
@@ -4,6 +4,7 @@
 import csv
 import json
 import os
+import time
 from logging import Logger
 from typing import IO, Dict
 
@@ -56,6 +57,7 @@ def output_pdf(all_cve_data: Dict[ProductInfo, CVEData], outfile):
     pdfdoc = pdfbuilder.PDFBuilder()
     cm = pdfdoc.cm
     severity_colour = {
+        "UNKNOWN": pdfdoc.grey,
         "LOW": pdfdoc.blue,
         "MEDIUM": pdfdoc.green,
         "HIGH": pdfdoc.orange,
diff --git a/cve_bin_tool/output_engine/pdfbuilder.py b/cve_bin_tool/output_engine/pdfbuilder.py
index 1dfc80c9ed..097a770bc1 100644
--- a/cve_bin_tool/output_engine/pdfbuilder.py
+++ b/cve_bin_tool/output_engine/pdfbuilder.py
@@ -72,6 +72,14 @@ class PDFBuilder:
 
     body = PS(name="body", fontSize=12, fontName="Helvetica", leading=12)
 
+    body_unknown = PS(
+        name="body",
+        fontSize=12,
+        textColor=colors.grey,
+        fontName="Helvetica-Bold",
+        leading=12,
+    )
+
     body_low = PS(
         name="body",
         fontSize=12,
@@ -123,6 +131,7 @@ class PDFBuilder:
         ]
     )
 
+    grey = colors.grey
     blue = colors.blue
     red = colors.red
     green = colors.green