如何使用Django annotate实现带条件取值的查询?
解决方案
要实现给ModelA查询集添加符合规则的value字段,你可以结合Django的Case、When、Exists和子查询/聚合函数来完成,具体分两种场景处理:
场景1:每个ModelA最多关联一个ModelB和ModelC
如果业务逻辑中一个ModelA只会对应一条ModelB和一条ModelC记录,可直接通过关联字段取值:
from django.db.models import Case, When, IntegerField, Exists # 构建带annotate的查询集 model_a_qs = ModelA.objects.annotate( value=Case( # 判断是否存在关联的ModelC记录 When(Exists(ModelC.objects.filter(model_A=OuterRef('pk'))), then='modelc__value'), # 不存在则取ModelB的value default='modelb__value', output_field=IntegerField() ) )
场景2:一个ModelA对应多条ModelB/ModelC记录
如果存在一对多关联,需先通过聚合函数(如Max、Min)确定要取的value,再应用规则:
from django.db.models import Case, When, IntegerField, Exists, Max model_a_qs = ModelA.objects.annotate( # 聚合出ModelC的最大value(可按需替换为Min/Avg等) modelc_value=Max('modelc__value'), # 聚合出ModelB的最大value modelb_value=Max('modelb__value'), # 应用取值规则 value=Case( When(Exists(ModelC.objects.filter(model_A=OuterRef('pk'))), then='modelc_value'), default='modelb_value', output_field=IntegerField() ) )
补充:精准获取指定关联记录的value
如果需要取某一条特定关联记录的value(比如最新创建的),可以用Subquery实现:
from django.db.models import Subquery, OuterRef # 取最新创建的ModelC的value latest_modelc_value = ModelC.objects.filter(model_A=OuterRef('pk')).order_by('-id').values('value')[:1] # 取最新创建的ModelB的value latest_modelb_value = ModelB.objects.filter(model_A=OuterRef('pk')).order_by('-id').values('value')[:1] model_a_qs = ModelA.objects.annotate( value=Case( When(Exists(ModelC.objects.filter(model_A=OuterRef('pk'))), then=Subquery(latest_modelc_value)), default=Subquery(latest_modelb_value), output_field=IntegerField() ) )
内容的提问来源于stack exchange,提问作者Andi Fathul Mukminin
相关产品推荐
相关产品推荐

