循环中的Django prefetch_related，出了什么问题？

Question

我正在尝试减少 Django 应用程序中的查询量，但不知道如何以正确的方式进行。我有一个产品模型，所有产品都有默认价格，现在我有一个客户产品，其中包含一些与客户相关的产品价格

我愿意重新设计 ;)

class Product(models.Model)
    name = models.CharField(max_length=255)
    price = models.DecimalField(default=0, max_digits=12, decimal_places=2)
    ....

    def get_prices(self, customer=None):
        '''
        get prices for the current product based on the price matrix given by bbp
        '''
        if customer:
            prices = self.get_customer_prices()
            if prices:
                self.price = self.customerproduct_set.filter(deleted=0).last().price

        return self

class CustomerProduct(AbstractProductPrice):
    customer = models.ForeignKey(Customer, on_delete=models.CASCADE)
    product = models.ForeignKey(Product, on_delete=models.CASCADE)
    price = models.DecimalField(default=0, max_digits=12, decimal_places=2)
    ....

现在我想要所有具有相关客户价格的产品。

products = Product.objects.all().prefetch_related('customerproduct_set')

    for product in products:
        product.get_prices(customer=1)

预取不“工作”我不明白如何使用它..请帮助。

我的数据库中有 31 个产品，django_debug_tool 产生 35 个查询。

Answer 1

仅为您感兴趣的客户预取CustomerProducts，然后使用该结果。类似的东西：

products = Product.objects.all().prefetch_related(
    models.Prefetch(
        'customerproduct_set',
        queryset=CustomerProduct.objects.filter(customer=customer, deleted=0).order_by('id')
        to_attr='customer_prices'
    )
)

def get_product_prices(product):
    customer_prices = getattr(product, 'customer_prices', None)
    if customer_prices:
        return customer_prices[-1].price
    return product.price


for product in products:
    print get_product_prices(product)