给定一个查询集,我使用以下内容添加相关对象(ModelA)的计数:
qs = User.objets.all()
qs.annotate(modela__count=models.Count('modela'))
但是,有没有办法计算只符合标准的ModelA?例如,计算ModelA,其中deleted_at为null?
我尝试了两种无法正常工作的解决方案。
1)正如@knbk建议的那样,在注释之前使用过滤器。
qs = User.objects.all().filter(modela__deleted_at__isnull=True).annotate(modela__count=models.Count('modela', distinct=True))
以下是django生成的查询的简化版本:
SELECT COUNT(DISTINCT "modela"."id") AS "modela__count", "users".*
FROM "users"
LEFT OUTER JOIN "modela" ON ( "users"."id" = "modela"."user_id" )
WHERE "modela"."deleted_at" IS NULL
GROUP BY "users"."id"
问题来自WHERE子句。 实际上,有一个LEFT JOIN,但后来的WHERE条件迫使它成为一个简单的JOIN。我需要将条件提升到JOIN子句中,以使其按预期工作。
所以,而不是
LEFT OUTER JOIN "modela" ON ( "users"."id" = "modela"."user_id" )
WHERE "modela"."deleted_at" IS NULL
当我在纯SQL中直接执行它时,我需要以下内容。
LEFT OUTER JOIN "modela" ON ( "users"."id" = "modela"."user_id" )
AND "modela"."deleted_at" IS NULL
如何在不执行原始查询的情况下更改查询集以获取此内容?
2)正如其他人所说,我可以使用条件聚合。
我尝试了以下内容:
qs = User.objects.all().annotate(modela__count=models.Count(Case(When(modela__deleted_at__isnull=True, then=1))))
变成以下SQL查询:
SELECT COUNT(CASE WHEN "modela"."deleted_at" IS NULL THEN 1 ELSE NULL END) AS "modela__count", "users".*
FROM "users" LEFT OUTER JOIN "modela" ON ( "users"."id" = "modela"."user_id" )
GROUP BY "users"."id"
通过这样做,我获得所有用户(因此LEFT JOIN正常工作)但是对于没有任何ModelA的所有用户,modela__count
得到“1”(而不是0) 。
如果没有什么可计算的,为什么我得到1而不是0?
怎么能改变呢?
答案 0 :(得分:37)
在LEFT JOIN
中,modela
的每个字段都可能是NULL
,因为没有相应的行。所以
modela.deleted_at IS NULL
...不仅适用于匹配的行,而且对于那些没有相应users
行的modela
也是如此。
我认为正确的SQL应该是:
SELECT COUNT(
CASE
WHEN
`modela`.`user_id` IS NOT NULL -- Make sure modela rows exist
AND `modela`.`deleted_at` IS NULL
THEN 1
ELSE NULL
END
) AS `modela__count`,
`users`.*
FROM `users`
LEFT OUTER JOIN `modela`
ON ( `users`.`id` = `modela`.`user_id` )
GROUP BY `users`.`id`
在Django 1.8中,这应该是:
from django.db import models
qs = User.objects.all().annotate(
modela_count=models.Count(
models.Case(
models.When(
modela__user_id__isnull=False,
modela__deleted_at__isnull=True,
then=1,
)
)
)
)
<强>通知强>:
INNER JOIN
而不是LEFT JOIN
,因此您将丢失没有相应外键的行关系。 使用Django 1.8.2或更高版本版本来修复它。
答案 1 :(得分:2)
在Django 1.8中,我相信这可以通过conditional aggregation 来实现。但是对于以前的版本,我会使用.extra
ModelA.objects.extra(select={
'account_count': 'SELECT COUNT(*) FROM account WHERE modela.account_id = account.id AND account.some_prop IS NOT NULL'
})
答案 2 :(得分:0)
您可以在注释前简单过滤:
from django.db.models import Q, Count
qs = ModelA.objects.filter(account__prop1__isnull=False).annotate(account_count=Count('account'))