Django：按月查询组

Question

如何在不使用额外的情况下按月计算总数？

我目前正在使用：

• django 1.8
• postgre 9.3.13
• Python 2.7

示例。

到目前为止我所尝试的。

#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"))

还有这个，答案似乎很好，但我无法导入 TruncMonth 模块。

Django: Group by date (day, month, year)

---

附：我知道这个问题已经被问过多次，但我没有看到任何答案。

谢谢！

---

解决方案：

感谢@Vin-G 的回答。

Answer 1

我不知道我的解决方案是否比你的更快。你应该分析它。尽管如此，我只查询数据库一次而不是 12 次。

#utils.py
from django.db.models import Count, Sum


def get_total_per_month_value():
    """
    Return the total of sales per month

    ReturnType: [Dict]
    {'December': 3400, 'February': 224, 'January': 792}
    """
    result= {}
    db_result = Sale.objects.values('price','created')
    for i in db_result:
        month = str(i.get('created').strftime("%B"))
        if month in result.keys():
            result[month] = result[month] + i.get('price')
        else:
            result[month] = i.get('price')
    return result

#models.py
class Sale(models.Model):
    price = models.PositiveSmallIntegerField()
    created = models.DateTimeField(_(u'Published'), default="2001-02-24")

#views.py
from .utils import get_total_per_month_value
# ...
result = get_total_per_month_value()

test.py

  #
    import pytest
    from mixer.backend.django import mixer
    #Don't try to write in the database
    pytestmark = pytest.mark.django_db
    def test_get_total_per_month():
        from .utils import get_total_per_month_value
        selected_date = ['01','02','03','01','01']
        #2016-01-12 == YYYY-MM-DD
        for i in selected_date:
            mixer.blend('myapp.Sale', created="2016-"+i+"-12")
        values = get_total_per_month_value() #return a dict
        months = values.keys()
        assert 'January' in months, 'Should include January'
        assert 'February' in months, 'Should include February'
        assert len(months) == 3, 'Should aggregate the months'

Answer 2

itertools.groupby 是 Python 中的高性能选项，可用于单个数据库查询：

from itertools import groupby

invoices = Invoice.objects.only('date', 'total').order_by('date')
month_totals = {
    k: sum(x.total for x in g) 
    for k, g in groupby(invoices, key=lambda i: i.date.month)
}
month_totals
# {1: 100, 3: 100, 4: 500, 7: 500}

我不知道纯 django ORM 解决方案。 date__month 过滤器非常有限，不能用于values、order_by 等。

Answer 3

首先，您必须创建一个可以为您提取月份的函数：

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()

之后你需要做的就是

1. 用月份注释每一行
2. 使用values()按带注释的月份对结果进行分组
3. 使用Sum()用总计的总和来注释每个结果

重要提示：如果您的模型类在元选项中指定了默认排序，那么您将必须添加一个空的 order_by() 子句。这是因为https://docs.djangoproject.com/en/1.9/topics/db/aggregation/#interaction-with-default-ordering-or-order-by

查询集的order_by() 部分中提到的字段（或在模型的默认排序中使用的字段）在选择输出数据时使用，即使在values() 调用中未另行指定它们也是如此.这些额外的字段用于将“喜欢”的结果组合在一起，它们可以使原本相同的结果行看起来是分开的。

如果您不确定，您可以添加空的 order_by() 子句，而不会产生任何不利影响。

即

from django.db.models import Sum

summary = (Invoice.objects
              .annotate(m=Month('date'))
              .values('m')
              .annotate(total=Sum('total'))
              .order_by())

在此处查看完整要点：https://gist.github.com/alvingonzales/ff9333e39d221981e5fc4cd6cdafdd17

如果您需要更多信息：

有关创建您自己的 Func 类的详细信息：https://docs.djangoproject.com/en/1.8/ref/models/expressions/#func-expressions

values() 子句的详细信息，（注意它如何与 annotate() 交互关于子句的顺序）： https://docs.djangoproject.com/en/1.9/topics/db/aggregation/#values

annotate() 和 values() 子句应用于查询的顺序很重要。如果 values() 子句在 annotate() 之前，注释将使用 values() 子句描述的分组计算。

Answer 4

不要忘记 Django 查询集提供了一个原生的datetimes manager，它可以让您轻松地从任何带有日期时间字段的模型的查询集中提取所有天/周/月/年。因此，如果上面的 Invoice 模型有一个 created 日期时间字段，并且您想要查询集中每个月的总计，您可以这样做：

    invoices = Invoice.objects.all()
    months = invoices.datetimes("created", kind="month")
    for month in months:
        month_invs = invoices.filter(created__month=month.month)
        month_total = month_invs.aggregate(total=Sum("otherfield")).get("total")
        print(f"Month: {month}, Total: {month_total}")

不需要外部功能或部门。

Answer 5

result = (
    invoices.objects
        .all()
        .values_list('created_at__year', 'created_at__month')
        .annotate(Sum('total'))
        .order_by('created_at__year', 'created_at__month')
)