根据请求在 django-filter ModelChoiceFilter (select) 和 ModelMultip

Question

我在两个地方使用 django-filter：我的 Django Rest Framework API 和我的 FilterViews(Django Filter 的通用列表视图(。在我的FilterViews中，我同时显示要过滤的选择框(ModelChoiceFilter(和多选框(ModelMultipleChoiceFilter(。

我需要能够根据请求中的字段限制这些选择和多选输入中的内容。

更改 FilterSet 中相关字段中列为 kwarg 的内容相对简单。例如，这是我的 FilterSet，其中查询集设置为 kwarg：

class FieldFilter(django_filters.FilterSet):
"""Filter for the field list in the API"""
dataset = ModelChoiceFilter(queryset=Dataset.objects.all())
class Meta(object):
"""Meta options for the filter"""
model = Field
fields = ['dataset']

在get_queryset()方法中限制 DRF中的结果相对简单。例如，这是我的 DRF 视图集：

class FieldViewSet(viewsets.ReadOnlyModelViewSet):
"""A ViewSet for viewing dataset fields"""
queryset = Field.objects.all()
serializer_class = FieldSerializer
filter_class = FieldFilter
def get_queryset(self):
"""Get the queryset"""
queryset = super(FieldViewSet, self).get_queryset()
queryset = queryset.filter(
dataset__organization=self.request.organization)
return queryset

显示视图时，我只是找不到任何地方来编辑filter_class中的Dataset字段。

这在 DjangoFormView通用视图中非常简单，但似乎FieldViewSet遵循与通用视图相同的get_form()结构。在管理中也相对简单，但DRF/Django-Filter似乎也不遵循这种结构。

有没有办法在每个请求的基础上自定义这些输入中的查询集？最好在 FilterViews 和 HTML API 浏览器中，但如果对于 HTML API 浏览器来说太复杂，那么仅在 FilterViews 中就可以了。

Answer 1

经过数小时的搜索，我在这里的官方文档中找到了解决方案！

ModelChoiceFilter和ModelMultipleChoiceFilter的查询集参数支持可调用行为。如果传递了一个可调用对象，它将以request作为其唯一参数进行调用。

import django_filters as filters
from django.utils.translation import gettext as _
def ourBranches(request):
if request is None:
return Branch.objects.none()
company = request.user.profile.company
return Branch.objects.filter(company=company)

class UnitFilter(filters.FilterSet):
branch = filters.ModelChoiceFilter(
queryset=ourBranches, empty_label=_("All Branches"))
class Meta:
model = Unit
fields = ('branch', )

在看来，我也确保通过request

qs = Unit.objects.all()
filter = UnitFilter(self.request.GET, request=self.request, queryset=qs) 
table = UnitTable(filter.qs)

Answer 2

我也在寻找解决方案时遇到了问题。

我通过以下方式解决了它(我认为(：

views.py

table_filter = ExampleFilter(request.GET, kwarg_I_want_to_pass=request.user, queryset=qs)

filters.py

class ExampleFilter(django_filters.FilterSet):
def __init__(self, *args, **kwargs):
self.user = kwargs.pop('kwarg_I_want_to_pass', None)
super(ExampleFilter, self).__init__(*args, **kwargs)
self.filters['field_to_filter'].extra.update({
'queryset': Supplier.objects.filter(related_user=self.user),
'empty_label': '',
'help_text': False
})
class Meta:
model = ExampleModel
fields = ['related_user', 'field_to_filter', ... other fields]