如何在不破坏 pk 查询的情况下正确覆盖模型视图集中的 get_queryset？

Question

我有一个模型视图集：

class OrderViewSet(viewsets.ModelViewSet):
serializer_class = OrderSerializer
pagination_class = DefaultPagination
queryset = Order.objects.all()

def get_queryset(self):
    userId = self.request.query_params.get('userId')
    if userId is not None:
        query = Order.objects.filter(owner__id=userId).order_by('-createdAt')
    else:
        query = Order.objects.order_by('-createdAt')
    return query

这里是url注册

router = routers.DefaultRouter()
router.register('api/v1/order', OrderViewSet, basename='order')

我发现在覆盖get_queryset之后，我不能再像下面这样通过一个订单id查询了：/api/v1/order/1，它只是returns detail: not found

我确实看到 django 说它支持：

^api/v1/order/$ [name='order-list']
^api/v1/order\.(?P<format>[a-z0-9]+)/?$ [name='order-list']
^api/v1/order/(?P<pk>[^/.]+)/$ [name='order-detail']
^api/v1/order/(?P<pk>[^/.]+)\.(?P<format>[a-z0-9]+)/?$ [name='order-detail']

执行此操作的正确方法应该是什么？谢谢！

Answer 1

使用 Django 的基于函数的视图和基于 class 的视图如何对您有利。我可以想到一个解决方案，您可以创建另一个基于函数的视图来处理上述功能，并让基于 class 的视图 get 方法处理 pk 查询集。

Answer 2

要解决此问题，请在覆盖 get_queryset()

时调用 super()

试试这个

def get_queryset(self):
    queryset = super(OrderViewSet, self).get_queryset()

    userId = self.request.query_params.get('userId')
    if userId is not None:
         queryset = queryset.filter(owner__id=userId).order_by('-createdAt')
    else:
        queryset = queryset.order_by('-createdAt')
    return query

如何在不破坏 pk 查询的情况下正确覆盖模型视图集中的 get_queryset？

How to properly override get_queryset in a modelviewset without breaking the query by pk?

django

django-rest-framework