python - Django Rest Framework ViewSet 的其他 View

Question

我有一个模型的简单 DRF ViewSet，位于 /gen_req/

class GenerationRequestViewSet(viewsets.ModelViewSet):
    queryset = GenerationRequest.objects
    serializer_class = GenerationRequestSerializer

它有默认的POST/GET/等。处理程序。但是，我想为 GET 添加另一个用于不同的 url 模式 (/gen_req/created_list:

class GenerationRequestViewSet(viewsets.ModelViewSet):
    queryset = GenerationRequest.objects
    serializer_class = GenerationRequestSerializer

    @action(methods=['get'])
    def special_get_handler(self, request):
        queryset = GenerationRequest.filter(...) # Some extra filtering here
        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data)

有没有办法在 ViewSet 中创建这样的 View ，或者应该创建另一个 APIView？

Answer 1

你可以做到这一点，只需将这样的记录添加到你的 urls.py 文件中即可。

path('/gen_req/created_list',
     GenerationRequestViewSet.as_view({'get': 'special_get_handler'}),),