Django ORM : is it possible to inject subqueries?

Question

我有一个看起来像这样的 Django 模型:

class Result(models.Model):
    date = DateTimeField()
    subject = models.ForeignKey('myapp.Subject')
    test_type = models.ForeignKey('myapp.TestType')
    summary = models.PositiveSmallIntegerField()
    # more fields about the result like its location, tester ID and so on

有时我们想要检索所有测试结果，有时我们只想要每个科目的特定测试类型的最新结果。 This answer有一些很棒的 SQL 选项可以找到最新的结果。

此外，我们有时希望将结果分成不同的时间 block ，以便我们可以绘制每天/每周/每月的结果数量图表。

我们还想过滤各种字段，为了优雅起见，我想要一个 QuerySet，然后我可以对其进行所有 filter() 调用，并对计数进行注释，而不是进行原始 SQL 调用。

我已经走到这一步了:

qs = Result.objects.extra(select = {
         'date_range': "date_trunc('{0}', time)".format("day"), # Chunking into time buckets
         'rn' : "ROW_NUMBER() OVER(PARTITION BY subject_id, test_type_id ORDER BY time DESC)"})
qs = qs.values('date_range', 'result_summary', 'rn')
qs = qs.order_by('-date_range')

这会产生以下 SQL:

SELECT (ROW_NUMBER() OVER(PARTITION BY subject_id, test_type_id ORDER BY time DESC)) AS "rn", (date_trunc('day', time)) AS "date_range", "myapp_result"."result_summary" FROM "myapp_result" ORDER BY "date_range" DESC

这有点接近我想要的，但现在我需要以某种方式过滤以仅获取 rn = 1 的行。我尝试在 extra() 中使用“where”字段，这给了我以下 SQL和错误:

SELECT (ROW_NUMBER() OVER(PARTITION BY subject_id, test_type_id ORDER BY time DESC)) AS "rn", (date_trunc('day', time)) AS "date_range", "myapp_result"."result_summary" FROM "myapp_result" WHERE "rn"=1 ORDER BY "date_range" DESC                                                                                                                            ;
ERROR:  column "rn" does not exist

所以我认为找到“rn”的查询需要是一个子查询 - 但是否可以以某种方式做到这一点，也许使用 extra() ？

我知道我可以使用原始 SQL 来做到这一点，但它看起来很丑!我很想找到一种简洁的方法来获得可过滤的查询集。

我想另一个选择是在模型中包含一个字段，指示它是否实际上是该主题的该测试类型的最新结果......

Answer 1

我找到办法了!

qs = Result.objects.extra(where = ["NOT EXISTS(SELECT * FROM myapp_result as T2 WHERE (T2.test_type_id = myapp_result.test_type_id AND T2.subject_id = myapp_result.subject ID AND T2.time > myapp_result.time))"])

这基于与 the answer I referenced earlier 不同的选项。我可以用我想要的任何内容过滤或注释 qs。

---

顺便说一句，在解决这个问题的过程中我尝试了这个:

qq = Result.objects.extra(where = ["NOT EXISTS(SELECT * FROM myapp_result as T2 WHERE (T2.test_type_id = myapp_result.test_type_id AND T2.subject_id = myapp_result.subject ID AND T2.time > myapp_result.time))"])
qs = Result.objects.filter(id__in=qq)

Django 按照您的需要嵌入子查询:

SELECT ...some fields... FROM "myapp_result" 
WHERE ("myapp_result"."id" IN (SELECT "myapp_result"."id" FROM "myapp_result" 
WHERE (NOT EXISTS(SELECT * FROM myapp_result as T2 
WHERE (T2.subject_id = myapp_result.subject_id AND T2.test_type_id = myapp_result.test_type_id AND T2.time > myapp_result.time))))) 

我意识到这有比我需要的更多的子查询，但我在这里注意到它，因为我可以想象知道你可以用另一个查询集过滤一个查询集并且 Django 在嵌入查询集方面完全符合你的期望是有用的子查询(而不是执行它并嵌入返回值，这将是可怕的。)