Skip to content

Commit fbff4f6

Browse files
committed
Don't use Q object for bulk vulnerability api
Signed-off-by: Shivam Sandbhor <[email protected]>
1 parent 202cd75 commit fbff4f6

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

vulnerabilities/api.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -171,16 +171,15 @@ class VulnerabilityViewSet(viewsets.ReadOnlyModelViewSet):
171171
# TODO: Find a good name for this endpoint
172172
@action(detail=False, methods=["post"])
173173
def fetch(self, request):
174-
filter_list = Q()
174+
filter_list = []
175175
response = {}
176176
# TODO: Do some validation here of request body
177177
for cve_id in request.data["vulnerabilities"]:
178-
filter_list |= Q(cve_id=cve_id)
179-
178+
filter_list.append(cve_id)
180179
# This handles the case when the said cve doesnt exist in db
181180
response[cve_id] = {}
182181

183-
res = Vulnerability.objects.filter(filter_list)
182+
res = Vulnerability.objects.filter(cve_id__in=[cve_id])
184183
for vuln in res:
185184
response[vuln.cve_id] = VulnerabilitySerializer(vuln, context={"request": request}).data
186185

0 commit comments

Comments
 (0)