Skip to content

Commit f56a9f0

Browse files
committed
fix: use proper response type in swagger schema
Signed-off-by: Keshav Priyadarshi <git@keshav.space>
1 parent 02720af commit f56a9f0

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

‎vulnerabilities/api_v3.py‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,11 @@ class PackageV3ViewSet(viewsets.GenericViewSet):
275275
filter_backends = [filters.DjangoFilterBackend]
276276
throttle_classes = [AnonRateThrottle, PermissionBasedUserRateThrottle]
277277

278-
@extend_schema(request=PackageQuerySerializer)
278+
@extend_schema(
279+
operation_id="v3_packages_retrieve",
280+
request=PackageQuerySerializer,
281+
responses={200: PackageV3Serializer(many=True)},
282+
)
279283
def create(self, request, *args, **kwargs):
280284
serializer = PackageQuerySerializer(data=request.data)
281285
serializer.is_valid(raise_exception=True)
@@ -411,7 +415,11 @@ class AdvisoryV3ViewSet(viewsets.GenericViewSet):
411415
filter_backends = [filters.DjangoFilterBackend]
412416
throttle_classes = [AnonRateThrottle, PermissionBasedUserRateThrottle]
413417

414-
@extend_schema(request=AdvisoryQuerySerializer)
418+
@extend_schema(
419+
operation_id="v3_advisories_retrieve",
420+
request=AdvisoryQuerySerializer,
421+
responses={200: AdvisoryV3Serializer(many=True)},
422+
)
415423
def create(self, request, *args, **kwargs):
416424
serializer = AdvisoryQuerySerializer(data=request.data)
417425
serializer.is_valid(raise_exception=True)

0 commit comments

Comments
 (0)