在Django查询中算出一个以上的事件

时间:2019-06-24 23:29:25

标签: django

我有两个模型:玩家和锦标赛。

玩家还玩一种游戏(dota,lol,魔术等)。他们可以同时参加许多比赛(每场比赛只能参加一次)。为了管理铭文,我使用了另一个名为TournamentMatch的模型,该模型为具有锦标赛ID和玩家ID的每个铭文创建一个新对象。

class Player(models.Model):
    name = models.CharField(_('name'), max_length=150, null=True, blank=True)
    email = models.EmailField(_('email address'), unique=True)
    is_dota_player = models.BooleanField(default=False)
    is_lol_player = models.BooleanField(default=False)
    is_magic_player = models.BooleanField(default=False)

class Tournament(models.Model):
    name = models.CharField(max_length=200)
    date_start = models.DateField()
    date_end = models.DateField()

class TournamentMatch(models.Model):
    tournament = models.ForeignKey(Tournament)
    player = models.ForeignKey(Player)
    date_added = models.DateTimeField(auto_now_add=True)
    date_modified = models.DateTimeField(auto_now=True)

我想计算比赛中有两个以上铭文的玩家人数,例如dota玩家。

对于每个玩家,我都可以轻松地通过for循环实现这一点,但是出于性能方面的考虑,我想使用Django查询来实现这一点。

例如,如果我想计算具有一个或多个题字的Dota玩家,我将这样做:

TournamentMatch.objects.filter(
    player__is_dota_player=True
).distinct(
   'player'
).count()

我肯定是有可能的,但是我不知道如何计算比赛中不同球员的每次出现率,而只过滤那些拥有多个(而不是只有一个)的球员。

我将不胜感激!

1 个答案:

答案 0 :(得分:0)

# subquery for counting match per player (note OuterRef which will be linked to the outer query row)
player_match_count_subquery = TournamentMatch.objects.filter(player_id=OuterRef('pk'), ).\
    annotate(match_count=Count('pk'), ).values('match_count', )[:1]

# the main (outer) query which uses subquery output for filtering
Players.objects.filter(is_dota_player=True, ).\
    annotate(match_count=Subquery(player_match_count_subquery), ).\
    filter(match_count__gt=2, )

https://docs.djangoproject.com/en/2.2/ref/models/expressions/#subquery-expressions

此代码将在逻辑上产生此SQL查询:

SELECT p.*, tm.match_count
FROM players p
CROSS APPLY (
    SELECT COUNT(id) match_count
    FROM tournamentmatch tm
    WHERE tm.player_id = p.id
) tm
WHERE p.is_dota_player = $true
    AND tm.match_count > 2