Faster Weakness Lookup

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/models.py

@@ -57,9 +57,6 @@
 from vulnerabilities.utils import normalize_purl
 from vulnerabilities.utils import purl_to_dict
 from vulnerablecode import __version__ as VULNERABLECODE_VERSION
-from cwe2.weakness import Weakness as DBWeakness
-from cwe2.mappings import xml_database_path
-import xml.etree.ElementTree as ET
 
 logger = logging.getLogger(__name__)
 
@@ -473,6 +470,7 @@ def get_severity_vectors_and_values(self):
 
         return severity_vectors, severity_values
 
+
 def get_cwes(self):
     """Yield CWE Weakness objects"""
     for cwe_category in self.cwe_files:
@@ -485,7 +483,16 @@ def get_cwes(self):
     for tag_num in [1, 2]:  # Categories , Views
         tag = root[tag_num]
         for child in tag:
-            yield DBWeakness(*[child.attrib["ID"], child.attrib.get("Name"),None,child.attrib.get("Status"),child[0].text])
+            yield DBWeakness(
+                *[
+                    child.attrib["ID"],
+                    child.attrib.get("Name"),
+                    None,
+                    child.attrib.get("Status"),
+                    child[0].text,
+                ]
+            )
+
 
 Database.get_cwes = get_cwes
 

vulnerabilities/views.py

@@ -224,8 +224,8 @@ def get_context_data(self, **kwargs):
                 NotImplementedError,
             ):
                 logging.error(f"CVSSMalformedError for {severity.scoring_elements}")
-        
-        epss_severity = vulnerability.severities.filter(scoring_system='epss').first()
+
+        epss_severity = vulnerability.severities.filter(scoring_system="epss").first()
         epss_data = None
         if epss_severity:
             epss_data = {