通过POST请求返回列表

时间:2018-10-30 16:08:05

标签: python django django-rest-framework

我是django和python的新手,我想返回所有具有发布请求提供的外键的对象。

这是我的模特:

class Product(models.Model):
    name = models.CharField(max_length=200)
    image = models.CharField(max_length=400)
    price = models.CharField(max_length=200)
    isFavorite = models.BooleanField(default=False)
    category = models.ForeignKey(Category, on_delete=models.CASCADE)

这是我的序列化器:

class ProductSerializer(serializers.ModelSerializer):
    class Meta:
        model = Product
        fields = ('id', 'name', 'image', 'price', 'isFavorite')

这是我在views.py中的代码:

class ListProductsOfCategory(generics.ListAPIView):
    serializer_class = ProductSerializer()

    def post(self, request, *args, **kwargs):
        # catch the category id of the products.
        category_id = request.data.get("category_id", "")
        # check if category id not null
        if not category_id:
            """

            Do action here 

            """
        # check if category with this id exists     
        if not Category.objects.filter(id=category_id).exists():
            """

            Do action here 

            """

        selected_category = Category.objects.get(id=category_id)
        # get products of this provided category.
        products = Product.objects.filter(category=selected_category)
        serialized_products = []
        # serialize to json all product fetched 
        for product in products:
            serializer = ProductSerializer(data={
                "id": product.id,
                "name": product.name,
                "image": product.image,
                "price": product.price,
                "isFavorite": product.isFavorite
            })
            if serializer.is_valid(raise_exception=True):
                serialized_products.append(serializer.data)
            else:
                return
        return Response(
            data=serialized_products
            ,
            status=status.HTTP_200_OK
        )

此代码部分起作用,它返回以下响应。

enter image description here

问题是产品的主键“ id”丢失,我希望响应是这样的:

enter image description here

P.S。如果有人可以增强代码并简化代码,我将不胜感激。

预先感谢

1 个答案:

答案 0 :(得分:1)

您以错误的方式使用了序列化器。您应该传入实例,它将为您提供序列化的数据;传递数据并检查is_valid是用于提交数据,而不是发送数据。另外,您可以使用many=True传入整个查询集:

serialized_products = ProductSerializer(products, many=True)

因此您不需要for循环。

但是实际上DRF甚至可以为您完成所有这些操作,因为您正在使用ListAPIView。您需要做的就是告诉它要使用get_queryset方法中的哪个查询集。因此,您所需要做的就是:

class ListProductsOfCategory(generics.ListAPIView):
    serializer_class = ProductSerializer()

    def get_queryset(self):
        return Product.objects.filter(category__id=self.request.data['category_id'])