使用 Django ORM 注释查询以计算 2 个表值的总和

Question

我有两张桌子

Class Billing(models.Model):
 id=models.AutoField(primary_key=True)
 .....
 #Some more fields
 ....


Class BillInfo(models.Model):
  id=models.AutoField(primary_key=True)
  billing=models.ForeignKey(Billing)
  testId=models.ForeignKey(AllTests)
  costOfTest=models.IntegerField(default=0)
  concession=models.IntegerField(default=0)

这里BillInfo 是垂直表，即一个Billing 有多个BillInfo。在这里，我想计算单个Billing 的Sum(costOfTest - concession)。

我可以使用单个查询来实现吗？

需要帮助，提前致谢。

Answer 1

你可以这样写：

from django.db.models import F, Sum

Billing.objects.annotate(
    the_sum=Sum(F('billinfo__costOfTest') - F('billinfo__concession'))
)

这里QuerySet 中的每个Billing 对象都将有一个额外的属性.the_sum，它是所有costOfTests 减去所有相关BillingInfo 的concession 的总和对象。

计算此值的 SQL 查询大致如下：

SELECT billing.*
       SUM(billinginfo.costOfTest - billinginfo.concession) AS the_sum
FROM billing
LEFT OUTER JOIN billinginfo ON billinginfo.billing_id = billing.id
GROUP BY billing.id

因此，当您“实现”查询时，查询将在一次调用中获得所有Billing 对象的总和。

对于没有任何相关BillingInfo的Billing对象，the_sum属性将为None，我们可以通过使用Coalesce [Django-doc]函数来避免这种情况：

from django.db.models import F, Sum, Value
from django.db.models.functions import Coalesce

Billing.objects.annotate(
    the_sum=Coalesce(
        Sum(F('billinfo__costOfTest') - F('billinfo__concession')),
        Value(0)
    )
)