Django M2M QuerySet对多个外键进行过滤

时间:2012-12-01 21:09:27

标签: python django many-to-many django-queryset

我有两个类用于我正在处理的消息传递模块。这个想法是一个参与者由一组参与者(两个或更多)代表。我正在努力寻找一种通过逻辑来查找对话的方法,说明我想要找到的所需对话有以下参与者。我尝试Conversation.objects.filter(participants__in=[p1, p2])然而这是一个OR样式查询,p1是参与者或p2是参与者。我想要p1和p2和... pN是参与者。有任何帮助吗?

class Conversation(models.Model):
    date_started = models.DateTimeField(auto_now_add=True)
    participants = models.ManyToManyField(User)

    def get_messages(self):
        return Message.objects.filter(conversation=self)

    def new_message(self, sender, body):
        Message.objects.create(sender=sender, body=body, conversation=self)
        self.save()


class Message(models.Model):
    sender = models.ForeignKey(User)
    body = models.TextField()
    date = models.DateTimeField(auto_now_add=True)
    conversation = models.ForeignKey(Conversation)

    def __unicodde__(self):
        return body + "-" + sender 

4 个答案:

答案 0 :(得分:3)

我认为你只需要迭代过滤。这可能完全是胡说八道,因为我有点睡眠不足,但也许是这样的经理方法:

class ConversationManager(models.Manager):
    def has_all(self, participants):
        # Start with all conversations
        reducedQs = self.get_query_set()
        for p in participants:
            # Reduce to conversations that have a participant "p" 
            reducedQs = reducedQs.filter(participants__id=p.id)
        return reducedQs

一般来说,你应养成制作表级查询管理器方法的习惯,而不是类方法。通过这种方式,你留下了一个查询集,你可以根据需要进一步过滤。

受到the documentationthis answer中成员名称为Paul的所有群组的查询启发。

答案 1 :(得分:2)

如果在同一个相关模型上链接多次filter(),生成的查询将对同一个表有一个额外的JOIN。

所以你有:Conversation.objects.filter(participants=p1).filter(participants=p2)

您可以通过查看生成的查询print Conversation.objects.filter(participants=p1).filter(participants=p2).query

来确认此行为

请参阅:https://docs.djangoproject.com/en/dev/topics/db/queries/#spanning-multi-valued-relationships

由于它仍然相当简单和有效,我会避免在查询后使用python逻辑,这需要从数据库中提取太多数据,然后通过迭代再次过滤。

答案 2 :(得分:0)

首先,我会在participants字段中添加一个相关名称:

participants = models.ManyToManyField(User, related_name='conversations')

这不是必需的,但更具可读性的IMO。

然后你可以做类似的事情:

p1.conversations.filter(participants__in=p2)

这将返回p2也参与的所有p1对话。

我不确定这种过滤方法的数据库效率,也许使用其他类型的数据库(可能是图形数据库,如Neo4j)更合适。

答案 3 :(得分:0)

这样做的一种方法可能是使用python集:

#Get the set of conversation ids for each participant
    p1_conv_set = set(Converstation.objects.filter(participants = p1).values_list('id', flat=True))
    p2_conv_set = set(Converstation.objects.filter(participants = p2).values_list('id', flat=True))
    .
    .
    pn_conv_set = set(Converstation.objects.filter(participants = pN).values_list('id', flat=True))
    #Find the common ids for all participants
    all_participants_intersection = p1_conv_set & p2_conv_set & ....pN_conv_set
    #Get all the conversation for all the calculated ids
    Conversation.objects.filter(id__in = all_participants_intersection)
相关问题