高级 django 查询

Question

我有三个模型：公司、资产和用户。每个公司都有很多资产（ForeignKey），每个资产都有一个用户（ForeignKey，因为一个用户可能拥有多个资产）。

我想运行一个查询，返回给定用户拥有资产的公司列表，以及他们在每家公司拥有的资产数量。

我曾尝试在 python 中这样做，如下所示：

companies = Company.objects.filter(asset__owner=self.request.user)
context['investments'] = [(x, Asset.objects.filter(company=x, owner=self.request.user).count()) for x in companies] 

这可能并不令人惊讶，因为一家公司可以拥有 100,000 项资产。似乎在数据库级别这样做会更好，这导致我查看注释和聚合，但我无处可去。有人能指出我正确的方向吗？

Answer 1

您可以使用条件进行注释：

from django.db.models import Count, Case, When, IntegerField

qs = Company.objects.filter(asset__owner=self.request.user)
qs = qs.objects.annotate(
    num_assets=Count(Case(
        When(asset__owner=self.request.user, then=1),
        output_field=IntegerField(),
    ))
)

阅读文档conditional-aggregation

Answer 2

您可以通过.filter(..) 和.annotate(..) 的组合来做到这一点：

from django.db.models import Count

Company.objects.filter(
    asset__owner=some_user
).annotate(
    num_assets=Count('asset')
)

所以这里我们返回一个包含Company 对象的QuerySet，每个对象都有一个额外的属性：num_assets，它包含Company 中给定用户拥有的资产数量。

从这里开始 - 如果我理解正确 - 资产始终属于单一公司，没有重复计算：如果它是多对- many 关系，即属于两家公司的用户所拥有的 assed，会增加两家公司的计数。

因此，查询集仅返回用户拥有资产的公司，Companys 没有用户拥有的任何资产，因此被过滤掉。 p>