Django：按多个ID过滤queryset

Question

My query is quite simple, I have a model Vendor in my Django REST app. What I want is to use a get response with a few ID and get back all the respective models with these ID. The GET url pattern could be something like this: r'^api/vendors?id=1,2,3' .

What I'm thinking of right now is to use ListAPIView , and in the list method filter my queryset with all the id in the url. But I'm not sure exactly how to achieve this (filtering queryset with a list of id, I'm very new to both Python and Django), so if anyone can provide any advice on this it would be greatly appreciated.

Answer 1

(Unfortunately I do not know django REST, so here is a pure django solution)

Using the ListAPIView you can access the URL (or GET) params and modify the queryset.

class MyVendorView(ListAPIView):
    # attributes

    def get_queryset(self):
        id_string = self.request.GET.get('id')
        if id_string is not None:
            ids = [int(id) for id in id_string.split(',')]
            return Vendor.objects.filter(id__in=ids)

        else:
            return Vendor.objects.all()

    # other methods

please note that I'm ingoring any attributes or other attributes needed

What's happening here then?

• Overriding get_queryset will control what results we get from hitting the view

• self.request.GET.get('id') Will extract the value of the id query parameter from the url like so localhost:8000/api/vendors?id=1,2,3 the result will be a string "1,2,3".

• filter(id__in=ids) lets you say select stuff that has an value in this list of ids