时间戳

首页 > 解决方案 > 如何为 django_filters DateRangeFilter 函数指定自定义选项？

django - 如何为 django_filters DateRangeFilter 函数指定自定义选项？

问题描述

我正在使用 django_filters DateRangeFilter() 函数。我的过滤器类如下所示：

class ListingFilter(django_filters.FilterSet):
    ship_docking = django_filters.DateRangeFilter()
    class Meta:
        model = Listing
        fields = ['orig', 'dest', 'ship_sailing', 'ship_docking']

我想更改范围的选择 - 默认选择是今天、昨天、过去 7 天等。如何指定不同的选择列表？

谢谢和问候......保罗（Django新手）

标签： djangodjango-filters

解决方案

查看django-filter和的源代码DateRangeFilter，我们可以看到__init__它检查是否已传入选项集和过滤器集，如果它们为空，则使用默认值。

class DateRangeFilter(ChoiceFilter):
    choices = [
            ('today', _('Today')),
            ('yesterday', _('Yesterday')),
            ('week', _('Past 7 days')),
            ('month', _('This month')),
            ('year', _('This year')),
        ]
    
        filters = {
            'today': lambda qs, name: qs.filter(**{
                '%s__year' % name: now().year,
                '%s__month' % name: now().month,
                '%s__day' % name: now().day
            }),
            'yesterday': lambda qs, name: qs.filter(**{
                '%s__year' % name: (now() - timedelta(days=1)).year,
                '%s__month' % name: (now() - timedelta(days=1)).month,
                '%s__day' % name: (now() - timedelta(days=1)).day,
            }),
            'week': lambda qs, name: qs.filter(**{
                '%s__gte' % name: _truncate(now() - timedelta(days=7)),
                '%s__lt' % name: _truncate(now() + timedelta(days=1)),
            }),
            'month': lambda qs, name: qs.filter(**{
                '%s__year' % name: now().year,
                '%s__month' % name: now().month
            }),
            'year': lambda qs, name: qs.filter(**{
                '%s__year' % name: now().year,
            }),
        }
    
        def __init__(self, choices=None, filters=None, *args, **kwargs):
            if choices is not None:
                self.choices = choices
            if filters is not None:
                self.filters = filters

您所要做的就是为您的选择创建一个集合列表，然后复制自定义过滤器的格式，然后：

class ListingFilter(django_filters.FilterSet):
    // Put custom choices and filter variables here
    choices = [...]
    filters = { ... }
    ship_docking = django_filters.DateRangeFilter(choices=choices, filters=filters)
    class Meta:
        model = Listing
        fields = ['orig', 'dest', 'ship_sailing', 'ship_docking']

推荐阅读