Django 如何为Document查询集注解关联对象聚合计算的total_price字段

实现方案 #

可以在查询集层面直接实现该需求，通过带过滤条件的聚合注解即可完成，无需复杂的预取或嵌套子查询操作。 #

前置修正（原有代码笔误）

先修正现有代码的几处小问题，避免后续运行报错：

• Document类的total_price方法缺少self参数，方法内引用的DocumentLine应为实际模型类Line
• LineQuerySet.total_price方法内调用的with_total_prices多了后缀s，应为with_total_price

核心实现（Django 3.2+ 推荐方案）

Django 3.2及以上版本支持聚合函数的filter参数，可以直接在注解中完成过滤+求和逻辑：

# 导入依赖
from django.db.models import Sum, F, Q, DecimalField, Value
from django.db.models.functions import Coalesce

# 构造带总价注解的Document查询集
document_qs = Document.objects.annotate(
    total_price=Coalesce(
        Sum(
            F('sections__lines__quantity') * F('sections__lines__price'),
            filter=Q(
                sections__deleted=False,
                sections__lines__deleted=False
            ),
            output_field=DecimalField()
        ),
        Value(0, output_field=DecimalField())
    )
)

注解生成的total_price字段可以直接用于后续的筛选、排序操作，示例：

# 筛选总价大于100的单据，按总价倒序排列
filtered_qs = document_qs.filter(total_price__gt=100).order_by('-total_price')

逻辑和你原有单实例计算逻辑完全一致：自动过滤已删除的Section和Line，无有效行项目时总价默认为0。 #

兼容低版本Django方案

如果使用Django <3.2版本，不支持聚合的filter参数，可以用子查询实现相同效果：

from django.db.models import Subquery, OuterRef

# 构造子查询计算每个Document的总价
line_total_subq = Line.objects.filter(
    section__document_id=OuterRef('pk'),
    section__deleted=False,
    deleted=False
).annotate(
    line_total=F('quantity') * F('price')
).values('section__document_id').annotate(
    total=Sum('line_total')
).values('total')

# 注解到Document查询集
document_qs = Document.objects.annotate(
    total_price=Coalesce(Subquery(line_total_subq), Value(0, output_field=DecimalField()))
)

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