Django ManyToMany按集合中的集大小或成员进行过滤

时间:2012-12-28 09:26:34

标签: python django django-models django-queryset

我正在使用django来维护邮件数据库 其中我有以下模型

class User(models.Model):
    id = models.IntegerField(primary_key=True)
    name = models.CharField(max_length=10)

class Message(models.Model):
    id = models.IntegerField(primary_key=True)
    body = models.CharField(max_length=200)
    users = models.ManyToManyField(User)

我正在尝试编写一个实用程序方法,为给定的用户提供他(和他一个人)相关联的消息。

即。 for:

m1 = Message(id=1, body='Some body')
m1.save()
m2 = Message(id=2, body='Another body')
m2.save()
m3 = Message(id=3, body='And yet another body')
m3.save()

u1 = User(name='Jesse James')
u1.save()
u2 = User(name='John Doe')
u2.save()

m1.users.add(u1, u2)
m2.users.add(u1)
m3.users.add(u2)

getMessagesFor('Jesse James')

仅返回m2 假设我在user中有正确的模型实例,它归结为一行,我尝试了以下这些:

    user.message_set.annotate(usr_cnt=Count('users')).filter(usr_cnt__lte=1)

或者:

    messages = Message.objects.filter(users__id__in=[user.id])

并且:

    messages = Message.objects.filter(users__id__exact=user.id)

并且:

    messages = Message.objects.filter(users__contains=user)

等等......我总是得到m2 AND m1
尝试注释,排除,过滤等。

有人可以帮我这个吗?

2 个答案:

答案 0 :(得分:5)

qs = Message.objects.annotate(cc=Count('users')).filter(cc=1)

上面的查询将返回只有一个用户与之关联的所有消息。

要按用户过滤,请在末尾添加另一个过滤器,以根据用户过滤带注释的查询:

qs = Message.objects.annotate(cc=Count('users')).filter(cc=1).filter(users__id=user.id)
# if user user.id=1, this will return only m2

答案 1 :(得分:0)

这样的事可能吗? (未经测试)

for msg in Messages.objects.all():
    if (user in msg.users_set.all() and len(msg.users_set.all()) == 1):
        # do something