在 Django ORM 中过滤关系字段

Question

我有一个模型媒体，它与模型 UserMedia（用户评级）有关系。还有一个与问题相关的称为 UserMatchScore（用户匹配分数）的模型。

在我正在查询媒体表的视图中，在此视图中，有一个选项可以仅获取我的比赛已评分但我尚未评分的媒体。此外，还会根据我和我的匹配项的子集返回平均评分，而不是所有对媒体进行评分的用户。我用注释来做到这一点。

我所做的是过滤媒体表中我未评级但我的匹配已评级的元素，这很简单，但它只完成了一半的工作。返回所有我未评级的媒体，但我的匹配已评级，但关系字段 UserMedia 仍包含所有评级，此未过滤，因此无法计算我的匹配子集的平均评级. 这是我正在描述的查询：

queryset = models.Media.objects
queryset = queryset.filter(
               Q(usermedia__user__id__in=my_matches) & ~Q(usermedia__user=user)
            )

获得预期结果的唯一方法是遍历查询集并过滤 UserMedia 关系的每个元素，但这太慢了，因此必须使用数据库查询来完成。

for el in queryset:                    
    el.usermedia_set.filter(~Q(user=user)).filter(user=my_matches)

有人知道如何用 Django ORM 做到这一点吗？

Answer 1

使用Prefetch，您可以使用预过滤的查询集获取关系的相关对象（对于多对多和反向外键关系）：

queryset = queryset.filter(
               Q(usermedia__user__id__in=my_matches) & ~Q(usermedia__user=user)
            )

prefetch = UserMedia.objects.filter(user_id__in=my_matches).exclude(user=user)
queryset = queryset.prefetch_related(
    Prefetch('usermedia_set', prefetch, to_attr='filtered_usermedia')
)

for el in queryset:
    for usermedia in el.filtered_usermedia:
        # iterate over the filtered usermedia
        # without any additional queries
        calculate_something(usermedia)                    

请注意，与之前对主查询集中的每个对象的额外查询相比，这将导致一个额外的查询用于预取所有相关对象（因此总共有两个查询，无论您提取多少行）。