如何覆盖list_filter中提供过滤器的查询集?

时间:2012-09-21 00:08:37

标签: django admin

鉴于以下模型

class AnotherModel(models.Model):
    n = models.IntegerField()

class MyModel(models.Model):
    somefield = models.ForeignKey(AnotherModel)

和admin

class MyModelAdmin(admin.ModelAdmin):        
    list_filter = ('somefield',)

如何过滤AnotherModel的实例,以便仅显示管理过滤器中具有给定n值的实例?

我需要类似的东西:

过滤

by somefield

所有

[给定n的AnotherModel实例列表]

4 个答案:

答案 0 :(得分:23)

请参阅ModelAdmin.querysetModelAdmin.formfield_for_foreignkey。来自文档:

  

ModelAdmin上的queryset方法返回可由管理站点编辑的所有模型实例的QuerySet。覆盖此方法的一个用例是显示登录用户拥有的对象:

class MyModelAdmin(admin.ModelAdmin):
    def queryset(self, request):
        qs = super(MyModelAdmin, self).queryset(request)
        if request.user.is_superuser:
            return qs
        return qs.filter(author=request.user)
  

ModelAdmin上的formfield_for_foreignkey方法允许您覆盖外键字段的默认表单字段。例如,要根据用户返回此外键字段的对象子集:

class MyModelAdmin(admin.ModelAdmin):
    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == "car":
            kwargs["queryset"] = Car.objects.filter(owner=request.user)
        return super(MyModelAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
  

这使用HttpRequest实例过滤Car外键字段,仅显示User实例拥有的汽车。

[更新]

抱歉,我没有阅读“过滤器”部分。在Django> = 1.4中,您可以在list_filter argument list中传递django.contrib.admin.SimpleListFilter的子类,您可以使用它来覆盖查找和查询集方法。

from datetime import date

from django.contrib import admin
from django.utils.translation import ugettext_lazy as _

class DecadeBornListFilter(admin.SimpleListFilter):
    # Human-readable title which will be displayed in the
    # right admin sidebar just above the filter options.
    title = _('decade born')

    # Parameter for the filter that will be used in the URL query.
    parameter_name = 'decade'

    def lookups(self, request, model_admin):
        """
        Returns a list of tuples. The first element in each
        tuple is the coded value for the option that will
        appear in the URL query. The second element is the
        human-readable name for the option that will appear
        in the right sidebar.
        """
        return (
            ('80s', _('in the eighties')),
            ('90s', _('in the nineties')),
        )

    def queryset(self, request, queryset):
        """
        Returns the filtered queryset based on the value
        provided in the query string and retrievable via
        `self.value()`.
        """
        # Compare the requested value (either '80s' or '90s')
        # to decide how to filter the queryset.
        if self.value() == '80s':
            return queryset.filter(birthday__gte=date(1980, 1, 1),
                                birthday__lte=date(1989, 12, 31))
        if self.value() == '90s':
            return queryset.filter(birthday__gte=date(1990, 1, 1),
                                birthday__lte=date(1999, 12, 31))

class PersonAdmin(admin.ModelAdmin):
    list_filter = (DecadeBornListFilter,)

答案 1 :(得分:7)

编辑 - 已指出此方法存在问题,请参阅下文

你可以这样做:

假设你有一个名为Animal的模型,它有一个名为Species的模型的ForeignKey字段。在特定的管理列表中,您希望仅允许在动物过滤器选项中显示某些物种。

首先,在Animal的ModelAdmin中指定一个名为SpeciesFilter的自定义ListFilter:

class AnimalAdmin(ModelAdmin):
    list_filter = (('species', SpeciesFilter),)

然后定义SpeciesFilter:

from django.contrib.admin.filters import RelatedFieldListFilter

class SpeciesFilter(RelatedFieldListFilter):
    def __init__(self, field, request, *args, **kwargs):
        """Get the species you want to limit it to.
        This could be determined by the request,
        But in this example we'll just specify an
        arbitrary species"""
        species = Species.objects.get(name='Tarantula')

        #Limit the choices on the field
        field.rel.limit_choices_to = {'species': species}

        #Let the RelatedFieldListFilter do its magic 
        super(SpeciesFilter, self).__init__(field, request, *args, **kwargs)

应该这样做。

答案 2 :(得分:0)

我必须从db表创建我的查找字段。我创建了如下所示的自定义过滤器类,并仅向登录用户显示相关值并相应地过滤:

class ShiftFilter_Org(admin.SimpleListFilter):
title = 'Organisation'
parameter_name = 'org'

def lookups(self, request, model_admin):
"""Get the organisations you want to limit"""
    qs_org = Organisation.objects.filter(users=request.user)
    list_org = []
    for og in qs_org:
        list_org.append(
            (og.id, og.name)
        )
    return (
        sorted(list_org, key=lambda tp:tp[1])
    )

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

更多访问Getting the most out of Django Admin filters

答案 3 :(得分:0)

我发现了另一种类似于@seddonym的方法,但不会使缓存混乱。它基于this Django code,但使用未公开的方法field_choices,在将来的Django版本中,该方法可能会发生变化。 @seddonym的情况的代码为:

from django.contrib.admin.filters import RelatedFieldListFilter

class SpeciesFilter(RelatedFieldListFilter):
    def field_choices(self, field, request, model_admin):
        return field.get_choices(include_blank=False, limit_choices_to={'name': 'Tarantula'})

或者在我的情况下,工作代码是:

from django.contrib.admin.filters import RelatedFieldListFilter

class UnitFilter(RelatedFieldListFilter):
    def field_choices(self, field, request, model_admin):
        return field.get_choices(include_blank=False, limit_choices_to={'pk__in': request.user.administrated_units.all()})