使用Django SimpleListFilter的自定义方法

时间:2013-08-16 10:42:11

标签: django function django-models filter admin

本质上,我的问题是以下设置是在管理页面中返回整个查询集,而不管我过滤的是什么。

models.py

from django.db import models

class Booking(models.model):
    client = models.ForeignKey(Client)
    reference = models.CharField(max_length=100)
    ....

    def __unicode__(self):
        return self.reference

class Client(models.Model):
    name = models.CharField(max_length=100)
    email = models.EmailField(null=True, blank=True)
    ...

    def __unicode__(self):
        return self.name  

    def total_bookings(self):
        return len(self.booking_set.all())

    def bookings0(self):
        if self.total_bookings() == 0:
            return True

    def bookings1(self):
        if self.total_bookings() == 1:
            return True

    def bookings2(self):
        if self.total_bookings() == 2:
            return True

    def bookings3plus(self):
        if self.total_bookings() > 2:
            return True

...

admin.py

from django.contrib import admin
from django.contrib.admin import SimpleListFilter
from django.utils.translation import ugettext_lazy as _
from clakes.models import Client, Booking ...

class BookingAdmin(admin.ModelAdmin):
    ...

class NumBookingListFilter(SimpleListFilter):
    title = _('Number of Bookings')
    parameter_name = 'numofbooks'

    def lookups(self, request, model_admin):
        return (
            ('no', _('No Bookings')),
            ('1', _('One')),
            ('2', _('Two')),
            ('3plus', _('Three or more')),
        )

    def queryset(self, request, queryset):
        if self.value() == 'no':
            return [x for x in queryset if x.bookings0()]
        if self.value() == '1':
            return [x for x in queryset if x.bookings1()]
        if self.value() == '2':
            return [x for x in queryset if x.bookings2()]
        if self.value() == '3plus':
            return [x for x in queryset if x.bookings3plus()]

class ClientAdmin(admin.ModelAdmin):
    list_display = ('name', 'email', 'total_bookings', ...)
    search_fields = ['name']
    list_filter = (NumBookingListFilter,)
    ...

...
admin.site.register(Client, ClientAdmin)
admin.site.register(Booking, BookingAdmin)

有人可以向我解释我哪里出错了。过滤器按预期显示在“客户端”管理页面上,但是当选择一个选项时,过滤器不起作用 - 网址将更改为http:... / client /?e = 1,但未按预期进行过滤。我的列表理解在shell中有效 - 例如[x for queryset中的x,如果x.bookings1()]返回正确的客户列表。

我很乐意考虑使用经理或其他方法的不同方法。无论谁善意回答,请不要仅仅指出我的文档,因为我花了几个小时现在阅读并重新阅读它,所以如果我错过了这一点,那是由于缺乏理解而不是努力。

1 个答案:

答案 0 :(得分:4)

所以,可能有更简洁的方式来处理整个事情,但以下对我有用:

admin.py

if self.value() == 'no':
        l = [x.id for x in queryset if x.bookings0()]
        return queryset.filter(pk__in=l)
if self.value() == '1':
        l = [x.id for x in queryset if x.bookings1()]
        return queryset.filter(pk__in=l)
...etc...

我希望这会在某些时候帮助某人。