Skip to content

Commit c923840

Browse files
committed
Add reference score to package endpoint
Signed-off-by: Keshav Priyadarshi <[email protected]>
1 parent ef1df71 commit c923840

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

vulnerabilities/api.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,14 +155,26 @@ class VulnSerializerRefsAndSummary(BaseResourceSerializer):
155155
many=True, source="filtered_fixed_packages", read_only=True
156156
)
157157

158-
references = VulnerabilityReferenceSerializer(many=True, source="vulnerabilityreference_set")
158+
references = serializers.SerializerMethodField()
159159

160160
aliases = serializers.SerializerMethodField()
161161

162162
def get_aliases(self, obj):
163163
# Assuming `obj.aliases` is a queryset of `Alias` objects
164164
return [alias.alias for alias in obj.aliases.all()]
165165

166+
def get_references(self, vulnerability):
167+
references = vulnerability.vulnerabilityreference_set.all()
168+
severities = vulnerability.severities.all()
169+
170+
serialized_references = VulnerabilityReferenceSerializer(
171+
references,
172+
context={"severities": severities},
173+
many=True,
174+
).data
175+
176+
return serialized_references
177+
166178
class Meta:
167179
model = Vulnerability
168180
fields = ["url", "vulnerability_id", "summary", "references", "fixed_packages", "aliases"]

0 commit comments

Comments
 (0)