django - 如何使用 Django Rest Framework 为列表和详细 View 设置不同的序列化器？

Question

在使用 viewsets.ModelViewSet 和 HyperlinkedSerializer 时，如何为列表和详细 View 设置不同的序列化器？

我知道如何通过定义 list 和 retrive 来使用 viewsets.ViewSet 来实现这一点，( here's an example )，但我不知道知道如何优雅地使其适应 viewsets.ModelViewSet

Answer 1

我改编了“Django rest framework, use different serializers in the same ModelViewSet”的答案，它对我很有用，希望您会发现有用:

class MyModelViewSet(viewsets.MyModelViewSet):

    queryset = MyModel.objects.all()
    serializer_class = MyModelListSerializer
    detail_serializer_class = MyModelDetailSerializer

    def get_serializer_class(self):
        if self.action == 'retrieve':
            if hasattr(self, 'detail_serializer_class'):
                return self.detail_serializer_class

        return super(MyModelViewSet, self).get_serializer_class()

在这种情况下，您只需指定两个序列化器并根据操作使用一个序列化器。但是，这可以变得更通用(对于所有操作)，如下所示:

class MyModelViewSet(viewsets.MyModelViewSet):

    queryset = MyModel.objects.all()
    serializer_class = MyModelSerializer

    action_serializers = {
        'retrieve': MyModelDetailSerializer,
        'list': MyModelListSerializer,
        'create': MyModelCreateSerializer
    }

    def get_serializer_class(self):

        if hasattr(self, 'action_serializers'):
            return self.action_serializers.get(self.action, self.serializer_class)

        return super(MyModelViewSet, self).get_serializer_class()