我在一个视图中构建自定义查询集时遇到问题。
这是我模型的简化版本。我的想法是要跟踪不同商店中不同产品的价格。一个商店可以有多个产品,并且一个产品可以存在于多个商店中。产品可以设置is_special
标记以突出显示它们。选择以下结构是因为它很容易在以后添加新产品和新商店:
class Product(models.Model):
product_id = models.CharField(max_length=255, primary_key=True)
is_special = models.BooleanField(default=False)
product_name = models.CharField(max_length=255, default='', blank=False, null=False)
class ShopProductPrice(models.Model):
product = models.ForeignKey("Product", on_delete=models.CASCADE)
shop = models.ForeignKey("Shop", on_delete=models.CASCADE, related_name="shop_entry")
price = models.FloatField(default=0.0, null=False)
class Shop(models.Model):
name = models.CharField(max_length=255, default='', blank=False, null=False)
location = models.CharField(max_length=255, default='', blank=False, null=False)
这些是我的序列化器:
class ShopProductPriceSerializer(serializers.ModelSerializer):
class Meta:
model = ShopProductPrice
fields = ['product', 'price']
class ShopSerializer(serializers.ModelSerializer):
shopProductPriceData = ShopProductPriceSerializer(many=True, source='shopProductPrice_set')
class Meta:
model = Shop
fields = ['name', 'location', 'shopProductPriceData']
最后,这是我要使用自定义queryset的对应视图:
class ShopViewSet(viewsets.ModelViewSet):
queryset = Shop.objects.all()
serializer_class = ShopSerializer
def get_queryset(self):
queryset = Shop.objects.filter(shopproductpricedata__product__is_special=False)
return queryset
在重要的一点是现在的get_queryset()
功能。这是我实际上正在做的事情,我不知道该如何继续。我想要实现的是使用属性控制,无论特殊产品是否应作为输出的一部分。is_special
这是我期望的输出的两个示例(该属性is_special
实际上未在此处显示,但假定名称方案为“ Special Product XY”的产品已is_special
设置为True
)
我在寻找两个不同的查询集:
1:一个查询集,其中特殊产品隐藏在输出中(也不应包括在shopProductPriceData中没有元素的商店)
[
{
"name": "Shop A",
"location": "New York",
"shopProductPriceData": [
{
"product_name": "Product 2",
"price": 100.0
}
]
},
{
"location": "Shop B",
"date": "Berlin",
"shopProductPriceData": [
{
"product_name": "Product 3",
"price": 100.0
}
]
}
]
2:包含特殊产品的查询集:
[
{
"name": "Shop A",
"location": "New York",
"shopProductPriceData": [
{
"product_name": "Special Product 1",
"price": 5.0
},
{
"product_name": "Product 2",
"price": 100.0
}
]
},
{
"location": "Shop B",
"date": "Berlin",
"shopProductPriceData": [
{
"product_name": "Special Product 1",
"price": 8.0
},
{
"product_name": "Product 3",
"price": 100.0
},
{
"product_name": "Special Product 2",
"price": 100.0
}
]
}
]
但是,使用上面给出的查询集,此方法不起作用:
queryset = Shop.objects.filter(shopproductpricedata__product__is_special=True)
实际上返回所有至少具有一种is_special=True
产品的商店
queryset = Shop.objects.filter(shopproductpricedata__product__is_special=True)
返回具有至少一种is_special=False
产品的所有商店。
在这两种情况下,如果都显示了商店,那么无论是否是特殊产品,也会显示所有嵌套产品。
您能帮我提出一个生成所需输出的自定义查询集生成器吗?
编辑:这是有效的解决方案:
queryset = Shop.objects.filter(shopproductpricedata__product__is_special=False).prefetch_related(
Prefetch('shopProductPrice_set', queryset=ShopProductPrice.objects.filter(product__is_special=False)))
尝试使用,prefetch_related
以便相关字段已被获取,Prefetch
您还可以使用过滤相关对象。
from django.db.models import Prefetch
queryset = Shop.objects.filter(shopproductpricedata__product__is_special=False).prefetch_related(
Prefetch('shopproductpricedata__product', queryset=Product.objects.filter(is_special=False)))
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句