Export new severities relations

Signed-off-by: Keshav Priyadarshi <[email protected]>

Author: keshav-space

vulnerabilities/management/commands/export.py

@@ -22,20 +22,12 @@
 
 
 def serialize_severity(sev):
-    # inlines refs
-    ref = sev.reference
-    sevref = {
-        "url": ref.url,
-        "reference_type": ref.reference_type,
-        "reference_id": ref.reference_id,
-    }
-
     return {
         "score": sev.value,
         "scoring_system": sev.scoring_system,
         "scoring_elements": sev.scoring_elements,
         "published_at": sev.published_at,
-        "reference": sevref,
+        "url": sev.url,
     }
 
 
@@ -44,7 +36,7 @@ def serialize_vulnerability(vuln):
     Return a plain data mapping seralized from ``vuln`` Vulnerability instance.
     """
     aliases = list(vuln.aliases.values_list("alias", flat=True))
-    severities = [serialize_severity(sev) for sev in vuln.severities]
+    severities = [serialize_severity(sev) for sev in vuln.severities.all()]
     weaknesses = [wkns.cwe for wkns in vuln.weaknesses.all()]
 
     references = list(
@@ -161,11 +153,11 @@ def packages_by_type_ns_name():
             "affected_by_vulnerabilities",
             "affected_by_vulnerabilities__references",
             "affected_by_vulnerabilities__weaknesses",
-            "affected_by_vulnerabilities__references__vulnerabilityseverity_set",
+            "affected_by_vulnerabilities__severities",
             "fixing_vulnerabilities",
             "fixing_vulnerabilities__references",
             "fixing_vulnerabilities__weaknesses",
-            "fixing_vulnerabilities__references__vulnerabilityseverity_set",
+            "fixing_vulnerabilities__severities",
         )
         .paginated()
     )

vulnerabilities/tests/test_export.py

@@ -50,7 +50,7 @@ def vulnerability_severity(vulnerability_reference):
         scoring_system="cvssv3_vector",
         value="7.0",
         scoring_elements="CVSS:3.0/AV:N/AC:L/PR:N/UI:N/S:U/C:N/I:N/A:H",
-        reference_id=vulnerability_reference.id,
+        url=f"https://..",
     )
 
 
@@ -66,6 +66,7 @@ def vulnerability(db, vulnerability_reference, vulnerability_severity):
 
     weakness = Weakness.objects.create(cwe_id=15)
     vulnerability.weaknesses.add(weakness)
+    vulnerability.severities.add(vulnerability_severity)
 
     return vulnerability