Django按Party外键统计两模型字段差值的ORM查询实现方法

前置导入 #

首先需要导入ORM聚合相关的工具类：

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

查询语句 #

直接基于Party模型做关联聚合查询，返回格式完全匹配需求，转列表即可得到目标结果：

result = list(Party.objects.annotate(
    # 计算对应发票总金额，无发票记录时默认返回0
    total_invoice = Coalesce(Sum('invoice__invoice_amount'), Value(0.0)),
    # 计算对应收款总金额，无收款记录时默认返回0
    total_received = Coalesce(Sum('recievedamount__recieved_amount'), Value(0.0)),
    # 计算应收差值
    total_difference = F('total_invoice') - F('total_received'),
    # 为PartyID设置别名，匹配返回要求的key
    party = F('id')
).values('party', 'total_difference'))

说明 #

• 全程仅触发1次数据库查询，性能足以支撑千级以上的Party数量查询，不会出现N+1性能问题
• Coalesce 用于处理某Party没有对应发票/收款记录的场景，避免空值参与计算得到异常结果
• 关联外键的反向查询名默认是模型名小写，你定义的收款模型名为RecievedAmount，所以反向关联字段为recievedamount__recieved_amount，如果后续你修改了模型命名或自定义了related_name，对应调整关联字段即可

内容的提问来源于stack exchange，提问作者Pulkit Arora