[英]Build request URL to filter Django queryset multiple times by the same field
我想使用Q
多次过滤同一字段的 Django 查询集,以在该字段中包含/排除具有特定值的记录。
我将使用示例模型来说明我的情况。 假设我有一个带有字段status
的Record
模型。 该字段可以是A
、 B
、 C
三种状态之一。
class Record(models.Model):
STATUS_A = 'A'
STATUS_B = 'B'
STATUS_C = 'C'
SOME_STATUSES = (
(STATUS_A, 'Something A'),
(STATUS_B, 'Something B'),
(STATUS_C, 'Something C'),
)
status = models.CharField(
max_length=1,
choices= SOME_STATUSES,
default= STATUS_A,
)
我有一个DRF ViewSet
,它负责返回Record
对象的过滤查询集。
现在我正在过滤由单个状态设置的查询,所以我的URL
看起来像:
.../?status=A
.../?status=B
.../?status=C
但是说我想通过多个状态过滤查询集:返回状态为A
和B
所有记录。 或者我想返回除状态为C
之外的所有记录。 我不确定在这些情况下如何构建 URL。 我知道在 URL 中复制参数是一种非常糟糕的做法:
.../?status=A&status=B
如何请求A
AND
B
或NOT
C
?
如何处理这些多个值的问题的其余部分不清楚,可能是因为我首先不了解如何构建这样的查询。
通过编写 CustomDjangoFilter 你可以实现这一点。
示例代码
URL 示例和用法
URL : localhost:8000/records/?status_include=A,B
URL : localhost:8000/records/?status_exclude=A
URL : localhost:8000/records/?status_include=A,B,C&status_exclude=D,E,F
代码片段
视图.py
from django_filters.rest_framework import DjangoFilterBackend
from .filters import CustomRecordFilter
class RecordViewSet(viewsets.ModelViewSet):
queryset = Record.objects.all()
serializer_class = RecordSerializer
# django-filter-backend and custom-filter-class
filter_backends = (DjangoFilterBackend, )
filter_class = CustomRecordFilter
过滤器.py
import django_filters
class CustomRecordFilter(django_filters.FilterSet):
status_exclude = django_filters.CharFilter(field_name='status', method='filter_status_exclude')
status_include = django_filters.CharFilter(field_name='status', method='filter_status_include')
def filter_status_include(self, queryset, name, value):
if not value:
return queryset
values = ''.join(value.split(' ')).split(',')
queryset = queryset.filter(status__in=values)
return queryset
def filter_status_exclude(self, queryset, name, value):
if not value:
return queryset
values = ''.join(value.split(' ')).split(',')
# exclude status
queryset = queryset.exclude(status__in=values)
return queryset
class Meta:
model = UserRoleGroup
fields = ('status', 'status_include', 'status_exclude')
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.