Django 将正值与负值分开

Question

您可以将正值与负值分开求和吗？

我有一个看起来像这样的表：

class Order(models.Model):
    date = models.DateField()
    amount = models.DecimalField()

我正在寻找一个查询集，它按周将正金额与负金额和组分开：

Year
Week
Sum of positive amounts
Sum of negative amounts

我可以通过以下方式每周分别获得所有正数：

Order.objects.filter(amount__gt=0).annotate(year_week=Trunc('date', 'week', output_field=DateField())).\
annotate(year=ExtractYear('year_week')).annotate(week=ExtractWeek('year_week')).\
values("year_week", "year", "week").annotate(value=Sum("amount")).order_by("-year_week")

我可以为负值再次执行此操作，但是我有两个单独的查询集。 我可以在不手动循环的情况下有效地合并它们吗？或者有没有办法在一个查询中同时完成这两者？

Answer 1

是的，从django-2.0 开始，大多数聚合函数都有一个filter=…参数 [Django-doc] ：

from django.db.models import Q, Sum
from django.db.models.functions import ExtractYear, ExtractWeek

Order.objects.values(
    year=ExtractYear('date'),
    week=ExtractWeek('date')
).annotate(
    sum_pos=Sum('amount', filter=Q(amount__gt=0)),
    sum_neg=Sum('amount', filter=Q(amount__lt=0))
).order_by('year', 'week')