ARTS打卡计划第四周-TIPS-自定义一个LikeFilter

django-restframe-work的searchfilter，可以从search_fields 的一个或者多个字段中搜索，语法类似http://example.com/api/products?search=123。

目前有一个需求，假设search_fields = ('username', 'email')，username 要包含‘bob’ ,email要包含'qq.com'，这个时候search语法将失效。下面分享一种可以LikeFilter，可以分别对每个字段进行like.

首先定义一个LikeFilter

from rest_framework.filters import BaseFilterBackend


class LikeFilter(BaseFilterBackend):
    def filter_queryset(self, request, queryset, view):
        search_fields = getattr(view, 'like_fields', None)
        kwargs = {}
        for query_fields in search_fields:
            query_value = request.query_params.get(query_fields,'')
            if query_value:
                kwargs['{0}__{1}'.format(query_fields, 'contains')]=query_value

        return queryset.filter(**kwargs)

　　

使用也很简单，在你的viewset中加入属性：like_fields=('username','email')，然后你可以在你的url中使用了：http://example.com/api/products?username=bob&email=qq.com