Django筛选含指定标签的站点并保留所有关联标签

问题原因 #

你当前的查询使用filter(tags__key__in=['type-1', 'type-2'])时，Django会执行内连接过滤关联的SiteTag和Tag记录，只保留与指定标签关联的条目。后续的ArrayAgg只能基于这些过滤后的记录聚合，因此无法获取站点的其他标签。

解决方案 #

方法一：使用Exists子查询（推荐） #

通过Exists子查询判断站点是否包含指定标签，主查询不直接过滤关联的标签表，确保聚合时能获取站点所有标签：

from django.db.models import Exists, OuterRef, ArrayAgg

# 定义子查询：匹配包含目标标签的站点关联记录
target_tag_subquery = SiteTag.objects.filter(
    site=OuterRef('pk'),
    tag__key__in=['type-1', 'type-2']
)

# 筛选符合条件的站点，并聚合所有标签
sites = Site.objects.filter(
    Exists(target_tag_subquery)
).annotate(
    tags=ArrayAgg("tags__key", distinct=True)
)

方法二：先获取符合条件的站点ID再查询 #

分两次查询：先拿到包含指定标签的站点ID列表，再基于ID查询站点并聚合所有标签：

from django.db.models import ArrayAgg

# 第一步：获取包含指定标签的站点ID集合
valid_site_ids = Site.objects.filter(
    tags__key__in=['type-1', 'type-2']
).values_list('pk', flat=True).distinct()

# 第二步：查询这些站点，聚合所有标签
sites = Site.objects.filter(pk__in=valid_site_ids).annotate(
    tags=ArrayAgg("tags__key", distinct=True)
)

两种方法都能实现：筛选出包含type-1或type-2标签的站点，同时在tags聚合字段中保留站点的所有标签（包括type-3这类不在筛选条件中的标签）。

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