django-cte 2.0.1.dev20251006193645__py3-none-any.whl → 2.0.1.dev20251027172651__py3-none-any.whl

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/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

{django_cte-2.0.1.dev20251006193645.dist-info → django_cte-2.0.1.dev20251027172651.dist-info}/METADATA

@@ -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.dist-info → django_cte-2.0.1.dev20251027172651.dist-info}/RECORD

@@ -1,12 +1,12 @@
-django_cte/__init__.py,sha256=gMYzdKjI_095EcobSVAElk83rrLsqHQMs9Bz-5ZOOa0,141
+django_cte/__init__.py,sha256=i9NSCeEHNvcjP4gHaBF3kZkBWjKPStC-rPC5YVAIV_k,141
 django_cte/_deprecated.py,sha256=OAEmWVqnoi-27m2nyrfTIwVwPE1J8VLrfspgRHW-EZI,5623
-django_cte/cte.py,sha256=3DnlrjZ8CDbClCwDGAI_LHFFK0mhP7Fiz_s1p9buam4,8214
+django_cte/cte.py,sha256=navhZjCcEqdSN9ICC2D8z12FqTPuftv8qOGvarxklg0,8892
 django_cte/jitmixin.py,sha256=JRTRlbBhD8XD0GHwxg8wKdJSbcTv_zUrSyqtOSfgM5M,885
 django_cte/join.py,sha256=2b1iBKFFiqBqYBmX4W9q2RfX17xXjZbbzClpUdbnUtA,3156
 django_cte/meta.py,sha256=gNBtegkvg5CXcEPyt1nZMIF2UlsPMTUPStVJSsRni3I,3491
 django_cte/query.py,sha256=djY7Bknl_NQSd1g6zy8N3fU3h2r8OfpyeJY_Zt9a7fg,5582
 django_cte/raw.py,sha256=6_YDCKBsgXx1oQku4EcyK4qJmnmiyzEeZRxiYdkg8Y8,1045
-django_cte-2.0.1.dev20251006193645.dist-info/licenses/LICENSE,sha256=mkLNw_QhpZ40jBEbuAosqH4ciA3KMrwb8aSYbTmy5gc,1508
-django_cte-2.0.1.dev20251006193645.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-django_cte-2.0.1.dev20251006193645.dist-info/METADATA,sha256=TZ4PjzPc6egcGj0xx0PT8e6SILuqvm1StuF7p5Oewos,2646
-django_cte-2.0.1.dev20251006193645.dist-info/RECORD,,
+django_cte-2.0.1.dev20251027172651.dist-info/licenses/LICENSE,sha256=mkLNw_QhpZ40jBEbuAosqH4ciA3KMrwb8aSYbTmy5gc,1508
+django_cte-2.0.1.dev20251027172651.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+django_cte-2.0.1.dev20251027172651.dist-info/METADATA,sha256=H0FKQAlA2Fh_KJY-9STA2Yrde2FFG-sC2LlHJXzJQ8E,2646
+django_cte-2.0.1.dev20251027172651.dist-info/RECORD,,