如何为Django的Calendar查询集注解ui2字段（无外键）

我有3个Django模型：

1. Calendar模型：包含supplier、subassort字段，示例数据：

supplier  | subassort
----------------------
12345     | subassort1
67890     | subassort2

2. SupplierProductAssortments模型：包含subassort、supplier、ui2_code字段，示例数据：

supplier  | subassort  | ui2_code
---------------------------------
12345     | subassort1 | ui-1
12345     | subassort1 | ui-2
67890     | subassort2 | ui-3

3. UI2模型：包含code、name字段，示例数据：

code | name
-------------
ui-1 | milk
ui-2 | cheese
ui-3 | meat

需要为Calendar的查询集注解ui2字段，现有循环处理逻辑效率低，期望输出每条Calendar记录关联对应的UI2名称列表：

supplier  | subassort  | ui2
---------------------------------------
12345     | subassort1 | [milk, cheese] 
67890     | subassort2 | [meat]

由于需要通过supplier和subassort双字段过滤，无法直接用单字段OuterRef实现，请问如何用Annotate实现该需求？注：模型间无外键关联。

可以通过**嵌套子查询+ArrayAgg**实现双字段匹配的关联聚合，以下是两种可行写法：

写法一：双层子查询关联 #

先通过子查询匹配SupplierProductAssortments，再关联UI2获取名称并聚合：

from django.db.models import OuterRef, Subquery, ArrayAgg, CharField
from django.db.models.functions import Coalesce

# 构建子查询：先匹配当前Calendar记录的supplier和subassort，再关联UI2取名称
ui2_subquery = UI2.objects.filter(
    code=SupplierProductAssortments.objects.filter(
        supplier=OuterRef('supplier'),
        subassort=OuterRef('subassort')
    ).values('ui2_code')
).values_list('name', flat=True)

# 为Calendar查询集注解ui2字段，用Coalesce处理无匹配时返回空列表
annotated_calendars = Calendar.objects.annotate(
    ui2=Coalesce(
        ArrayAgg(Subquery(ui2_subquery), output_field=CharField()),
        []
    )
)

写法二：单分子查询聚合 #

直接在SupplierProductAssortments的子查询中完成与UI2的关联和聚合：

from django.db.models import OuterRef, Subquery, ArrayAgg, F
from django.db.models.functions import Coalesce

# 子查询：匹配双字段后，关联UI2取名称并聚合为列表
ui2_names_subquery = SupplierProductAssortments.objects.filter(
    supplier=OuterRef('supplier'),
    subassort=OuterRef('subassort')
).annotate(
    ui2_name=F('ui2_code__name')  # 无外键仍可通过字段关联写法生成JOIN语句
).values('supplier', 'subassort').annotate(
    ui2_list=ArrayAgg('ui2_name')
).values('ui2_list')

# 注解Calendar查询集
annotated_calendars = Calendar.objects.annotate(
    ui2=Coalesce(Subquery(ui2_names_subquery), [])
)

关键说明 #

• 两种写法均通过OuterRef同时引用Calendar的supplier和subassort字段，实现双字段精准匹配；
• ArrayAgg负责将多条匹配的UI2名称聚合为列表；
• Coalesce用于处理无匹配结果的场景，返回空列表而非NULL，保证字段类型统一。

内容的提问来源于stack exchange，提问作者Альберт Александров