Apply filter to search results

[igorpejic]

Apply filter to search results

[igorpejic]

Ok, thank you. I modified your code and the final solution would be this code:

class AppleViewSet(HaystackViewSet):

    # See http://drf-haystack.readthedocs.org/en/latest/basic_usage.html#views-py for how to set up the HaystackViewSet

    def filter_queryset(self, *args, **kwargs):
        queryset = super(AppleViewSet, self).filter_queryset(self.get_queryset())
        return queryset.filter(eater=self.request.user.pk)

[rhblind]

Hi,
Absolutely! But keep in mind that the QuerySet is a SearchQuerySet not a regular one, so you need to have the field you want to filter on in the index.

So you are almost there in the example provided, but you should make sure that the eater field in the index is holding the user id (We can't index python models). Then make the filter like this:

queryset=self.list(request, *args, **kwargs).filter(eater=request.user.pk)

and you should be good to go!

[igorpejic]

@rhblind Thank you for your help, but maybe I am doing something wrong.
I can't do what you suggested because

 self.list(request, *args, **kwargs)

is a rest framework response of type: <class 'rest_framework.response.Response'>, not a SearchQuerySet.

[rhblind]

Ah, right =)
In that case I'd suggest to sublcass from the HaystackViewSet where you can override ie. the filter_queryset() method.

Something like this should do it:

class AppleViewSet(HaystackGenericAPIView):

    # See http://drf-haystack.readthedocs.org/en/latest/basic_usage.html#views-py for how to set up the HaystackViewSet

    def filter_queryset(self):
        queryset = super(AppleViewSet, self).filter_queryset()
        return queryset.filter(eater=self.request.user.pk)

[rhblind]

Yeah, sorry about the typo there.
That was what I meant :)