问ForeignKey元素的Django Queryset滤波器
EN

Stack Overflow用户

提问于 2021-02-09 05:09:42

回答 1查看 35关注 0票数 0

在我的视图中构建自定义查询集时，我遇到了问题。

下面是我的模型的简化版本。我的想法是，我想跟踪不同商店的不同产品的价格。商店可以有多个产品，产品可以出现在多个商店中。产品可以设置一个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实际上不在这里显示，但假设名称为的产品名称为“特殊产品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生成器来产生我想要的输出吗？

编辑:这是工作解决方案：

queryset = Shop.objects.filter(shopproductpricedata__product__is_special=False).prefetch_related(
     Prefetch('shopProductPrice_set', queryset=ShopProductPrice.objects.filter(product__is_special=False)))

django

django-views

django-queryset

语音识别特惠，低至14.9元！

提供业界非常具有性价比的语音识别服务，超高识别准确率，适用多场景

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-02-09 05:51:23

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