注释过滤 - 仅对一些相关对象求和'字段

时间:2016-07-25 12:01:07

标签: django django-models django-orm

让我们说是作者,他有书。为了将作者与书面页面的数量一起获取,可以完成以下操作:

Author.objects.annotate(total_pages=Sum('book__pages'))

但是如果我想分别对科幻和奇幻书籍的页面进行总结呢?我希望最终得到一个具有total_pages_books_scifi_pages和total_pages_books_fantasy_pages属性的作者。

我知道我可以做以下事情:

Author.objects.filter(book__category='scifi').annotate(total_pages_books_scifi_pages=Sum('book__pages'))
Author.objects.filter(book__category='fantasy').annotate(total_pages_books_fantasy_pages=Sum('book__pages'))

但是在一个查询集中怎么做?

2 个答案:

答案 0 :(得分:10)

from django.db.models import IntegerField, F, Case, When, Sum

categories = ['scifi', 'fantasy']
annotations = {}

for category in categories:
    annotation_name = 'total_pages_books_{}'.format(category)
    case = Case(
        When(book__category=category, then=F('book__pages')),
        default=0,
        output_field=IntegerField()
    )
    annotations[annotation_name] = Sum(case)

Author.objects.filter(
    book__category__in=categories
).annotate(
    **annotations
)

答案 1 :(得分:3)

尝试:

Author.objects.values("book__category").annotate(total_pages=Sum('book__pages'))

来自Django docs: https://docs.djangoproject.com/en/1.10/topics/db/aggregation/#values

  

值()

     

通常,注释是基于每个对象生成的 - 带注释的QuerySet将为原始QuerySet中的每个对象返回一个结果。但是,当使用values()子句约束结果集中返回的列时,评估注释的方法略有不同。原始结果不是为原始QuerySet中的每个结果返回带注释的结果,而是根据values()子句中指定的字段的唯一组合进行分组。然后为每个唯一组提供注释;注释是在该组的所有成员上计算的。

相关问题