Preserve old API response for reference and severity

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

Author: keshav-space

vulnerabilities/api.py

@@ -62,9 +62,11 @@ class Meta:
         fields = ["reference_url", "reference_id", "reference_type", "scores", "url"]
 
     def get_scores(self, instance):
-        matching_scores = VulnerabilitySeverity.objects.filter(url=instance.url)
+        severities_related_to_reference = []
+        if vulnerability := self.context.get("vulnerability"):
+            severities_related_to_reference = vulnerability.severities.filter(url=instance.url)
 
-        return VulnerabilitySeveritySerializer(matching_scores, many=True).data
+        return VulnerabilitySeveritySerializer(severities_related_to_reference, many=True).data
 
 
 class BaseResourceSerializer(serializers.HyperlinkedModelSerializer):
@@ -204,8 +206,7 @@ class VulnerabilitySerializer(BaseResourceSerializer):
         many=True, source="filtered_fixed_packages", read_only=True
     )
     affected_packages = MinimalPackageSerializer(many=True, read_only=True)
-
-    references = VulnerabilityReferenceSerializer(many=True, source="vulnerabilityreference_set")
+    references = serializers.SerializerMethodField()
     aliases = AliasSerializer(many=True, source="alias")
     exploits = ExploitSerializer(many=True, read_only=True)
     weaknesses = WeaknessSerializer(many=True)
@@ -219,6 +220,17 @@ def to_representation(self, instance):
 
         return data
 
+    def get_references(self, vulnerability):
+        references = vulnerability.vulnerabilityreference_set.all()
+
+        serialized_references = VulnerabilityReferenceSerializer(
+            references,
+            context={"vulnerability": vulnerability},
+            many=True,
+        ).data
+
+        return serialized_references
+
     def get_severity_range_score(self, instance):
         severity_vectors = []
         severity_values = set()

vulnerabilities/api_extension.py

@@ -126,7 +126,6 @@ class V2VulnerabilitySerializer(ModelSerializer):
 
     aliases = SerializerMethodField("get_aliases")
     weaknesses = V2WeaknessSerializer(many=True, source="weaknesses_set")
-    scores = V2VulnerabilitySeveritySerializer(many=True, source="vulnerabilityseverity_set")
     references = V2VulnerabilityReferenceSerializer(many=True, source="vulnerabilityreference_set")
     exploits = V2ExploitSerializer(many=True, source="weaknesses")
     severities = V2VulnerabilitySeveritySerializer(many=True)
@@ -145,7 +144,6 @@ class Meta:
             "vulnerability_id",
             "aliases",
             "status",
-            "scores",
             "weaknesses",
             "summary",
             "exploits",