对中间模型 django 的一个字段进行注释

Question

我有以下方案：

class User(AbstractUser):
    pass

class Task(models.Model):
    pass

class Contest(models.Model):
    tasks = models.ManyToManyField('Task',
                               related_name='contests',
                               blank=True,
                               through='ContestTaskRelationship')

    participants = models.ManyToManyField('User',
                                      related_name='contests_participated',
                                      blank=True,
                                      through='ContestParticipantRelationship')

class ContestTaskRelationship(models.Model):
    contest = models.ForeignKey('Contest', on_delete=models.CASCADE)
    task = models.ForeignKey('Task', on_delete=models.CASCADE)
    cost = models.IntegerField()


class ContestParticipantRelationship(models.Model):
    contest = models.ForeignKey('Contest', on_delete=models.CASCADE)
    user = models.ForeignKey('User', on_delete=models.CASCADE)
    task = models.ForeignKey('Task', on_delete=models.CASCADE, related_name='contests_participants_relationship')
    is_solved = models.BooleanField()

现在我得到了contest 对象，需要获取tasks 字段上的所有任务，用用户数注释的rach 解决了它。所以，我需要计算ContestParticipantRelationship 的数量，需要task，需要contest 和is_solved 设置为True。如何进行这样的查询？

Answer 1

大概是这样的：

from django.db.models import IntegerField, Value, Sum
from django.db.models.functions import Cast, Coalesce


Task.objects.filter(
    contests__contest=some_contest,
).annotate(
    nsolved=Cast(Coalesce(
        Sum('contests_participants_relationship__is_solved'), Value(0)
    ),IntegerField())
)

所以这里我们首先过滤任务的竞赛是some_contest。接下来我们在is_solved 列上执行Sum(..)。由于存在可能是NULL 的极端情况（如果没有用户进行尝试等），那么我们将其转换为0，此外我们将其转换为IntegerField，因为否则某些实例可能会使用True 和False 进行注释，以防零个或一个用户解决它。