我是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
)
此代码部分起作用,它返回以下响应。
问题是产品的主键“ id”丢失,我希望响应是这样的:
P.S。如果有人可以增强代码并简化代码,我将不胜感激。
预先感谢
答案 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'])