Fix null constraint violations in multiple v1 exploit collection pipelines

Signed-off-by: ziad hany <ziadhany2016@gmail.com>
Signed-off-by: Tedsig42 <teddams047@gmail.com>

Author: ziadhany

vulnerabilities/pipelines/enhance_with_exploitdb.py

@@ -88,7 +88,8 @@ def add_vulnerability_exploit(row, logger):
     for raw_alias in aliases:
         try:
             if alias := Alias.objects.get(alias=raw_alias):
-                vulnerabilities.add(alias.vulnerability)
+                if alias.vulnerability:
+                    vulnerabilities.add(alias.vulnerability)
         except Alias.DoesNotExist:
             continue
 

vulnerabilities/pipelines/enhance_with_kev.py

@@ -74,7 +74,8 @@ def add_vulnerability_exploit(kev_vul, logger):
     vulnerability = None
     try:
         if alias := Alias.objects.get(alias=cve_id):
-            vulnerability = alias.vulnerability
+            if alias.vulnerability:
+                vulnerability = alias.vulnerability
     except Alias.DoesNotExist:
         logger(f"No vulnerability found for aliases {cve_id}")
         return 0

vulnerabilities/pipelines/enhance_with_metasploit.py

@@ -79,7 +79,8 @@ def add_vulnerability_exploit(record, logger):
     for ref in interesting_references:
         try:
             if alias := Alias.objects.get(alias=ref):
-                vulnerabilities.add(alias.vulnerability)
+                if alias.vulnerability:
+                    vulnerabilities.add(alias.vulnerability)
         except Alias.DoesNotExist:
             continue