编辑:我更新了代码以反映查询集过滤器中的一个错误,其中我有 user__user_type,已被正确的 userprofile__user_type 替换。
我正在使用 Django 1.4,并且我知道有一个新功能可以为管理员创建自定义 list_filters,它取代了 FilterSpec API。
我已经阅读了无数关于创建自定义管理员 list_filters 的 SO 帖子,但我仍然卡住了。
我的用例是我的用户对象的 list_filter 引用 UserProfile 模型中的一个字段。
所以在models.py中:
class UserProfile(models.Model):
user = models.OneToOneField(User, unique=True)
user_type = models.CharField(max_length=25, choices=USER_TYPES, default='Client')
...
在 admin.py 中:
from django.contrib import admin
from django.contrib.admin import site, ModelAdmin, ChoicesFieldListFilter
from django.contrib.auth.models import User
from django.contrib.auth.admin import UserAdmin
from models import UserProfile
class UserTypeFilter(ChoicesFieldListFilter):
title = 'User Type'
parameter_name = 'user_type'
def lookups(self, request, model_admin):
usertypes = set([c.user_type for c in UserProfile.objects.all()])
return [(c.id, c.user_type) for c in usertypes]
def queryset(self, request, queryset):
if self.value():
return queryset.filter(userprofile__user_type=self.value())
else:
return queryset
class UserAdmin(UserAdmin):
list_filter = ('is_staff', UserTypeFilter)
inlines = (UserProfileInline, )
admin.site.unregister(User)
admin.site.register(User, UserAdmin)
这是我得到的错误:
'UserAdmin.list_filter[1]' is 'UserTypeFilter' which is of type FieldListFilter but is not associated with a field name.
我最初尝试使用 SimpleListFilter,并收到此错误
'unicode' object has no attribute 'id'
这是有道理的,因为我的 user_type 字段是一个选择字段,而选择是 unicode 对象。
文档显示了这个例子:
class PersonAdmin(UserAdmin):
list_filter = ('company__name',)
但这似乎暗示“公司”是用户模型上的一个字段。我是否需要重新定义我的用户模型以包含用于 UserProfile 的 OneToOneField?还是我错过了一些明显的方式来引用用户的个人资料?
谢谢,