自定义 list_filter 结果

Question

我想限制 django "list_filter" 在 Django admin 中显示的结果。

根据我下面的模型，我只希望它显示类类别中的类别（在过滤器中）并包含层整数“1”。

这可能吗？

class Category(models.Model):
    category_name = models.CharField(max_length=100, blank=False)
    tier = models.IntegerField(blank=false)
    def __str__(self):
        return self.category_name

class Product(models.Model):
    name = models.CharField ("Name", max_length=400)
    link = models.URLField("Link", max_length=900)
    store = models.ForeignKey(Store)
    category = models.ManyToManyField(Category, blank=True)

在管理员中，我有：

list_filter = ('category')

我认为我需要以某种方式合并它？

category=Category.objects.get(tier__iexact='1')

Answer 1

这样的事情应该可以工作：

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


class CategoryListFilter(admin.SimpleListFilter):
    title = _('Category')
    parameter_name = 'category'

    def lookups(self, request, model_admin):
        categories = Category.objects.filter(tier=1)
        for obj in categories:
            yield (str(obj.pk), smart_text(obj))

    def queryset(self, request, queryset):

        if self.value():
            return queryset.filter(category__id=self.value())


class ProductAdmin(admin.ModelAdmin):
    list_filter = (CategoryListFilter,)

文档：https://docs.djangoproject.com/en/dev/ref/contrib/admin/#django.contrib.admin.ModelAdmin.list_filter