如何在 django 的过滤器类的后端设置过滤器范围

Question

我正在使用 DRF 构建一个 REST API。我正在使用 django_filters 过滤结果集。在一个api中，我希望用户发送他的坐标（纬度，经度），并在后端创建一个范围为（纬度+2到纬度-2）并返回结果，我不想从用户那里获取范围字段。所以我可以在后端轻松更改范围。

我在过滤器类中创建了两个范围过滤器，它们工作正常，但 url 看起来像这样：/posts/?latitude_min=22&latitude_max=28&longitude_min=80&longitude_max=84。在这里用户负责决定范围。我希望用户只发送纬度和经度，我决定最大和最小范围。

class PostFilter(django_filters.FilterSet):
    latitude = django_filters.RangeFilter(field_name='latitude')
    longitude = django_filters.RangeFilter()

    class Meta:
        model = Post
        fields = ['latitude', 'longitude', 'country', 'state', 'district', ]

Answer 1

我认为你可以像这样使用“MethodFilter”：

from django.db.models import Q

class PostFilter(django_filters.FilterSet):
    latitude = django_filters.MethodFilter(action="filter_latitude")
    longitude = django_filters.MethodFilter(action="filter_longitude")

    class Meta:
        model = Post
        fields = ['latitude', 'longitude', 'country', 'state', 'district']

    def filter_latitude(self, qs, value):
        try:
            return qs.filter(Q(latitude__gte=value-2)&Q(latitude__lte=value+2))
        except BaseException:
            return qs 

    def filter_longitude(self, qs, value):
        try:
            return qs.filter(Q(longitude__gte=value-2)&Q(longitude__lte=value+2))
        except BaseException:
            return qs 

Answer 2

只需几行代码，您就可以编写一个自定义过滤器字段来执行此操作。它可以让您更灵活地在以后添加验证，或者围绕 lat/lng 的细节添加额外的逻辑。

class ApproximateNumberFilter(Filter):
    field_class = FloatField  # use django base float validation

    def __init__(self, under_over=2, **kwargs):
        super().__init__(**kwargs)
        self._under_over = under_over

    def filter(self, qs, value):
        if value in EMPTY_VALUES:
            return qs

        return qs.filter(**{
            f"{self.field_name}__gte": value - self._under_over,
            f"{self.field_name}__lte": value + self._under_over,
        })