Django CaptureQueriesContext无法捕获SELECT查询SQL的问题及原因

核心原因 #

Django的QuerySet是惰性求值的：

• 执行all_objects = TestTableTwo.objects.all()时，只是创建了一个QuerySet对象，并没有真正触发数据库查询，所以不会生成SELECT语句被CaptureQueriesContext捕获。
• 而print(TestTableTwo.objects.all())会强制QuerySet执行求值（因为要把对象转为字符串输出），这时候才会向数据库发送SELECT请求，自然能被捕获。

捕获所有SQL语句的方法 #

要让所有类型的SQL（CREATE、SELECT、UPDATE等）都被记录，需要确保QuerySet被实际执行。常见触发求值的方式有：

• 用list()把QuerySet转为列表：list(TestTableTwo.objects.all())
• 遍历QuerySet：for obj in TestTableTwo.objects.all(): pass
• 调用exists()、count()这类直接返回结果的方法：TestTableTwo.objects.count()
• 访问QuerySet的具体元素：TestTableTwo.objects.all()[0]

另外，要确保所有ORM操作都在CaptureQueriesContext的上下文范围内完成，示例测试用例结构如下：

from django.test import TestCase, CaptureQueriesContext
from django.db import connection
from myapp.models import TestTableTwo

class MyTestCase(TestCase):
    def test_capture_all_queries(self):
        with CaptureQueriesContext(connection) as ctx:
            # INSERT操作会立即触发数据库查询
            TestTableTwo.objects.create(name="test_data")
            # 触发SELECT查询的求值
            all_objects = list(TestTableTwo.objects.all())
            
            # 输出所有捕获的SQL语句
            for query in ctx.captured_queries:
                print(query['sql'])

内容的提问来源于stack exchange，提问作者Ewro