如何在django admin.SimpleListFilter中访问已过滤的查询集

Enu*_*nuy 8 django django-admin

假设我们有两个对象:

class Author(models.Model):
    name = models.CharField(length=50)

class Book(models.Model):
    author = models.ForeignKey(Author)
    is_bestseller = models.BooleanField()
Run Code Online (Sandbox Code Playgroud)

在BookAdmin中,如果我们指定

list_filter = ('author', 'is_bestseller')
Run Code Online (Sandbox Code Playgroud)

"作者"给你的选择将永远是数据库中的所有作者,无论他们是否写过畅销书.

我希望我的过滤器的选择受到当前选择的限制.我尝试使用admin.SimpleListFilter以一般方式执行此操作,但是,我被卡住了,因为:

model_admin.get_queryset()
Run Code Online (Sandbox Code Playgroud)

返回未过滤的查询集(即所有对象,而不仅仅是当前由用户过滤掉的对象).

如何获取BookAdmin 的筛选查询集?

编辑:为了更好地说明问题,这是我的代码:

class ForeignFieldFilter(admin.SimpleListFilter):
    field = None

    def lookups(self, request, model_admin):
        queryset = model_admin.get_queryset(request).select_related(self.field)
        # ^^^ !!! Returns UNFILTERED queryset - it's not affected by other filters

        field_ids = queryset.values_list(self.field, flat=True)

        for field_value in self.model..objects.filter(id__in=field_ids):
            yield (field_value.pk, str(field_value))

    def queryset(self, request, queryset):
        value = self.value()

        # Check that any value was passed, if not, return unmodified queryset
        if not value:
            return queryset

        return queryset.filter({'{0}__pk'.format(self.field): value})


def foreign_field_filter_factory(field, model, title=None):
    title = title or (field[:1].upper() + field[1:])
    return type(
        "ForeignFieldFilter_{0}".format(field),
        (ForeignFieldFilter,),
        {
            'field': field,
            'title': title,
            'parameter_name': field,
            'model': model
        })
Run Code Online (Sandbox Code Playgroud)

war*_*der 3

对于图书管理员

class AuthorBestSeller(admin.SimpleListFilter):
    title = 'Best selling authors'
    parameter_name = 'bestseller'

    def queryset(self, request, queryset):
        if self.value():
            return queryset.filter(author_id=self.value())
        else:
            return queryset

    def lookups(self, request, model_admin):
        qs = model_admin.get_queryset(request)
        query_attrs = dict([(param, val) for param, val in request.GET.items()])
        qs = qs.filter(**query_attrs)
        for book in qs.filter(is_bestseller=1):  # or might be able to use yeild here
            ret.append((book.author_id, book.author))
        return ret
Run Code Online (Sandbox Code Playgroud)

编辑:嗯...这也不太管用。它只会为您提供畅销书作者的列表。

然后添加到BookAdmin

list_filter = (AuthorBestSeller, )
Run Code Online (Sandbox Code Playgroud)