Django-rest-framework中过滤器的定制

Django-rest-framework中过滤器的定制

1. 定义一个自定义的filter.py模块，增加一个新的过滤类

import django_filters#这个Q可以支持表查询，单下划线获取表字段，双下划线获取关联表，from django.db.models import Q#引入自己的模型from app.item.models import ItemCategory#自定义的过滤类，需要继承django_filter.rest_framework中的FilterSet类class ItemCategoryFilter(django_filters.rest_framework.FilterSet):    Filter_category = django_filters.NumberFilter(method='category_filter', label='Filter_category')    #django_filters.NumberFilter类似，ModelForm中字段类型的控制    #其中method指向自己定义的过滤函数，label用于标识在测试API界面中的过滤界面字段，Filter_category控制查询字段    def category_filter(self, queryset, name, value):        #这里用到多级联表查询        return queryset.filter(Q(category_id=value)|Q(category__parent_id=value)|Q(category__parent__parent_id=value)|Q(category__parent__parent__parent_id=value))    class Meta:        model = ItemCategory        fields = []

2.将视图处理类中的，过滤器后端重新指定，将过滤器类连接到我们自定义实现的处理类上

class ItemsListByCategory(generics.ListCreateAPIView):    '''        根据category编号获取相关模型no。        请求中请带有参数category        @:require param: category        such as: GET /itemsListByCategory/?Filter_category=2    '''    from django_filters.rest_framework import DjangoFilterBackend    from app.item.filters import ItemCategoryFilter    queryset = ItemCategory.objects.all()    serializer_class = ItemCategorySerializer    pagination_class = CommonPagination    # 在这里哦    filter_backends = (DjangoFilterBackend,)    filter_class =ItemCategoryFilter    permission_classes = (ReadOnly,)