与 django 查询集注释中的先前对象的区别

Question

假设我有以下模型：

class TestModel(models.Model):
    some_integer = models.IntegerField()

我有这个模型的 3 个实例：

TestModel.objects.create(some_integer=100)
TestModel.objects.create(some_integer=50)
TestModel.objects.create(some_integer=20)

我想以某种方式对查询集进行注释以便能够获得以下结果：

for obj in TestModel.objects.annotate(difference=...):
    print(obj.difference)
=> 50  # 100 - 50
=> 30  # 50 - 30
=> None  # we don't have anything created after this record

是否有机会使用 Django(2.0) 查询集执行此操作，或者我应该在 Python 中“手动”执行此操作？我们可以放心地假设对象是按 pk 或某个日期排序的，因此保留了排序。

Answer 1

基于这个Answer，如果你想使用原始SQL，你可以这样做：

SELECT *,
       LEAD("some_integer") OVER(ORDER BY "id") AS "next_val",
       "some_integer" - "next_val" AS "difference"
FROM "myapp_testmodel";

但是，从 Django 版本 2.0 开始，您可以使用 Window Functions 创建上述查询：

from django.db.models import Window, F
from django.db.models.functions import Lead

q = TestModel.objects.annotate(
    next_val=Window(
        expression=Lead('some_integer', offset=1, default=0),
        order_by=F('id').asc()
    ), 
    difference=F('some_integer')-F('next_val')
)

注意：我已经使用0作为默认值（将应用于最后一项），以简化答案以防止从Integer中减去Null值引起FieldError。您可以使用 Django 的 Case 查询为最后一项返回 None 。