Django注释对另一个注释有副作用

时间:2019-06-11 20:30:16

标签: python django django-annotate

我遇到了一个问题,其中向QuerySet添加注释会更改先前注释的结果。

以下是(简化的)设置:

class Player(models.Model):
    name = models.CharField()

class Unit(models.Model):
    player = models.ForeignKey(Player, on_delete=models.CASCADE,
                               related_name='unit_set')

    rarity = models.IntegerField()

class Gear(models.Model):
    pass

class UnitGear(models.Model):
    unit = models.ForeignKey(Unit, on_delete=models.CASCADE,
                             related_name='ug_set')
    gear = models.ForeignKey(Gear, on_delete=models.PROTECT)

用稀有7个单位注释玩家很有效:

Player.objects.annotate(
    rarity7_count=Count(unit_set__rarity=7)
).values_list('name', 'rarity7_count')

[('Player1', 170),
 ('Player2', 172),
 ('Player3', 164),
 ...,
)]

上面为rarity7_count返回的值是正确的。

如果我添加以下附加注释,则不再是这种情况:

Player.objects.annotate(
    rarity7_count=Count(unit_set__rarity=7),
    gear_count=Count(unit_set__ug_set)
).values_list('name', 'rarity7_count', 'gear_count')

[('Player1', 476, 456),
 ('Player2', 490, 466),
 ('Player3', 422, 433),
 ...,
)]

请注意rarity7_count值的更改方式-这些值不再正确!但是,gear_count的值是正确的。

那是为什么?如何使两个注释都能正常工作而不互相干扰?我已经尝试过各种方法,但现在却对如何执行此方法感到迷茫。

1 个答案:

答案 0 :(得分:2)

,因为现在有两个JOIN,并且由于Count(..) [Django-doc]是行数,所以它将充当某种乘数。

不过,我们可以通过指定distinct=True来解决此问题,例如:

Player.objects.annotate(
    rarity7_count=Count('unit_set', distinct=True, filter=Q(unit_set__rarity=7)),
    gear_count=Count('unit_set__ug_set')
).values_list('name', 'rarity7_count', 'gear_count')

请注意,如果您希望gear_count也根据稀有性进行过滤,则需要再次指定filter=部分。