用于 POST 请求的 Django rest 框架自定义过滤器

pun*_*lly 6 python django post filter django-rest-framework

在filters.py 中,我CustomFilter定义了一个类型为ComboSortFilter和的值IntegerListFilter

在views.py 中,我定义了一个ViewSet,filter_class = CustomFilter它对GET请求非常有用,在url 中使用查询字符串参数,使用自定义过滤进行处理。

我还需要支持POST请求,因为有些过滤条件太长,无法放入 URL。

所以我post在我的 ViewSet 中添加了一个方法,我从中提取参数,request.DATA然后过滤,序列化和分页它们......

在views.py中我的ViewSet的post方法中:

queryset = MyModel.objects.filter(**filter_args)
page = self.paginate_queryset(queryset)
serializer = self.get_pagination_serializer(page)
return Response(serializer.data)
Run Code Online (Sandbox Code Playgroud)

对于简单的过滤,上述工作正常。但是,我在CustomFilter过滤字段中定义的过滤使用ComboSortFilterIntegerListFilter有点复杂:

在过滤器.py中:

class IntegerListFilter(django_filters.Filter):
    def filter(self, qs, value):
        if value not in (None, ''):
            integers = [int(v) for v in value.split(',')]
            return qs.filter(**{'{0}__{1}'.format(self.name, self.lookup_type): integers})
        return qs

class ComboSortFilter(django_filters.Filter):
    def __init__(self, threshold, lookup_type, order='ASC'):
        super(ComboSortFilter, self).__init__(lookup_type=lookup_type)
        self.threshold = threshold
        self.order = order

    def filter(self, qs, value):
        if value not in (None, ''):
            fields = [str(v) for v in value.split(',')]
            for field in fields:  # filters each field's values relative to threshold
                qs = qs.filter(**{'{0}__{1}'.format(field, self.lookup_type): self.threshold})
            if self.order == 'DESC':
                fields = ['-{0}'.format(f) for f in fields]
            qs = qs.order_by(*fields)
        return qs

class CustomFilter(django_filters.FilterSet):
    thing_ids = IntegerListFilter(name="thing_id", lookup_type='in')
    sort_desc = ComboSortFilter(lookup_type='gte', threshold=100, order='DESC')
    sort_asc = ComboSortFilter(lookup_type='lte', threshold=100)
Run Code Online (Sandbox Code Playgroud)

有没有办法CustomFilter从 ViewSet 的 post 方法中引用FilterSet 或我的两个自定义过滤器中的任何一个?我不想复制所有这些过滤代码views.py才能让它工作。

所有关于Django-rest-framework:filters的文档只包含简单的例子,要么在视图中滚动你自己的,要么使用过滤器类。有没有办法直接从视图调用过滤器类,覆盖它通常从查询字符串中获取的值?

pun*_*lly 7

弄清楚了:

class MyPostViewSet(BaseModelViewSet):

    def post(self, request, *args, **kwargs):
        queryset = self.queryset
        filter_params = request.DATA or request.GET
        if filter_params:
            queryfilter = self.filter_class(filter_params, queryset=queryset)
            queryset = queryfilter.qs
        page = self.paginate_queryset(queryset)
        serializer = self.get_pagination_serializer(page)
        return Response(serializer.data)
Run Code Online (Sandbox Code Playgroud)