如何在Django REST框架中按浏览器输入的日期范围过滤数据？

嘿，我来帮你搞定浏览器输入日期范围过滤图书数据的需求！结合你现有的代码，有两种常用的实现方式，我都给你列出来：

方法一：使用django-filter库（推荐，灵活扩展性强） #

这个库是Django REST Framework常用的过滤工具，能快速实现各种过滤场景，包括日期范围。

步骤1：安装并配置django-filter #

• 先安装依赖包：

  pip install django-filter
  

• 在项目的settings.py中，把django_filters添加到INSTALLED_APPS：

  INSTALLED_APPS = [
      # 其他已有的app
      'django_filters',
  ]
  

步骤2：定义过滤规则类 #

在你的app目录下（比如和views.py同目录）新建或修改一个filters.py文件，定义图书的过滤规则，其中包含日期范围的过滤：

import django_filters
from .models import Book

class BookFilter(django_filters.FilterSet):
    # 定义起始日期过滤：date字段大于等于start_date
    start_date = django_filters.DateFilter(field_name='date', lookup_expr='gte')
    # 定义结束日期过滤：date字段小于等于end_date
    end_date = django_filters.DateFilter(field_name='date', lookup_expr='lte')

    class Meta:
        model = Book
        # 保留你原来需要过滤的字段，加上新增的日期范围参数
        fields = ['title', 'language', 'authors', 'start_date', 'end_date']

步骤3：修改你的BookView视图 #

更新views.py，引入DjangoFilterBackend并指定过滤类：

from rest_framework import generics, filters
from django_filters.rest_framework import DjangoFilterBackend
from .models import Book
from .serializers import BookSerializers
from .filters import BookFilter  # 导入刚才定义的过滤类

class BookView(generics.ListAPIView):
    queryset = Book.objects.all()
    serializer_class = BookSerializers
    # 把DjangoFilterBackend加入过滤后端
    filter_backends = [filters.SearchFilter, DjangoFilterBackend]
    search_fields = ['title', 'language', 'authors', 'date']
    # 指定我们定义的过滤类
    filterset_class = BookFilter

这样配置后，你就可以在浏览器里通过类似这样的URL来过滤日期范围了：
http://your-domain/api/books/?start_date=2023-01-01&end_date=2023-12-31

方法二：手动处理查询参数（轻量，适合简单场景） #

如果不想引入额外依赖，也可以直接在视图里重写get_queryset方法，手动处理日期参数：

修改你的views.py：

from rest_framework import generics, filters
from .models import Book
from .serializers import BookSerializers

class BookView(generics.ListAPIView):
    queryset = Book.objects.all()
    serializer_class = BookSerializers
    filter_backends = [filters.SearchFilter]
    search_fields = ['title', 'language', 'authors', 'date']

    def get_queryset(self):
        # 先获取基础的查询集
        queryset = super().get_queryset()
        # 从请求参数中获取起始和结束日期
        start_date = self.request.query_params.get('start_date')
        end_date = self.request.query_params.get('end_date')
        
        # 如果有起始日期，过滤出date大于等于该日期的记录
        if start_date:
            queryset = queryset.filter(date__gte=start_date)
        # 如果有结束日期，过滤出date小于等于该日期的记录
        if end_date:
            queryset = queryset.filter(date__lte=end_date)
        
        return queryset

同样，你可以用和方法一一样的URL参数格式来过滤数据。

两种方法各有优劣：方法一更适合后续可能扩展更多过滤规则的场景，方法二则更轻量，不需要额外依赖。你可以根据自己的需求选择~

内容的提问来源于stack exchange，提问作者mitus