使用 Django REST 框架从多个模型返回结果

发布于09月10日

我有三种模式——文章、作者和推文.我最终需要使用Django REST框架来构建一个提要，该提要使用文章和推特模型将所有对象聚合到一个反向按时间顺序排列的提要中.

知道我该怎么做吗？我觉得我需要创建一个新的序列化程序，但我真的不确定.

谢谢!

这是我到目前为止所做的.

app/serializers.py:个

class TimelineSerializer(serializers.Serializer):
    pk = serializers.Field()
    title = serializers.CharField()
    author = serializers.RelatedField()
    pub_date = serializers.DateTimeField()

app/views.py:个

class TimelineViewSet(viewsets.ModelViewSet):
    """
    API endpoint that lists all tweet/article objects in rev-chrono.
    """
    queryset = itertools.chain(Tweet.objects.all(), Article.objects.all())
    serializer_class = TimelineSerializer

推荐答案

它看起来离我很近.我个人还没有在DRF中使用ViewSet，但我认为如果您将代码更改为这个，您应该找到一些地方(抱歉，这两个都没有经过测试):

class TimelineViewSet(viewsets.ModelViewSet):
    """
    API endpoint that lists all tweet/article objects in rev-chrono.
    """
    def list(self, request):
        queryset = list(itertools.chain(Tweet.objects.all(), Article.objects.all()))
        serializer = TimelineSerializer(queryset, many=True)
        return Response(serializer.data)

如果您不想使用ViewSet，那么generics.ListAPIView会更简单一些:

class TimeLineList(generics.ListAPIView):
    serializer_class = TimeLineSerializer

    def get_queryset(self):
        return list(itertools.chain(Tweet.objects.all(), Article.objects.all()))

注您必须将chain的输出转换为列表，这样才能正常工作.