Пользовательский FilterSet не фильтрует по двум полям одновременно - PullRequest
Новая
       17

Пользовательский FilterSet не фильтрует по двум полям одновременно

1 голос
/ 19 июня 2019

Я написал собственный FilterSet для фильтрации набора запросов по двум полям, но он не работает должным образом, когда он фильтрует по двум полям одновременно.

мой набор фильтров: 

class EventFilter(filters.FilterSet):
    values = None
    default = None

    category = filters.ModelMultipleChoiceFilter(
        queryset=EventCategory.objects.all(),
    )
    interval = filters.CharFilter(
        method='filter_interval'
    )

    class Meta:
        model = Event
        fields = ('category', 'interval')

    def filter_interval(self, queryset, name, value):
        if self.request.query_params.get('current_time'):
            try:
                interval = getattr(self, f'get_{value}_interval')()
                interval = list(map(lambda date: self.to_utc(date), interval))
                return self.queryset.filter(Q(status=Event.STARTED) | (Q(status=Event.NOT_STARTED, start_at__range=interval)))
            except Exception as e:
                pass
        return queryset

APIView: 

class ListEventsAPIView(generics.ListAPIView):
    serializer_class = ListEventsSerializer
    filter_class = EventFilter
    search_fields = 'title',
    filter_backends = filters.SearchFilter, DjangoFilterBackend

    def get_queryset(self):
        return Event.objects.filter(Q(status=Event.STARTED) | (Q(status=Event.NOT_STARTED) & Q(start_at__gte=date)))

Здесь генерируется SQL, когда я пытаюсь фильтровать только по категории: 

SELECT "*" FROM "events" WHERE (("events"."status" = 'started'
OR ("events"."status" = 'not_started' 
AND "events"."start_at" >= '2019-06-19T13:24:26.444183+00:00'::timestamptz)) 
AND "events"."category_id" = 'JNPIZF54n5q')

Когда я фильтрую оба: 

SELECT "*" FROM "events" WHERE (("events"."status" = 'started' 
OR ("events"."status" = 'not_started' AND "events"."start_at" >= '2019-06-19T13:24:26.444183+00:00'::timestamptz)) 
AND ("events"."status" = 'started' OR ("events"."start_at" BETWEEN '2019-06-19T07:16:48.549000+00:00'::timestamptz AND '2019-06-30T20:59:59.000059+00:00'::timestamptz AND "events"."status" = 'not_started')))

1 Ответ

1 голос
/ 19 июня 2019

Ваша проблема в этой строке: 

               return self.queryset.filter(Q(status=Event.STARTED) | (Q(status=Event.NOT_STARTED, start_at__range=interval)))

Вы используете набор запросов из самого класса FilterSet. К этому набору запросов не применены никакие предыдущие фильтры, поэтому, используя его, вы отменяете другой фильтр. Просто удалите self. из этой строки, чтобы использовать набор запросов, передаваемый этой функции в качестве параметра, и все будет работать нормально.

...