如何按日期过滤以从 QuerySet 中获取本周与前一周的计数？

Question

我有以下两个带注释的查询集，它们是通过从数据集中提取this_wk 和prev_wk 分别生成的，然后执行注释。我想在归还它们之前将它们合并在一起。

data = {
    'this_wk': QuerySet([{'this_total': 3, 'dept': 2}, 
                         {'this_total': 2, 'dept': 1}, 
                         {'this_total': 1, 'dept': 3}]),
    'prev_wk': QuerySet([{'prev_total': 2, 'dept': 3}])}

预期：

QuerySet([{'this_total': 3, 'prev_total': None, 'dept': 2},
          {'this_total': 2, 'prev_total': None, 'dept': 1}, 
          {'this_total': 1, 'prev_total': 2, 'dept': 3}]),

或者，有没有办法直接聚合它，以便我直接从一组带有日期字段的项目中获取？以下是我目前生成两个 QuerySet 的方式：

previous_items = Widget.objects.filter(start__gte=prev_start).filter(start__lte=prev_end)
prev_ranking = previous_items.values('dept').annotate(prev_total=Count('dept'))

this_items = Widget.objects.filter(start__gte=this_start
    ).filter(start__lte=this_end)
this_ranking = this_items.values('dept'
    ).annotate(this_total=Count('dept')
    ).order_by('-this_total')

有没有办法将这两个注释链接在一起，以便我最终得到一个单独的 QuerySet，根据本周使用的 Dept 对每个 Dept 进行分组和排名，但还添加了一个字段这表示前一周使用了多少Widgets？

我在想这样的事情：

items = Widget.objects.filter(start__gte=prev_start
    ).filter(start__lte=this_end
    ).annotate(prev_total=Count('dept')  # can I restrict this to the previous week?
    ).annotate(this_total=Count('dept')  # can I restrict this to the current week?
    ).order_by('-this_total')

Answer 1

查看条件聚合：

https://docs.djangoproject.com/en/1.10/ref/models/conditional-expressions/#conditional-aggregation

items = (Widget.objects.filter(start__range=(prev_start, this_end))
         .values('dept')
         .annotate(prev_total=Count(Case(When(start__lte=prev_end, then='dept'), output_field=IntegerField())),
                   this_total=Count(Case(When(start__gte=this_start, then='dept'), output_field=IntegerField()))
                   )
         .order_by('-this_total'))

Answer 2

您的想法似乎对您的情况是正确的。为了实现您的需要，您可以应用以下内容：

1. range 查找功能虽小但很有帮助，可让您在两个值之间进行过滤
2. 要限制annotate 参数，您需要使用conditional aggregation。

最后，您的查询将如下所示：

items = Widget.objects.filter(start__range=(prev_start, this_end))
                      .values('dept')
                      .annotate(
                           prev_total=Count(
                               Case(
                                   When(start__lte=prev_end, then=1)),                 
                                   output_field=IntegerField())),
                           this_total=Count(
                               Case(
                                   When(start__gte=this_start, then=1), 
                                   output_field=IntegerField())))
                       .order_by('-this_total')

---

因评论而编辑：

在这种情况下不需要此步骤，这将有助于构建一个注释来对过滤后的 dept 字段的值求和。

1. 我会使用F() 表达式来获取注解中每个dept 字段的值，并使用Sum 函数而不是Count 将这些值相加。