如何在Django的annotate中使用自定义查询统计指定记录？

下面提供几种可行的方案，帮你实现仅统计符合buzzed条件的推文数量：

方法一：直接在Count中使用filter参数（推荐） #

Django 2.0及以上版本支持在聚合函数（如Count）中传入filter参数，直接指定统计的筛选条件，这种方式简洁且复用性强：

首先可以把buzzed的条件封装到TweetQuerySet中避免重复代码：

class TweetQuerySet(models.QuerySet):
    _BUZZED_Q = Q(is_active=True, favorite__gte=100)

    def buzzed(self):
        return self.filter(self._BUZZED_Q)

然后在统计时直接引用这个条件：

from django.db.models import Count

users = User.objects.annotate(
    tweet_count=Count('tweet', filter=TweetQuerySet._BUZZED_Q)
)

如果不想修改原有TweetQuerySet，也可以直接在Count里写条件：

from django.db.models import Count, Q

users = User.objects.annotate(
    tweet_count=Count('tweet', filter=Q(is_active=True, favorite__gte=100))
)

方法二：使用FilteredRelation定义筛选后的关联 #

这种方式会先定义一个符合筛选条件的推文关联，再基于这个关联统计数量，适合需要复用该筛选关联做其他操作的场景：

from django.db.models import FilteredRelation, Count, Q

users = User.objects.annotate(
    # 定义筛选后的推文关联
    buzzed_tweets=FilteredRelation('tweet', condition=Q(is_active=True, favorite__gte=100)),
    # 统计该关联的记录数
    tweet_count=Count('buzzed_tweets')
)

方法三：子查询实现（复杂场景适用） #

如果筛选逻辑涉及更复杂的聚合或关联操作，可以用子查询来统计：

from django.db.models import Subquery, OuterRef, Count

# 子查询获取单个用户符合条件的推文数
buzzed_count_subquery = Tweet.objects.buzzed()\
    .filter(user=OuterRef('pk'))\
    .annotate(count=Count('pk'))\
    .values('count')

users = User.objects.annotate(
    tweet_count=Subquery(buzzed_count_subquery, output_field=models.IntegerField())
)

内容的提问来源于stack exchange，提问作者y.dai