python - Django 模型查询总和为给定数字的前 N ​​行

Question

我唱Django 3.2

我有一个这样的模型:

class MyModel(models.Model):
    name = models.CharField(max_length=255)
    num_visits  = models.PositiveSmallNumber()
    created_at  = models.DateTimeField()

    class Meta:
        ordering = ['created_at']

我想选择前 N 行，其中访问次数的累积总和 = 100(作为示例) 一个简单的(但可能是数据库密集型的方法)是迭代表中的行(即查询集的元素) - 但我不想这样做 - 原因应该是显而易见的。

如何编写一个查询来获取访问次数总和为指定数字的前 N ​​行？

def get_rows_satisfying_visit_count(cumulative_sum_in_field=100):
    MyModel.objects.filter(/* what ? */)

Answer 1

您可以对查询集使用切片运算符:

def get_rows_satisfying_visit_count(visits=100, rows=100):
    return MyModel.objects.filter(num_visits=number)[:rows]

编辑: 通过更新的问题，从 Django >= 2.0 开始，您可以使用 Window 和 Sum 来完成此操作:

MyModel.objects.annotate(
    cumulative_sum=Window(
        Sum('num_visits'),
        order_by=F('id').asc()
    )
).filter(cumulative_sum__lte=100)

这将获取累积和小于或等于 100 的前几行。