Django：如何为QuerySet添加基于request.user的用户反应注解字段

嘿，这个需求很常见，咱们直接用Django ORM里的Case和When就能完美解决，不用写复杂的自定义方法。下面是具体实现步骤：

1. 先确保数据约束（可选但强烈推荐） #

首先建议给Like模型加个唯一约束，保证同一个用户对同一条Tweet只能有一条反应记录，避免数据混乱：

class Like(models.Model):
    tweet = models.ForeignKey(Tweet, related_name='likes')
    user = models.ForeignKey(AUTH_USER_MODEL, related_name='my_likes')
    type = models.CharField(
        choices=(
            ('like', _"Like"),
            ('dislike', "Dislike")
        ),
        max_length=10
    )

    class Meta:
        # 确保一个用户对一条Tweet只能有一个反应
        unique_together = ('tweet', 'user')

2. 编写带注解的查询 #

导入需要的ORM工具，然后通过annotate结合Case/When给每个Tweet添加user_reaction字段：

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

# 获取当前请求用户
current_user = request.user

# 执行查询并添加注解
tweets = Tweet.objects.annotate(
    user_reaction=Case(
        # 匹配用户点赞的情况，返回1
        When(likes__user=current_user, likes__type='like', then=Value(1)),
        # 匹配用户点踩的情况，返回2
        When(likes__user=current_user, likes__type='dislike', then=Value(2)),
        # 默认情况：用户没有任何反应，返回3
        default=Value(3),
        output_field=IntegerField()
    )
)

3. 使用注解字段 #

现在你就可以直接访问每个Tweet对象的user_reaction属性了：

# 获取第11条Tweet的用户反应（索引从0开始）
reaction = tweets[10].user_reaction
# reaction的值会是1（点赞）、2（点踩）或3（无反应）

可选：返回字符串类型的反应 #

如果你希望直接返回字符串（比如'like'、'dislike'、'none'），只需要调整Value和output_field即可：

from django.db.models import Case, When, Value, CharField

tweets = Tweet.objects.annotate(
    user_reaction=Case(
        When(likes__user=current_user, likes__type='like', then=Value('like')),
        When(likes__user=current_user, likes__type='dislike', then=Value('dislike')),
        default=Value('none'),
        output_field=CharField()
    )
)

这个方案完全用Django ORM实现，不需要额外的自定义方法，性能也有保障——所有逻辑都在数据库层面完成，避免了N+1查询问题。

内容的提问来源于stack exchange，提问作者msln