仅在DRF中检索当前用户的数据

Question

我想知道如何将查询集限制为与当前用户关联的数据。以下是我的看法：

class TradingGroupList(generics.ListAPIView):
        queryset = Tradegroup.objects.all()
        serializer_class = TradeGroupSerializer
        name = 'tradegroup-list'

我会用母语django写一些类似queryset = Tradegroup.objects.filter(owner=self.request.user)的东西，但我想知道我如何在这里实现这一点

Answer 1

这实际上应该通过重写get_queryset方法来实现。只需将此方法添加到ListAPIView中，它就可以工作了

class TradingGroupList(generics.ListAPIView):
    
    serializer_class = TradeGroupSerializer
    name = 'tradegroup-list'

    def get_queryset(self)
        return Tradegroup.objects.all().filter(owner=self.request.user)

我希望这对你有用。如果你有问题，就给我一个评论