python - Django 查询集优化 : Reverse lookup on _set with filter

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 %}