如何解决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'),)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。