Django 分组和计数

Question

我正在尝试对我的模型进行分组，如下所示：

class Restaurant(models.Model):
    pass

class Order(models.Model):
    created = models.DateTimeField(auto_now_add=True)
    restaurant = models.ForeignKey('Restaurant')

现在我想知道每天创建了多少订单。这意味着我需要从 DateTime 字段中获取日期，然后进行计数。

如果相关，我会以这种方式获取Order 查询集：

restaurant = Restaurant.objects.get(id=request.data['restaurant_id'])
orders = restaurant.order_set.filter(created__lte=some_date)

现在，我怎样才能从orders 获得我想要的东西？我试过这样的事情：

orders.values('created').annotate(Count('created'))

使用TruncDate等。

我正在使用 Python 2 和 Django 1.11。

编辑为了更好地表达我的意图。我想要一些我可以在纯 python 中实现的东西，如下所示：

orders_by_date = {}
for order in orders:
    if orders_by_date.get(datetime.date(orders[0].created.year, orders[0].created.month, orders[0].created.day):
        orders_by_date[datetime.date(orders[0].created.year, orders[0].created.month, orders[0].created.day)] += 1
    else:
        orders_by_date[datetime.date(orders[0].created.year, orders[0].created.month, orders[0].created.day)] = 1

EDIT2我能够成功显示单个日期的计数：

orders.filter(created__date=datetime.date(2018, 6, 8)).aggregate(Count('id'))

给我{'id__count': 3}。现在可以完美地按所有日期进行分组，而不仅仅是单个日期。

Answer 1

您应该能够执行orders.filter(created__date=datetime.date(2018, 6, 8)).values('id__count')，这将为您提供字典列表：[{'id__count': 3},{'id__count': 1},...]。

这里是docs

---

编辑： 尝试注释然后获取值：

orders_with_counts = orders.annotate(Count('id'))
dates_and_counts = orders_with_counts.values_list('created__date','id__count')
dct = dict(dates_and_counts)

dates_and_counts 将是一个元组列表：[(date1,count1),(date2,count2)...]。 dct 将被转换为字典。

Answer 2

您可以通过以下方式在一个查询中获取数据：

from django.db.models import DateField, Sum
from django.db.models.functions import Cast

query = Order.objects.filter(
    restaurant=some_restaurant
).annotate(
    create_date=Cast('created', DateField())
).values('create_date').annotate(
    id_count=Count('id')
).order_by('create_date')

这将返回一个QuerySet 的字典，例如：

<QuerySet [{'create_date': datetime.date(2017, 1, 31), 'id_count': 14}, 
           {'create_date': datetime.date(2017, 2, 2), 'id_count': 25}, 
           {'create_date': datetime.date(2017, 2, 9), 'id_count': 13},
           {'create_date': datetime.date(2017, 2, 10), 'id_count': 2}, 
           {'create_date': datetime.date(2017, 2, 16), 'id_count': 17},
           {'create_date': datetime.date(2017, 2, 17), 'id_count': 89}, 
           {'create_date': datetime.date(2017, 2, 20), 'id_count': 20},
           {'create_date': datetime.date(2017, 2, 23), 'id_count': 18}, 
           {'create_date': datetime.date(2017, 2, 24), 'id_count': 20},
           {'create_date': datetime.date(2017, 2, 28), 'id_count': 20}, 
           {'create_date': datetime.date(2017, 3, 1), 'id_count': 3},
           {'create_date': datetime.date(2017, 3, 3), 'id_count': 9}, 
           {'create_date': datetime.date(2017, 3, 7), 'id_count': 9},
           {'create_date': datetime.date(2017, 3, 9), 'id_count': 1}, 
           {'create_date': datetime.date(2017, 3, 10), 'id_count': 7}, 
           {'create_date': datetime.date(2017, 3, 14), 'id_count': 2}, 
           {'create_date': datetime.date(2017, 3, 15), 'id_count': 7},
           {'create_date': datetime.date(2017, 3, 17), 'id_count': 9}, 
           {'create_date': datetime.date(2017, 3, 22), 'id_count': 2},
           {'create_date': datetime.date(2017, 3, 24), 'id_count': 8},
           '...
           (remaining elements truncated)...']>

（添加格式）

在幕后，它会生成如下查询：

SELECT CAST(`order`.`created` AS date) AS `create_date`,
       COUNT(`order`.`id`) AS `id_count`
FROM `order`
WHERE `order`.`restaurant_id` = 123
GROUP BY CAST(`order`.`created` AS date)
ORDER BY `create_date` ASC

（123 是这里的示例餐厅 ID）。

因此，您可以例如迭代结果并构造 JSON 对象等。

例如，我们可以将其翻译成字典，通过 字典理解 遍历查询，将 datetime.date 对象映射到计数：

result = { t['create_date']: t['id_count'] for t in query }

请注意，没有任何Order 的日期不会成为查询集的一部分（也不是result 字典，这是合乎逻辑的，因为我们将Order 表作为“根”，如果没有行，则不会有输出）