DRF 所需的搜索过滤器

Question

我有工作正常的视图集

class PlayerViewSet(ModelViewSet):
     queryset = Player.objects.all()
     serializer_class = PlayerSerializer
     filter_backends = [SearchFilter]
     search_fields = ['nickname']

网址配置：

    path('player', PlayerViewSet.as_view({'get': 'list', 'post': 'create'})),

当我尝试使用端点搜索对象时一切正常：'player?search=randomNickname'

但是有可能在没有搜索参数的情况下发送 GET 请求并返回所有可能的对象

有没有办法根据需要设置搜索字段？

Answer 1

由于ModelViewSet可以执行各种操作（获取、更新、删除 ..），SearchFilter如果没有传递搜索参数或出现错误，则扩展并使其返回空结果可能会更好。

from rest_framework import filters
from rest_framework.exceptions import ValidationError


class CustomSearchFilter(filters.SearchFilter):
    def filter_queryset(self, request, queryset, view):
        search_terms = self.get_search_terms(request)

        if not search_terms:
            # turn queryset in EmptyQuerySet
            # so no objects will be returned
            return queryset.none()

            # Alternatively, an error can be raise  
            #raise ValidationError(
            #    "%s parameter is required!" % self.search_param
            #)

        return super().filter_queryset(request, queryset, view)


class PlayerViewSet(ModelViewSet):
    queryset = Player.objects.all()
    serializer_class = PlayerSerializer
    filter_backends = [CustomSearchFilter]
    search_fields = ['nickname']