带有多个 ForeignKey 引用的 Django 注释问题

Question

我正在努力使用注释，但没有找到帮助我理解的示例。以下是我的模型的相关部分：

    class Team(models.Model):
        team_name = models.CharField(max_length=50)
    
    class Match(models.Model):
        match_time = models.DateTimeField()
        team1 = models.ForeignKey(
            Team, on_delete=models.CASCADE, related_name='match_team1')
        team2 = models.ForeignKey(
            Team, on_delete=models.CASCADE, related_name='match_team2')
        team1_points = models.IntegerField(null=True)
        team2_points = models.IntegerField(null=True)

我想最终得到的是 Teams 对象上的注释，它可以为我提供每个团队的总分。有时，一个团队是match.team1（所以他们的积分在match.team1_points），有时他们是match.team2，他们的积分存储在match.team2_points。

这是我得到的最接近的结果，大概尝试了一百次左右：

teams = Team.objects.annotate(total_points = 
Value(
  (Match.objects.filter(team1=21).aggregate(total=Sum(F('team1_points'))))['total'] or 0 + 
  (Match.objects.filter(team2=21).aggregate(total=Sum(F('team2_points'))))['total'] or 0, 
  output_field=IntegerField())
)

这很好用，但（当然）为 pk=21 的团队注释 total_points 到查询集中的每个团队。如果有更好的方法来解决这一切，我很乐意看到它，但除此之外，如果你能告诉我如何将那些“21”值转化为对外部团队 pk 的参考，我认为这会奏效吗？

编辑：我最终结合使用 elyas 的答案和注释原始 SQL 语句来解决我的问题。我无法阻止普通注释从查询集中删除非唯一分数，但原始 SQL 似乎可以工作。

这是原始注释：

teams = Team.objects.raw('select id, sum(points) as total_points from (select team1_id as id, team1_points as points from leagueman_match union all select team2_id as id, team2_points as points from leagueman_match) group by id order by total_points desc;')

Answer 1

注释 QuerySet 的另一种解决方案可能是使 total_points 成为 Team 模型的 @property（取决于用例）：

from django.db.models import Case, Q, Sum, When


class Team(models.Model):
    team_name = models.CharField(max_length=50)

    @property
    def total_points(self):
        return Match.objects.filter(Q(team1=self.id) | Q(team2=self.id)).aggregate(
            total_points=Sum(Case(
                When(team1=self.id, then='team1_points'),
                When(team2=self.id, then='team2_points')
            ))
        )['total_points']

缺点是它不能用于后续的 QuerySet 操作，例如.values(), .order_by().

Django 还有一个@cached_property decorator，它将在第一次调用时缓存属性的输出。

尝试了其他解决方案

最初我认为您可以利用 Team 模型中的反向关系 match_team1 和 match_team2 来生成一个简单的注释：

teams = Team.objects.annotate(
    total_points=(
        Sum('match_team1__team1_points', distinct=True)
        + Sum('match_team2__team2_points', distinct=True)
    )
)

不幸的是，此解决方案在处理重复项时遇到了困难。 distinct=True 参数消除了同一场比赛的分数被多次相加的问题。但它引入了一个不同的问题，即得分相同的不同比赛将被排除在外。

Answer 2

也许我会建议重构您的数据模型。即使您找到了解决此特定问题的方法，您也可能想提前考虑。 这是一个解决方案：

class Team(models.Model):
    team_name = models.CharField(max_length=50)

class Match(models.Model):
    match_time = models.DateTimeField()
    team1 = models.ForeignKey(
        Team, on_delete=models.CASCADE, related_name='match_team1')
    team2 = models.ForeignKey(
        Team, on_delete=models.CASCADE, related_name='match_team2')

class Points(models.Model):
    match = models.ForeignKey(
        Match, on_delete=models.CASCADE, related_name='match')
    team = models.ForeignKey(
        Team, on_delete=models.CASCADE, related_name='team')
    points = models.IntegerField(null=True)

有了这个，您可以轻松总结任何球队的积分，也可以按比赛进行过滤。