升级到Django 3.2后ArrayAgg字段注解的空数组overlap/contained_by查询报错

1. Django 3.2版本对ArrayAgg聚合的输出类型推断存在缺陷：当你添加过滤条件后无匹配记录、ArrayAgg返回空数组时，Django无法根据空值推断该字段对应的数据库数组类型，执行__overlap、__contained_by等数组专属查询操作时，就会抛出Cannot resolve expression type, unknown output_field错误。
2. 最初返回[None]的原因是，当Contact没有关联的RegisteredStaffContact记录时，左连接查询对应的staff_user字段值为NULL，ArrayAgg默认会将NULL值聚合到结果中，因此出现带None的数组。

第一步：为ArrayAgg显式指定输出类型与默认值 #

显式声明output_field告知Django该聚合字段的类型，同时添加default参数指定无匹配记录时返回空数组，避免NULL值和类型推断问题。
需要先导入依赖：

from django.contrib.postgres.fields import ArrayField
from django.db import models

修改注解部分的代码：

annotated_share_requests = cls._default_manager\
    .annotate(
        registered_staff_ids=ArrayAgg(
            'contact__registered_staffing__staff_user',
            distinct=True,
            # 过滤掉空值避免None进入聚合结果
            filter=models.Q(contact__registered_staffing__staff_user__isnull=False),
            # 显式指定输出为整数数组
            output_field=ArrayField(models.IntegerField()),
            # 无匹配记录时默认返回空数组
            default=list,
        )
    )\
    .annotate(
        primary_registered_staff_ids=ArrayAgg(
            'contact__registered_staffing__staff_user',
            filter=models.Q(contact__registered_staffing__is_primary=True, contact__registered_staffing__staff_user__isnull=False),
            distinct=True,
            output_field=ArrayField(models.IntegerField()),
            default=list,
        )
    )

第二步：优化过滤逻辑 #

添加default=list后聚合字段永远不会返回NULL，可以删除__isnull=True的判断逻辑；同时提前判断user_principals_ids是否为空，避免空数组参与查询导致的额外异常：

# 基础过滤条件：当前用户在主负责人/普通负责人列表中
filter_conditions = (
    models.Q(primary_registered_staff_ids__contains=[user.id])
    | models.Q(registered_staff_ids__contains=[user.id])
)
# 仅当用户有负责上级时，添加上级相关的过滤条件
if user_principals_ids:
    filter_conditions |= (
        models.Q(primary_registered_staff_ids__contained_by=user_principals_ids)
        | models.Q(registered_staff_ids__overlap=user_principals_ids)
    )
filtered_shared_requests = annotated_share_requests.filter(filter_conditions)

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