为包含相同多对多关系子集的对象过滤django

时间:2018-10-23 22:07:58

标签: django django-queryset python-3.5

在我的数据库中,我的用户对象上有两个多对多字段(消息和后续字段),它们都包含与另一个对象主题相关的多对多字段。

class User():
    messages = ManyToManyField('Message', related_name='users', blank=True, null=True)
    following = ForeignKey('Following', related_name='users', blank=True, null=True) 

class Message():
    date = DateField(blank=True, null=True)
    content = TextField(blank=True, null=True)
    topics = ManyToManyField('Topic', related_name='messages', blank=True, null=True)

class Following():
    name = CharField(max_length=255, blank=True, null=True)
    description = CharField(max_length=255, blank=True, null=True)
    topics = ManyToManyField('Topic', related_name='following', blank=True, null=True)

class Topic():
    name = CharField(max_length=255, blank=True, null=True)
    source = CharField(max_length=255, blank=True, null=True)

我要过滤所有附加了“消息”但不包含附加到用户“以下”对象的所有主题的“用户”。

现在我正在使用循环来完成此操作:

users = set()
for user in User.objects.filter(messages__isnull=False, following__isnull=False).iterator():
    if not set(user.following.values_list('topics', flat=True))
               ).issubset(set(user.messages.values_list('topics', flat=True)):
       users.add(user.pk)

是否有一种方法可以通过单个查询完成同一件事?

----编辑----

我有什么?

User.objects.filter(following__isnull=False
).annotate(following_count=Count('following__topics', distinct=True)
).filter(following__topics__exact=F('message__topics')
).annotate(missing_topics=ExpressionWrapper(
    F('following_count') - Count('message__topics', distinct=True),
                                 IntegerField())
).filter(missing_topics__gt=0)

如果有更好的方法可以执行此操作,或者出于某些原因我绝对不应该这样做,那么它们是什么?

----编辑----

This question帮助我理解和使用Håken Lid's answer

这是我的新模型和新查询:

class User():
    messages = ManyToManyField('Message', related_name='users', blank=True, null=True)
    following = ManyToManyField('Topic', through='Following', related_name='users', blank=True, null=True) 

class Message():
    date = DateField(blank=True, null=True)
    content = TextField(blank=True, null=True)
    topics = ManyToManyField('Topic', related_name='messages', blank=True, null=True)

class Following():
    name = CharField(max_length=255, blank=True, null=True)
    description = CharField(max_length=255, blank=True, null=True)
    user = ForeignKey('User', related_name='following', blank=True, null=True)
    topic = ForeignKey('Topic', related_name='following', blank=True, null=True)

class Topic():
    name = CharField(max_length=255, blank=True, null=True)
    source = CharField(max_length=255, blank=True, null=True)



User.objects.filter(~Q(messages__topics__in=F('following'))
).values('id').annotate(missing_topics=Count('following__topics', distinct=True))

1 个答案:

答案 0 :(得分:1)

这应该可以通过子查询来实现。

首先,确保Following.topics使用与Messages.topics不同的相关名称。

class Following(models.Model):
    topics = ManyToManyField('Topic', related_name='following')

然后应该可以创建一个subquery。像这样:

from django.db.models import OuterRef, Subquery
user_following_topic = Topic.objects.filter(following__users=OuterRef('pk'))
User.objects.exclude(messages__topics__in=Subquery(user_following_topics.values('pk')))

这可能无法正常工作,并且不能完全按照书面要求提供预期输出,但是我认为该原则也应适用于您的情况。

另一方面,我不太了解您的数据库结构。似乎您使用m2m关系,其中外键可能更合适和更简单。您的关系越复杂,创建这种高级查询就越困难。而且,与简单查询相比,具有大量数据库联接的查询可能会非常慢,因为它们可能必须处理大量数据。

例如,Following不是使用m2m分配,而是像这样对我来说更有意义:

class Following():
    topic = ForeignKey('Topic', on_delete=models.CASCADE)
    user = ForeignKey('User', on_delete=models.CASCADE)
    client = models.CharField(max_length=255, blank=True, null=True)
    duration = fields.DateRangeField(blank=False, null=False)

基本上就是“直通”模型,如django docs on model relationships所述,这里有一个类似的例子。