针对ForeignKey元素的Django Queryset过滤器

Question

我有一个问题，在我的一个视图中建立一个自定义查询集。

这里是一个简化的我的模型版本。我的想法是，我想跟踪不同商店不同产品的价格。商店可以有多个产品一个产品可以出现在多个商店。产品可以设置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']

最后，这里是相应的视图其中我想使用自定义查询集:

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，但是假设名称方案为"特殊产品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)))

Answer 1

尝试使用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)))