django-cte 2.0.1.dev20251006193645__tar.gz → 2.0.1.dev20251027172651__tar.gz

{django_cte-2.0.1.dev20251006193645 → django_cte-2.0.1.dev20251027172651}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-cte
-Version: 2.0.1.dev20251006193645
+Version: 2.0.1.dev20251027172651
 Summary: Common Table Expressions (CTE) for Django
 Author-email: Daniel Miller <millerdev@gmail.com>
 Requires-Python: >= 3.9

{django_cte-2.0.1.dev20251006193645 → django_cte-2.0.1.dev20251027172651}/django_cte/__init__.py

@@ -1,4 +1,4 @@
 from .cte import CTE, with_cte, CTEManager, CTEQuerySet, With  # noqa
 
-__version__ = "2.0.1.dev20251006193645"
+__version__ = "2.0.1.dev20251027172651"
 __all__ = ["CTE", "with_cte"]

{django_cte-2.0.1.dev20251006193645 → django_cte-2.0.1.dev20251027172651}/django_cte/cte.py

@@ -1,5 +1,6 @@
 from copy import copy
 
+import django
 from django.db.models import Manager, sql
 from django.db.models.expressions import Ref
 from django.db.models.query import Q, QuerySet, ValuesIterable
@@ -45,21 +46,30 @@ class CTE:
     """
 
     def __init__(self, queryset, name="cte", materialized=False):
-        self.query = None if queryset is None else queryset.query
+        self._set_queryset(queryset)
         self.name = name
         self.col = CTEColumns(self)
         self.materialized = materialized
 
     def __getstate__(self):
-        return (self.query, self.name, self.materialized)
+        return (self.query, self.name, self.materialized, self._iterable_class)
 
     def __setstate__(self, state):
-        self.query, self.name, self.materialized = state
+        if len(state) == 3:
+            # Keep compatibility with the previous serialization method
+            self.query, self.name, self.materialized = state
+            self._iterable_class = ValuesIterable
+        else:
+            self.query, self.name, self.materialized, self._iterable_class = state
         self.col = CTEColumns(self)
 
     def __repr__(self):
         return f"<{type(self).__name__} {self.name}>"
 
+    def _set_queryset(self, queryset):
+        self.query = None if queryset is None else queryset.query
+        self._iterable_class = getattr(queryset, "_iterable_class", ValuesIterable)
+
     @classmethod
     def recursive(cls, make_cte_queryset, name="cte", materialized=False):
         """Recursive Common Table Expression
@@ -73,7 +83,7 @@ class CTE:
         :returns: The fully constructed recursive cte object.
         """
         cte = cls(None, name, materialized)
-        cte.query = make_cte_queryset(cte).query
+        cte._set_queryset(make_cte_queryset(cte))
         return cte
 
     def join(self, model_or_queryset, *filter_q, **filter_kw):
@@ -124,24 +134,30 @@ class CTE:
         """
         cte_query = self.query
         qs = cte_query.model._default_manager.get_queryset()
+        qs._iterable_class = self._iterable_class
+        qs._fields = ()  # Allow any field names to be used in further annotations
 
         query = jit_mixin(sql.Query(cte_query.model), CTEQuery)
         query.join(BaseTable(self.name, None))
         query.default_cols = cte_query.default_cols
         query.deferred_loading = cte_query.deferred_loading
-        if cte_query.values_select:
+
+        if django.VERSION < (5, 2) and cte_query.values_select:
             query.set_values(cte_query.values_select)
-            qs._iterable_class = ValuesIterable
+
         if cte_query.annotations:
             for alias, value in cte_query.annotations.items():
                 col = CTEColumnRef(alias, self.name, value.output_field)
                 query.add_annotation(col, alias)
         query.annotation_select_mask = cte_query.annotation_select_mask
-        for alias in getattr(cte_query, "selected", None) or ():
-            if alias not in cte_query.annotations:
-                output_field = cte_query.resolve_ref(alias).output_field
-                col = CTEColumnRef(alias, self.name, output_field)
-                query.add_annotation(col, alias)
+
+        if selected := getattr(cte_query, "selected", None):
+            for alias in selected:
+                if alias not in cte_query.annotations:
+                    output_field = cte_query.resolve_ref(alias).output_field
+                    col = CTEColumnRef(alias, self.name, output_field)
+                    query.add_annotation(col, alias)
+            query.selected = {alias: alias for alias in selected}
 
         qs.query = query
         return qs