Django 查询集优化：使用过滤器对 _set 进行反向查找

Question

我有这些模型，我需要做一些计算并将它们呈现给用户。我渲染了大约 2-3k 行，这导致对数据库执行 4k 次查询（从调试工具栏显示）。有没有办法优化这个？我已经尝试过使用 prefetch_related 但它只是在 4k 之上添加了另一个正在完成的查询..

class Cart(models.Model):
    name = models.CharField(max_length=15)
    user = models.OneToOneField(User)


    def sum_for_this(self, taxtype, tax):
        return self.carttax_set.filter(tax__type__name=taxtype,
                                       tax__tax=tax).aggregate(
            sum=Coalesce(Sum('tax_amount'), Value('0'))
        ).get('sum')


class TaxType(models.Model):
    name = models.CharField(max_length=10)


class Tax(models.Model):
    name = models.CharField(max_length=100)
    type = models.ForeignKey(TaxType)
    tax = models.DecimalField()


class CartTax(models.Model):
    cart = models.ForeignKey(Cart)
    tax = models.ForeignKey(Tax)
    base = models.IntegerField()
    tax_amount = models.IntegerField()

我在模板中做的是：

{% for cart in cartlist %}
{{ cart.sum_for_this }}
{% endfor %}

我试过了，但没有效果：

Cart.objects.prefetch_related('carttax_set').all()

def sum_for_this 方法正在执行所有查询..

Answer 1

尝试使用conditional expressions 进行注释。您的查询将如下所示：

from django.db.models import Q, F, Sum, Case, When, IntegerField
from django.db.models.functions import Coalesce


cartlist = Cart.objects.annotate(
    sum=Coalesce(Sum(Case(
        When(Q(carttax__tax__type__name=taxtype) & Q(carttax__tax__tax=tax), then=F('carttax__tax_amount')),
        output_field=IntegerField()
        )), 0)
    )

在模板中：

{% for cart in cartlist %}
{{ cart.sum }}
{% endfor %}