DRF 將 URL 參數傳遞給序列化程序

Question

我們可以將 URL 查詢集參數傳遞給序列化程序進行過濾/計算嗎？

這是我的端點： ?start_date=20210312&end_date=20210317

這是我的觀點：

class BrandChartsViewSet(ModelViewSet):
serializer_class = BrandChartsSerializer
pagination_class = BrandChartsPagination

queryset = Brand.objects.all()

def get_queryset(self):
    start_date = self.request.query_params.get('start_date',None)
    end_date = self.request.query_params.get('end_date',None)
    if start_date is None:
        raise InvalidInputError()
    if end_date is None:
        raise InvalidInputError()

    start_date_obj = datetime.strptime(start_date,'%Y%m%d')
    end_date_obj = datetime.strptime(end_date,'%Y%m%d')

    serializer = BrandChartsSerializer(start_date_obj,end_date_obj)
    queryset = Brand.objects.all()

    return queryset

這是我的序列化程序：

class BrandChartsSerializer(serializers.ModelSerializer):
rankings = serializers.SerializerMethodField()
instagram = IGSerializer(allow_null=True)
facebook = FBSerializer(allow_null=True)
hashtags = BrandHashtagSerializer(allow_null=True, many=True)
# rename the json field
brand_uid = serializers.IntegerField(source='id')
brand_name = serializers.CharField(source='name')
origin = serializers.CharField(source="country")

class Meta:
    model = Brand
    fields = [
        "brand_uid",
        "brand_name",
        "origin",
        "rankings",
        "instagram",
        "facebook",
        "hashtags",
    ]

def get_rankings(self, instance):
    rank = instance.rankings.all().filter(created__gte=start_date_obj,
                                          created__lte=end_date_obj
                                          ).order_by('created')
    return BrandRankSerializer(rank, allow_null=True, many=True).data

並且它將在get_rankings中有Unresolved reference 'start_date_obj' ，但是我試圖通過class BrandChartsSerializer(serializers.ModelSerializer,start_date_obj,end_date_obj)將參數傳遞給序列化器

它仍然給出錯誤。

由於我的 function 設計，我認為它無法使用 DRF 過濾器來處理它，我可以將參數傳遞給 serizalizer 進行過濾嗎？ （我需要過濾范圍內的排名 model 並計算總和而不是返回）

Answer 1

在這種情況下，我們可以利用上下文。

嘗試這個。

更新get_rankings方法。

def get_rankings(self, instance):
    start_date_obj = self.context.get("request").query_params.get("start_date")
    end_date_obj = self.context.get("request").query_params.get("end_date")
    rank = instance.rankings.all().filter(created__gte=start_date_obj,
                                          created__lte=end_date_obj
                                          ).order_by('created')
    return BrandRankSerializer(rank, allow_null=True, many=True).data