如何为Django Machine QuerySet标注关联的唯一零件数量？

要统计机器关联的所有唯一零件数量，核心是先获取所有关联该机器的组件（引擎和传动系统的），再统计这些组件关联的零件的唯一数量，避免重复计数。这里有两种简洁的实现方式：

方法一：使用子查询（Subquery）一次性统计唯一零件数 #

from django.db.models import Count, OuterRef, Subquery, Q, IntegerField
from django.db.models.functions import Coalesce

def annotate_machine_query_with_num_parts(machine_query):
    # 子查询：筛选出当前机器所有关联的零件，去重后统计数量
    part_count_subquery = (
        Part.objects.filter(
            # 零件关联的组件属于当前机器的引擎或传动系统
            Q(components__machine_engine=OuterRef("pk")) |
            Q(components__machine_transmission=OuterRef("pk"))
        )
        .annotate(unique_part_count=Count("pk", distinct=True))
        .values("unique_part_count")
    )
    
    # 为QuerySet添加标注，用Coalesce处理无零件的情况（返回0）
    return machine_query.annotate(
        num_parts=Coalesce(Subquery(part_count_subquery, output_field=IntegerField()), 0)
    )

方法二：合并关联路径后直接计数 #

利用Django的Q对象合并两个组件关联，再通过Count(distinct=True)统计唯一零件：

from django.db.models import Count, Q
from django.db.models.functions import Coalesce

def annotate_machine_query_with_num_parts(machine_query):
    return machine_query.annotate(
        num_parts=Coalesce(
            Count(
                "engine_components__parts",
                distinct=True,
                # 筛选条件：零件来自引擎组件或传动组件
                filter=Q(engine_components__parts__isnull=False) | Q(transmission_components__parts__isnull=False)
            ),
            0
        )
    )

为什么之前的方法出错？ #

• 第一个子查询方法逻辑混乱：嵌套子查询没有正确关联机器和零件的层级关系，导致重复计数。
• 第二个直接相加的方法：两个Count分别统计不同组件集的唯一零件，但如果某个零件同时出现在引擎和传动组件中，会被两次计数，相加后结果偏大。

两种新方法都能确保零件只被计数一次，不管它关联多少个组件或属于哪个系统。

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