如何给Django的BuildItem查询集标注mandatory_skills对应Skill值总和？

要实现仅统计mandatory_skills中指定ID的Skill值总和，你需要结合**子查询（Subquery）和外部引用（OuterRef）**来关联BuildItem自身的mandatory_skills字段，同时过滤符合条件的Skill记录。

实现代码 #

首先导入所需的查询工具：

from django.db.models import Subquery, OuterRef, Sum, Coalesce
import django.db.models as models

然后构建查询集：

# 构建子查询：筛选当前BuildItem的base关联的Skill中，ID在mandatory_skills里的记录，并求和
skill_sum_subquery = Skill.objects.filter(
    item=OuterRef('base'),  # 关联到当前BuildItem的base字段
    id__in=OuterRef('mandatory_skills')  # 只保留ID在mandatory_skills数组中的Skill
).values('item').annotate(total=Sum('value')).values('total')

# 为BuildItem查询集标注求和结果，用Coalesce处理空值情况
q = BuildItem.objects.annotate(
    sum_mandatory_skill=Coalesce(Subquery(skill_sum_subquery, output_field=models.IntegerField()), 0)
)

代码解释 #

1. 子查询逻辑：

   • OuterRef('base') 引用当前BuildItem实例的base字段，确保只关联该Item下的Skill。
   • id__in=OuterRef('mandatory_skills') 筛选出Skill的ID存在于当前BuildItem的mandatory_skills数组中的记录。
   • 通过values('item').annotate(total=Sum('value'))对符合条件的Skill的value字段求和。

2. 主查询标注：

   • 使用Subquery将子查询的结果注入到主查询的标注字段中。
   • Coalesce用于处理当mandatory_skills为空或无匹配Skill时的情况，返回默认值0，避免出现None。

注意事项 #

• 确保你的Django版本支持Subquery和OuterRef（Django 1.11+ 已支持）。
• 若mandatory_skills包含null值，id__in会自动忽略这些空值，不影响统计结果。

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