python - 如何在 Django 中过滤对象以进行计数注释？

Question

考虑简单的 Django 模型 Event 和 Participant:

class Event(models.Model):
    title = models.CharField(max_length=100)

class Participant(models.Model):
    event = models.ForeignKey(Event, db_index=True)
    is_paid = models.BooleanField(default=False, db_index=True)

使用参与者总数注释事件查询很容易:

events = Event.objects.all().annotate(participants=models.Count('participant'))

如何用 is_paid=True 过滤的参与者数量进行注释？

无论参与者有多少，我都需要查询所有事件，例如我不需要按带注释的结果进行过滤。如果有 0 个参与者，那没关系，我只需要 0 在注释值中。

example from documentation在这里不起作用，因为它从查询中排除对象，而不是用 0 注释它们。

更新。 Django 1.8 有新的 conditional expressions feature ，所以现在我们可以这样做:

events = Event.objects.all().annotate(paid_participants=models.Sum(
    models.Case(
        models.When(participant__is_paid=True, then=1),
        default=0,
        output_field=models.IntegerField()
    )))

更新 2。 Django 2.0 有新的 Conditional aggregation功能，见 the accepted answer以下。这也适用于 Django 3.x

Answer 1

Conditional aggregation在 Django 2.0 中，您可以进一步减少过去的 faff 数量。这也将使用 Postgres 的 filter 逻辑，这比 sum-case 快一些(我见过像 20-30% 这样的数字)。

无论如何，在你的情况下，我们正在研究这样简单的事情:

from django.db.models import Q, Count
events = Event.objects.annotate(
    paid_participants=Count('participants', filter=Q(participants__is_paid=True))
)

文档中有一个单独的部分关于 filtering on annotations .它与条件聚合相同，但更像我上面的示例。不管怎样，这比我之前做的那些粗糙的子查询要健康得多。