Как переопределить запрос, задающий фильтры в list_filter?

Ответ 1

Смотрите ModelAdmin.queryset и ModelAdmin.formfield_for_foreignkey. Из документов:

Метод queryset в ModelAdmin возвращает 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)

Метод formfield_for_foreignkey в ModelAdmin позволяет переопределить поле формы по умолчанию для поля внешних ключей. Например, чтобы вернуть подмножество объектов для этого поля внешнего ключа на основе пользователя:

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, чтобы отображать только автомобили, принадлежащие экземпляру пользователя.

[обновление]

Извините, я не смог прочитать часть "фильтр". В Django >= 1.4 вы можете передать подкласс django.contrib.admin.SimpleListFilter в список list_filter, который вы можете использовать, чтобы переопределить поиск и методы набора запросов.

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,)

Ответ 2

Изменить - этот метод указал на наличие проблем, см. ниже

Вы можете сделать это следующим образом:

Скажем, у вас есть модель под названием Animal, которая имеет поле ForeignKey для модели под названием "Виды". В конкретном списке администраторов вы хотите разрешить показывать только определенные виды в выборе фильтров для животных.

Сначала укажите пользовательский ListFilter под названием SpeciesFilter в Animal ModelAdmin:

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)

Это должно сделать это.

Ответ 3

Я должен был создать свои поля поиска из таблицы БД. Я создал собственный класс фильтра, как показано ниже, и отображал только связанные значения для входа в систему пользователя и фильтровал соответственно:

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())

Для большего количества посещений Получение максимальной отдачи от фильтров администратора Django

Ответ 4

Я нашел другой метод, похожий на @seddonym, но не вмешивающийся в кеширование. Он основан на этом коде Django, но использует недокументированный метод 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()})