Как комментировать аннотированный подзапрос Django QuerySet - PullRequest
Новая
       24

Как комментировать аннотированный подзапрос Django QuerySet

1 голос
/ 12 марта 2019

У меня есть несколько моделей Django с отношениями FK между ними: 

from django.db import models


class Order(models.Model):
    notes = models.TextField(blank=True, null=True)


class OrderLine(models.Model):
    order = models.ForeignKey(Order, on_delete=models.CASCADE)
    quantity = models.PositiveIntegerField()
    price = models.DecimalField(max_digits=8, blank=True, decimal_places=2)

Для данной Линии Заказа вы можете рассчитать ее общее количество как цену: 

def get_order_line_total(order_line):
    return order_line.quantity * order_line.price

Для данного ордера вы можете рассчитать его итоговую сумму как сумму итогов по строкам ордеров: 

def get_order_total(order):
    order_total = 0
    for orderline_for in order.orderline_set.all():
        order_total += (order_line_for.quantity * order_line_for.price)
    return order_total

Я хочу аннотировать эти итоги в наборах запросов, чтобы я мог их фильтровать, сортировать и т. Д.

Для моделей OrderLine я нашел это довольно просто: 

from django.db.models import F, FloatField, Sum


annotated_orderline_set = OrderLine.objects.annotate(orderline_total=Sum(F('quantity') * F('price'), output_field=FloatField()))

Теперь я хочу аннотировать итоги в наборе запросов Order.objects. Я думаю, мне нужно будет использовать подзапрос, но я не могу заставить его работать. Я думаю (не работает): 

from django.db.models import F, FloatField, OuterRef, Subquery, Sum


Order.objects.annotate(
    order_total=Subquery(
        OrderLine.objects.filter(
            order=OuterRef('pk')
        ).annotate(
            orderline_total=Sum(F('quantity') * F('price'), output_field=FloatField())
        ).values(
            'orderline_total'
        ).aggregate(
            Sum('orderline_total')
        )['orderline_total__sum']
    )
)

# Not working, returns:
# ValueError: This queryset contains a reference to an outer query and may only be used in a subquery.

Как я мог решить это?

Ответы [ 2 ]

1 голос
/ 13 марта 2019

Как отметил комментарий @aedry, очень простое решение, позволяющее избежать подзапроса: 

Order.objects.annotate(total=models.Sum(F('orderline_set__quantity') * F('orderline_set__price'), output_field=models.DecimalField(max_digits=10, decimal_places=2)))

(я применил идею output_field = DecimalField из ответа @Todor для согласованности типов)

0 голосов
/ 13 марта 2019

Вы не можете использовать .aggregate, потому что это оценивает queryset немедленно, в то время как вам нужно отложить эту оценку до оценки внешнего запроса.

Таким образом, правильный подход - .annotate вместо.aggregate. 

class OrderQuerySet(models.QuerySet):
    def annotate_total(self):
        return self.annotate(
            total=models.Subquery(
                OrderLine.objects.filter(
                    order=models.OuterRef('pk')
                ).annotate_total()
                .values('order')
                .annotate(total_sum=models.Sum('total'))
                .values('total_sum')
            )
        )


class Order(models.Model):
    # ...
    objects = OrderQuerySet.as_manager()


class OrderLineQuerySet(models.QuerySet):
    def annotate_total(self):
        return self.annotate(
            total=models.ExpressionWrapper(
                models.F('quantity')*models.F('price'),
                output_field=models.DecimalField(max_digits=10, decimal_places=2)
            )
        )


class OrderLine(models.Model):
    #...
    objects = OrderLineQuerySet.as_manager()


# Usage:
>>> for l in OrderLine.objects.all().annotate_total():
...    print(l.id, l.order_id, l.quantity, l.price, l.total)
... 
1 1 3 20.00 60
2 1 9 10.00 90
3 2 18 2.00 36

>>> for o in Order.objects.all().annotate_total():
...    print(o.id, o.total)
... 
1 150
2 36
...