Django用交集计数注释查询集

Question

Djangonauts，我需要挖掘你的大脑。

简而言之，我有以下三个模型：

class Location(models.Model):
    name = models.CharField(max_length=100)


class Profile(models.Model):
    locations_of_interest = models.ManyToManyField(Location)


class Question(models.Model):
    locations = models.ManyToManyField(Location)

我想查找感兴趣的位置与为特定问题指定的位置相交的所有配置文件。这很简单：

question = Question.objects.first()

matching_profiles = Profile.objects.filter(
    locations_of_interest__in=question.locations.all()
)

但除此之外，我还想知道这些位置重叠到什么程度。

在普通的 python 中，我可以这样做：

question_location_names = [l['name'] for l in question.locations.all()]

for profile in matching_profiles:
    profile_location_names = [l['name'] for l in profile.locations_of_interest.all()]
    intersection = set(question_location_names).intersection(profile_location_names)
    intersection_count = len(list(intersection))
    # then proceed with this number

但是，如果可能的话，在我看来直接在数据库中进行操作似乎是有利的。

TL;DR

所以我的问题是：

有没有办法用这个交集计数注释配置文件查询集，然后在数据库中进行操作？

我尝试了几件事，但我认为它们对阅读本文并可能知道答案的人没有帮助。

Answer 1

您可以在locations_of_interest 号码上使用Count(..) 执行此操作：

from django.db.models import Count

matching_profiles = Profile.objects.filter(
    locations_of_interest__in=question.locations.all()
).annotate(
    locnom=Count('locations_of_interest')
)

现在每个matching_profiles 实例都有一个名为locnom 的属性，其中包含与过滤器匹配的兴趣位置的数量。

请注意，没有这些位置的Profiles 将不在查询集中，并且每个Profile 最多会出现一次。

编辑：计算多个相关的非重叠 (!) 字段

您可以通过使用distinct=True 计算非重叠连接来扩展此方法：

from django.db.models import Count

matching_profiles = Profile.objects.filter(
    locations_of_interest__in=question.locations.all(),
    industries_of_interest__in=question.industries.all()
).annotate(
    locnom=Count('locations_of_interest', distinct=True),
    indnom=Count('industries_of_interest', distinct=True)
)

但是请注意，这种方法通常会随着JOINs 的数量指数扩展，因此如果您要添加数十个我们数百个注释，这通常不可扩展.