Django admin添加自定义过滤器

时间:2018-02-05 10:14:32

标签: python django python-2.7 django-admin django-admin-filters

我正在使用django 1.10并且我需要显示数据并根据来自不同模型的值创建过滤器(其中有一个外键引用我在管理模板上使用的模型) 这些是我的2个型号: 这个用于生成模板:

class Job(models.Model):
    company = models.ForeignKey(Company)
    title = models.CharField(max_length=100, blank=False)
    description = models.TextField(blank=False, default='')
    store = models.CharField(max_length=100, blank=True, default='')
    phone_number = models.CharField(max_length=60, null=True, blank=True)

这是另一个持有我的第一个外键引用的引用:

class JobAdDuration(models.Model):
    job = models.ForeignKey(Job)
    ad_activated = models.DateTimeField(auto_now_add=True)
    ad_finished = models.DateTimeField(blank=True, null=True)

在我的模板中,我能够显示(最新的)开始和结束时间

def start_date(self,obj):
    if JobAdDuration.objects.filter(job=obj.id).exists():
        tempad = JobAdDuration.objects.filter(job=obj).order_by("-id")[0]
        return tempad.ad_activated

然后我只是在list_display中调用它,这样可以正常工作。 但是,我无法使用这些条件设置过滤器字段。

如果我只是将它添加到我的list_filter然后我得到一个错误,我的模型中没有这样的字段是真的(因为那个是在另一个引用我的作业表的表中)。所以我想知道解决这个问题的正确方法是什么?我是否需要为过滤器本身创建另一个函数,但即便如此我也不确定如何在list_filter中调用它。

这是我的Django管理页面的片段。

class JobAdmin(admin.OSMGeoAdmin, ImportExportModelAdmin):
    inlines = [
    ]

    readonly_fields = ( 'id', "start_date", )

    raw_id_fields = ("company",)

    list_filter = (('JobAdDuration__ad_activated', DateRangeFilter), 'recruitment', 'active', 'deleted', 'position', ('created', DateRangeFilter), 'town')
    search_fields = ('title', 'description', 'company__name', 'id', 'phone_number', 'town')
    list_display = ('title', 'id', 'description', 'active', 'transaction_number', 'company', 'get_position', 'town','created', 'expires', 'views', 'recruitment', 'recruits', 'paid', 'deleted', "start_date", "end_Date", "ad_consultant")


    def start_date(self,obj):
        if JobAdDuration.objects.filter(job=obj.id).exists():
            tempad = JobAdDuration.objects.filter(job=obj).order_by("-id")[0]
            return tempad.ad_activated

编辑: 在此期间,我试图用一个简单的列表过滤器解决它,但我无法让它工作。我想在2个输入字段中放置一个日历(比如默认的DateRangeFilter)来表示开始和结束时间,然后根据这些值返回数据。这是我的简单过滤器的“原型”功能,它可以工作,但它返回硬编码数据。

class StartTimeFilter(SimpleListFilter):
    title = ('Start date')
    parameter_name = 'ad_finished'

    def lookups(self, request, model_admin):
       #return JobAdDuration.objects.values_list("ad_finished")
       return (
       ('startDate', 'stest1'),
       ('startDate1', 'test2')
       )

    def queryset(self, request, queryset):
        if not self.value():
            return queryset


    assigned = JobAdDuration.objects.filter(ad_finished__range=(datetime.now() - timedelta(minutes=45000), datetime.now()))
    allJobs = Job.objects.filter(pk__in=[current.job.id for current in assigned])
    return allJobs

3 个答案:

答案 0 :(得分:9)

我会使用自定义的FieldListFilter,因为它允许根据您的要求将过滤器绑定到不同的模型字段。

接下来我们实际做了什么来实现这样的过滤器:

  • 构建lookup_kwargs gte和lte并将其指定为expected_parameters
  • 定义选项以返回空列表,否则NotImplementedError
  • 创建表单以关注字段验证
  • 创建仅输出表单的自定义模板,例如{{spec.form}}
  • 如果表单有效,则将其清除数据,过滤掉Nones并过滤查询集,否则执行错误操作(在下面的代码中出现错误)

过滤代码:

class StartTimeFilter(admin.filters.FieldListFilter):
    # custom template which just outputs form, e.g. {{spec.form}}
    template = 'start_time_filter.html'

    def __init__(self, *args, **kwargs):
        field_path = kwargs['field_path']
        self.lookup_kwarg_since = '%s__gte' % field_path
        self.lookup_kwarg_upto = '%s__lte' % field_path
        super(StartTimeFilter, self).__init__(*args, **kwargs)
        self.form = StartTimeForm(data=self.used_parameters, field_name=field_path)

    def expected_parameters(self):
        return [self.lookup_kwarg_since, self.lookup_kwarg_upto]

    # no predefined choices
    def choices(self, cl):
        return []

    def queryset(self, request, queryset):
        if self.form.is_valid():
            filter_params = {
                p: self.form.cleaned_data.get(p) for p in self.expected_parameters()
                if self.form.cleaned_data.get(p) is not None
            }
            return queryset.filter(**filter_params)
        else:
            return queryset

表格可以简单如下:

class StartTimeForm(forms.Form):

    def __init__(self, *args, **kwargs):
        self.field_name = kwargs.pop('field_name')
        super(StartTimeForm, self).__init__(*args, **kwargs)
        self.fields['%s__gte' % self.field_name] = forms.DateField()
        self.fields['%s__lte' % self.field_name] = forms.DateField()

答案 1 :(得分:5)

这不是您要求的,但您可以在JobAdDuration modelAdmin上设置过滤器。这样,您就可以根据ad_activatedad_finished字段过滤相应的作业。我添加了job字段的链接,因此您可以直接点击它以便于导航。

为了使它成为日期html5过滤器,我使用了django-admin-rangefilter库。

from django.urls import reverse
from django.contrib import admin
from .models import Job, JobAdDuration
from django.utils.html import format_html
from rangefilter.filter import DateRangeFilter


@admin.register(JobAdDuration)
class JobAdDurationAdmin(admin.ModelAdmin):

    list_filter = (('ad_activated', DateRangeFilter), ('ad_finished', DateRangeFilter))
    list_display = ('id', 'job_link', 'ad_activated', 'ad_finished')

    def job_link(self, obj):
        return format_html('<a href="{}">{}</a>', reverse('admin:job_job_change', args=[obj.job.id]), obj.job.title)
    job_link.short_description = 'Job'

如果你确实想要使用现有路线(在JobAdmin内进行过滤),那么事情会变得非常复杂。

答案 2 :(得分:3)

我最近遇到了类似的问题,我需要根据另一个模型的值过滤数据。这可以使用SimpleListFilter完成。您只需要在查找和查询集功能中进行一些调整。我建议你安装django调试工具栏,这样你就可以知道django在内部执行了什么sql查询。

#import your corresponding models first

class StartTimeFilter(SimpleListFilter):
title = ('Start date')
parameter_name = 'ad_finished'

  def lookups(self, request, model_admin):

   data = []
   qs = JobAdDuration.objects.filter()   # Note : if you do not have distinct values of ad_activated apply distinct filter here to only get distinct values
   print qs
   for c in qs:
       data.append([c.ad_activated, c.ad_activated])  # The first c.activated is the queryset condition your filter will execute on your Job model to filter data ... and second c.ad_activated is the data that will be displayed in dropdown in StartTimeFilter
   return data

  def queryset(self, request, queryset):
     if self.value():
       assigned = JobAdDuration.objects.filter(ad_activated__exact = self.value())  # add your custom filter function based on your requirement
       return Job.objects.filter(pk__in=[current.job.id for current in assigned])
     else:
       return queryset

和list_filter

list_filter = (StartTimeFilter) # no quotes else it will search for a field in the model 'job'.