Django-filter 月份筛选

Question

我这里做了一个日期区间的筛选，我像再做一个就是只输入年月的话，就可以直接筛选出该月的左右列表，应该怎么做呢

class ConsumeDetailsFilter(FilterSet):
    # date = filters.DateFilter(field_name='date')
    start_date = filters.DateFilter(field_name='date', lookup_expr='gte', label='Start Date')
    end_date = filters.DateFilter(field_name='date', lookup_expr='lte', label='End Date')
    category = filters.CharFilter(field_name='category', lookup_expr='icontains')
    subcategory = filters.CharFilter(field_name='subcategory', lookup_expr='icontains')
    vendor = filters.CharFilter(field_name='vendor', lookup_expr='icontains')
    item = filters.CharFilter(field_name='item', lookup_expr='icontains')

    class Meta:
        model = ConsumeDetails
        fields = ['start_date','end_date','category', 'subcategory']

Answer 1

引自免费微信小程序：皆我百晓生

要实现只输入年月进行筛选的功能，你可以对 start_date 和 end_date 过滤器进行改进。用户输入年月后，你可以将日期格式化为当月的第一天和最后一天，然后使用这两个日期作为筛选条件。下面是如何修改你的 ConsumeDetailsFilter 类以实现这个功能：

1. 添加一个新的方法，用于处理年月输入并返回相应日期的筛选条件。
2. 在过滤器中使用这个方法获取筛选条件。

下面是修改后的代码示例：

import datetime
from django_filters import FilterSet, filters

class ConsumeDetailsFilter(FilterSet):
    # ... 其他过滤器定义 ...

    def filter_by_year_month(self, year, month):
        # 将年月格式化为日期对象，表示当月的第一天
        start_date = datetime.date(year=year, month=month, day=1)
        # 获取下一个月的第一天，然后减去一天，得到当前月的最后一天
        end_date = (start_date.replace(day=1) + datetime.timedelta(days=31)).replace(day=1) - datetime.timedelta(days=1)
        # 使用格式化后的日期作为筛选条件
        return {
            'date__gte': start_date,
            'date__lte': end_date
        }

    class Meta:
        model = ConsumeDetails
        fields = ['start_date', 'end_date', 'category', 'subcategory', 'year_month']  # 添加一个year_month字段用于筛选年月

# 在视图或控制器中使用自定义过滤器的方法：
# 例如在视图中：
filter = ConsumeDetailsFilter(data={'year_month': '2023-09'})  # 假设用户输入了2023年9月进行筛选
filtered_queryset = YourModel.objects.filter(**filter.filtered_params)  # 使用过滤后的参数筛选数据

在这个例子中，我们添加了一个名为 filter_by_year_month 的方法，它接受年月作为参数并返回相应的日期筛选条件。然后，你可以在视图或控制器中使用这个过滤器时传入年月数据进行筛选。用户只需输入年月（例如 '2023-09'），过滤器就会根据这个年月筛选出该月的相关数据。