Django 过滤一个预取的查询集，返回一个空列表

问题描述

在我的应用程序中，我有一些用户和组，它们通过 GroupMember 以多对多关系相关联。

class User(models.Model):
    ...

class Group(models.Model):
    ...
    users = models.ManyToManyField(User,through='groups.GroupMember',related_name='groups')


class GroupMember(models.Model):
    ...
    user = models.ForeignKey(User,related_name='group_members',on_delete=models.CASCADE)
    group = models.ForeignKey(Group,related_name='members',on_delete=models.CASCADE)

现在我正在尝试构建一个端点，该端点检索您的用户所在的任何一个组中的用户实例列表。因此，如果用户 A 和 B 在 group1 中，则用户 A 和 C 在 group2 中，并且 C 和 D 在 group3 中，检索用户，因为 A 将返回 B 和 C（因为它们与 A 处于组中）。

过滤器部分可以通过多种方式完成，但我遇到的问题是如何显示哪些组是导致这种关系的原因。我尝试将 prefetch_related 与自定义过滤器一起使用，以仅获取其中包含用户 A 的组，如下所示：

User.objects.exclude(id=user.id).annotate(
    # Annotate the amount of groups related between the users
    related_group_count=Count('groups',filter=Q(groups__users=user))
).filter(
    # Only users that have at least 1 relation with this user
    related_group_count__gt=0
).prefetch_related(
    # Prefetch the specific groups to show
    Prefetch('groups',User.groups.all(),to_attr='related_groups'),)

返回的用户不会有任何related_groups，它将是一个空的qs。当我包含请求用户（通过删除查询中的排除部分）时，组将可用于该用户，但仍然不能用于任何其他用户。这是我当前的响应（包括请求用户“A”）：

{
    "first_name": "A","related_groups": [
        {
            "name": "group1",},{
            "name": "group2",}
    ]
},{
    "first_name": "B","related_groups": []
},{
    "first_name": "C","related_groups": []
}

但我正在努力实现这一目标：

{
    "first_name": "B","related_groups": [
        {
            "name": "group2",}
    ]
}

问题在于 Prefetch 查询集上的过滤器，但我无法弄清楚我做错了什么，或者这是否可能。我尝试了不同的过滤方式（也使用 Group.objects.filter(users=user)）甚至注释组成员，计算给定用户是否在该组中并对其进行过滤，但没有成功...

---

注意：这仅在过滤预取时发生，如上所述：用户（是组成员）-> 组 -> 用户（组成员）。看起来过滤器正在用初始用户（A）“覆盖”查询集中的用户。

在我的项目中，我使用多个类似组的概念，其中一个有 2 种不同类型的成员（可以说，在这个问题的上下文中，第二个 GroupMember 类型模型，在组模型中具有额外的多对多关系给用户）。当使用用户（组成员）-> 组 -> 用户（其他类型的成员）时，预取实际上按预期工作。仅当使用与用户相关的相同直通模型进行预取时才会出现此问题。

解决方法

通过使用一个单独的查询集来获取组 ID，我设法让它起作用。这样，预取过滤器就不会使用请求的用户 ID 进行过滤，从而导致所有其他用户的列表为空。

group_ids = user.groups.values('pk')

User.objects.exclude(id=user.id).filter(
    groups__in=group_ids
).prefetch_related(
    Prefetch('groups',Group.objects.filter(id__in=group_ids),to_attr='related_groups'),)