Django一对多关系过滤：全匹配与同实例条件查询实现

问题1 #

原代码行为 #

执行outcome = modelB.objects.filter(modelA__in=[1,2,3])时，返回的是关联的modelA实例中任意一个ID在[1,2,3]范围内的modelB——只要有一个符合条件的关联modelA，对应的modelB就会被筛选出来，不要求所有关联的modelA都在这个列表里。

实现“全部关联modelA实例都在[1,2,3]内”的方案 #

可以通过排除存在无效关联的modelB来实现，具体代码如下：

from django.db.models import Exists, OuterRef

# 定义子查询：查找存在关联modelA不在指定列表中的modelB
invalid_subquery = modelA.objects.filter(b_model=OuterRef('pk'), id__notin=[1,2,3])
# 排除这些包含无效modelA的modelB，剩余的就是所有关联modelA都在列表内的modelB
outcome = modelB.objects.annotate(has_invalid_a=Exists(invalid_subquery)).exclude(has_invalid_a=True)

另一种方案是通过计数对比，验证关联的modelA总数等于符合条件的数量：

from django.db.models import Count, Q, F

outcome = modelB.objects.annotate(
    total_a=Count('modelA'),  # 统计该modelB关联的所有modelA数量
    valid_a=Count('modelA', filter=Q(modelA__id__in=[1,2,3]))  # 统计符合条件的modelA数量
).filter(total_a=F('valid_a'))
# 若需要排除没有任何关联modelA的modelB，可添加total_a__gt=0条件

问题2 #

原代码行为 #

outcome = modelB.objects.filter(modelA__gt=1, modelA__lte=3)默认会匹配存在至少一个modelA满足id>1，且至少一个modelA满足id≤3的modelB，这两个条件可能作用在不同的modelA实例上。

实现“条件作用于同一个modelA实例”的方案 #

需要用子查询来限定条件必须同时满足于同一个关联的modelA，代码如下：

from django.db.models import Exists, OuterRef

# 子查询：查找关联当前modelB且同时满足两个ID条件的modelA
valid_a_subquery = modelA.objects.filter(
    b_model=OuterRef('pk'),
    id__gt=1,
    id__lte=3
)
# 筛选存在这样的modelA的modelB
outcome = modelB.objects.filter(Exists(valid_a_subquery))

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