Создание настраиваемых фильтров для list_filter в Django Admin

Я хотел бы создать собственные фильтры для django admin вместо обычных 'is_staff' и 'is_superuser'. Я прочитал этот list_filter в документах Django. Пользовательские фильтры работают следующим образом:

from datetime import date

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

class DecadeBornListFilter(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(ModelAdmin):
    list_filter = (DecadeBornListFilter,)

Но я уже сделал пользовательские функции для list_display следующим образом:

def Student_Country(self, obj):
    return '%s' % obj.country
Student_Country.short_description = 'Student-Country'

Возможно ли использовать пользовательские функции для list_display в list_filter вместо того, чтобы писать новую настраиваемую функцию для list_filter? Любые предложения или улучшения приветствуются. Вам нужно руководствоваться этим... Спасибо...

Ответ 1

Вы действительно можете добавить настраиваемые фильтры в админ-фильтры, расширив SimpleListFilter. Например, если вы хотите добавить фильтр континентов для "Африка" в фильтр страны-администратора, использованный выше, вы можете сделать следующее:

В admin.py:

from django.contrib.admin import SimpleListFilter

class CountryFilter(SimpleListFilter):
    title = 'country' # or use _('country') for translated title
    parameter_name = 'country'

    def lookups(self, request, model_admin):
        countries = set([c.country for c in model_admin.model.objects.all()])
        return [(c.id, c.name) for c in countries] + [
          ('AFRICA', 'AFRICA - ALL')]

    def queryset(self, request, queryset):
        if self.value() == 'AFRICA':
            return queryset.filter(country__continent='Africa')
        if self.value():
            return queryset.filter(country__id__exact=self.value())

class CityAdmin(ModelAdmin):
    list_filter = (CountryFilter,)

Ответ 2

Ваш метод list_display возвращает строку, но если я правильно понимаю, что вы хотите сделать, это добавить фильтр, который позволяет выбирать страны студентов, правильно?

Для этого фильтра простых отношений и, фактически, для столбца отображения списка "Студент-страна", вам не нужно создавать собственный класс фильтра или метод отображения пользовательского списка; этого было бы достаточно:

class MyAdmin(admin.ModelAdmin):
    list_display = ('country', )
    list_filter = ('country', )

Способ django делает list_filter, как объяснено в документах, сначала автоматически сопоставляя поля, которые вы предоставляете, к предварительно построенным классам фильтров; эти фильтры включают CharField и ForeignKey.

list_display аналогично автоматизирует совокупность столбца списка изменений, используя поле, переданное путем извлечения связанных объектов и возвращающее значение unicode (такое же, как и в приведенном выше методе).