Merge pull request #90 from Mogost/deferred-loading

Fix deferred loading
dimagi · Apr 29, 2024 · 3d4fffe · 3d4fffe
2 parents 88843ef + 42d29f7
commit 3d4fffe
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/django_cte/cte.py b/django_cte/cte.py
@@ -109,6 +109,7 @@ def queryset(self):
         query = CTEQuery(cte_query.model)
         query.join(BaseTable(self.name, None))
         query.default_cols = cte_query.default_cols
+        query.deferred_loading = cte_query.deferred_loading
         if cte_query.annotations:
             for alias, value in cte_query.annotations.items():
                 col = CTEColumnRef(alias, self.name, value.output_field)

diff --git a/tests/test_manager.py b/tests/test_manager.py
@@ -104,3 +104,12 @@ def test_cte_queryset_with_custom_queryset(self):
             ('venus', 22, 'sun'),
             ('venus', 23, 'sun'),
         ])
+
+    def test_cte_queryset_with_deferred_loading(self):
+        cte = With(
+            OrderCustomManagerNQuery.objects.order_by("id").only("id")[:1]
+        )
+        orders = cte.queryset().with_cte(cte)
+        print(orders.query)
+
+        self.assertEqual([x.id for x in orders], [1])