How to calculate total by month without using extra?
I’m currently using:
- django 1.8
- postgre 9.3.13
- Python 2.7
Example.
What I have tried so far.
#Doesn't work for me but I don't mind because I don't want to use extra truncate_month = connection.ops.date_trunc_sql('month','day') invoices = Invoice.objects.filter(is_deleted = False,company = company).extra({'month': truncate_month}).values('month').annotate(Sum('total')) ---- #It works but I think that it's too slow if I query a big set of data for current_month in range(1,13): Invoice.objects.filter(date__month = current__month).annotate(total = Sum("total"))
and also this one, the answer seems great but I can’t import the TruncMonth module.
Django: Group by date (day, month, year)
P.S. I know that this question is already asked multiple times but I don’t see any answer.
Thanks!
SOLUTION:
Thanks to @Vin-G’s answer.
Advertisement
Answer
First, you have to make a Function that can extract the month for you:
from django.db import models from django.db.models import Func class Month(Func): function = 'EXTRACT' template = '%(function)s(MONTH from %(expressions)s)' output_field = models.IntegerField()
After that all you need to do is
- annotate each row with the month
- group the results by the annotated month using
values()
- annotate each result with the aggregated sum of the totals using
Sum()
Important: if your model class has a default ordering specified in the meta options, then you will have to add an empty order_by()
clause. This is because of https://docs.djangoproject.com/en/1.9/topics/db/aggregation/#interaction-with-default-ordering-or-order-by
Fields that are mentioned in the
order_by()
part of a queryset (or which are used in the default ordering on a model) are used when selecting the output data, even if they are not otherwise specified in thevalues()
call. These extra fields are used to group “like” results together and they can make otherwise identical result rows appear to be separate.
If you are unsure, you could just add the empty order_by()
clause anyway without any adverse effects.
i.e.
from django.db.models import Sum summary = (Invoice.objects .annotate(m=Month('date')) .values('m') .annotate(total=Sum('total')) .order_by())
See the full gist here: https://gist.github.com/alvingonzales/ff9333e39d221981e5fc4cd6cdafdd17
If you need further information:
Details on creating your own Func classes: https://docs.djangoproject.com/en/1.8/ref/models/expressions/#func-expressions
Details on the values() clause, (pay attention to how it interacts with annotate() with respect to order of the clauses): https://docs.djangoproject.com/en/1.9/topics/db/aggregation/#values
the order in which annotate() and values() clauses are applied to a query is significant. If the values() clause precedes the annotate(), the annotation will be computed using the grouping described by the values() clause.