Django查询集优化:使用过滤器在_set上反向查找

时间:2017-06-28 10:02:33

标签: python django postgresql

我有这些模型,我需要做一些计算并将它们呈现给用户。我渲染大约2-3k行,这导致对DB完成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正在执行所有查询..

1 个答案:

答案 0 :(得分:3)

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

appleId

在模板中:

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)
    )