Django Rest Filterset — как фильтровать в т.ч по вложенному сериализатору?

Question

[Игорь]

Добрый вечер коллеги! Имею такой код:

serializators.py

class CoinCostsSerializer(serializers.ModelSerializer):
    class Meta:
        fields = ('price', 'timestamp',)
        model = CoinCosts
        
class CoinSerializer(serializers.ModelSerializer):
    class Meta:
        fields = ('symbol', 'crr', 'costs')
        model = Coins
    costs = CoinCostsSerializer(source='coincosts_set.all', many=True)

views.py

class CoinCostFilterSet(filters.FilterSet):

    class Meta:
        model = Coins
        fields = {
            'symbol': ['exact'],
        }


class CoinCostViewSet(viewsets.ModelViewSet):

    queryset = Coins.objects.all()
    serializer_class = CoinSerializer
    filter_backends = (filters.DjangoFilterBackend,)
    filterset_class = CoinCostFilterSet

И набирая адрес /CoinCost/?symbol=ZERO например, я имею фильтрацию от Джанго по данной монете.

Но т.к я вложил в сериализатор CoinSerializer цены и время с сериализатора CoinCostsSerializer, я не могу фильтровать в т.ч по периоду времени. А мне это нужно. А именно, По symbol с модели Coins и по Timestamp с модели CoinCossts.

Чтобы я мог набрать /CoinCost/?symbol=ZERO&timestamp_start=2019-12-14T00:00:00&timestamp_end=2019-12-17T00:00:00 и получить фильтрацию не только по монете Zero но и по данному периоду. Можете помочь? Как должен выглядеть мой код для этой задачи?

Пытался эксперементировать с кодом CoinCostFilterSet, но безуспешно.

Спасибо за помощь! Хороших выходных!

Answer 1

[Сергей Тихонов]

class DateTimeGteFilter(Filter):
    def filter(qs, value):
        return qs.prefetch_related(
            Prefetch(
                'coincost_set',
                to_attr='filtered_costs',
                queryset=CoinCost.objects.filter(timestamp__gte=value)
            )
        )

class CoinCostFilterSet(FilterSet):
    timestamp_start = DateTimeGteFilter()
    ...

На простейшем уровне как-то так

Comments

[Игорь]

Спасибо Сергей, пытаюсь использовать, но все ругается на первую строку вашего кода, или, Cannot use None as a query value., если пишу

class DateTimeGteFilter(filters.DateTimeFilter):
    def filter(self, qs, value):

[Сергей Тихонов]

Ну а Вы попробуйте доработать код так, чтобы при передаче None ничего не делалось.

[Игорь]

пробую так

class DateTimeGteFilter(filters.DateTimeFilter):
    def filter(self, qs, value):
        if value != None:
            return qs.prefetch_related(
                Prefetch(
                    'coincosts_set',
                    to_attr='filtered_costs',
                    queryset=CoinCosts.objects.filter(timestamp__gte=value)
                )
            )

а он мне --Expected 'CoinCostFilterSet.timestamp_start' to return a QuerySet, but got a NoneType instead.
-

[Сергей Тихонов]

А что ваш фильтр возвращает когда value=None?

[Игорь]

Сергей Тихонов,
Сергей, это намек на то, что нет else? Пробовал дописать else: pass и try except юзал и тут и в основном фильтре, ошибка на месте. И непонятно вообще почему передает Nontype или None..

[Сергей Тихонов]

return qs

[Игорь]

Сергей Тихонов, тааак, спасибо, пока вроде никаких ошибок и 200 страница. Но не фильтруются timestamp. Пробую так /CoinCost/?symbol=ZERO&timestamp_start=2020-04-09&timestamp_start=2020-04-10 и все равно выдаются все записи с любыми timestamp значениями

[Сергей Тихонов]

Дебагером пройдитесь.

[Игорь]

Сергей Тихонов, да вот ошибок нет нигде, и не хочет фильтровать..

class CoinCostFilterSet(filters.FilterSet):
    timestamp_start = DateTimeGteFilter()
    
    class Meta:
        model = Coins
        fields = {
            'symbol': ['exact'],
            #'timestamp_start': ['gt', 'lt'],
        }

тут вообще нужно поле 'timestamp_start': ['gt', 'lt'], ? И с ним и без него тестирую.. Но судя по всему дело непонятно в чем..

[Игорь]

Сергей Тихонов, Здравствуйте, больше решения проблемы тут нет? Вариант нерабочий получается, уже двое суток пробую его поправить

[Сергей Тихонов]

Нерабочий код в студию.

[Игорь]

Сергей Тихонов,

class DateTimeGteFilter(filters.IsoDateTimeFromToRangeFilter):
    def filter(self, qs, value):
        if value != None:
            return qs.prefetch_related(
                Prefetch(
                    'coincosts_set',
                    to_attr='filtered_costs',
                    queryset=CoinCosts.objects.filter(timestamp__gte=value)
                )
            )
        else:
            return qs
    
class CoinCostFilterSet(filters.FilterSet):
    timestamp_start = DateTimeGteFilter()
    
    class Meta:
        model = Coins
        fields = {
            'symbol': ['exact'],
            #'timestamp_start': ['gt', 'lt'],
        }


class CoinCostViewSet(viewsets.ModelViewSet):

    queryset = Coins.objects.all()
    serializer_class = CoinSerializer
    filter_backends = (filters.DjangoFilterBackend,)
    filterset_class = CoinCostFilterSet

А сериализаторы, как в стартпосте.

[Сергей Тихонов]

CoinCostsSerializer(source='coincosts_set.all', many=True)

Prefetch(
                    'coincosts_set',
                    to_attr='filtered_costs',
                    queryset=CoinCosts.objects.filter(timestamp__gte=value)
                )

Вы добавляете в один атрибут а читаете из другого.

[Игорь]

так выдает ошибку name 'value' is not defined

[Сергей Тихонов]

Игорь, а вы задачи исключительно копированием кусков кода из интернета решаете? Не пробовали почитать, что вам пишут и подумать своей головой?

[Игорь]

Сергей Тихонов, Спасибо, понял. Работает старт. А как присобачить еще и вторую дату? Дабы в диапазоне было.
Неужто создавать второй DateTimeGteFilter ?

[Игорь]

Сергей Тихонов, Сергей, здравствуйте, так как можно сделать и для END дату?