From 8f0b432c5a096917f7c252021f863d9bdffd7fbb Mon Sep 17 00:00:00 2001
From: Alexander Schepanovski <suor.web@gmail.com>
Date: Mon, 29 Jul 2024 11:43:11 +0700
Subject: [PATCH] Prevent dups in conjs and disjs by using proper
 data-structures

Related to #480
---
 cacheops/tree.py | 53 +++++++++++++++++++++++++++---------------------
 1 file changed, 30 insertions(+), 23 deletions(-)

diff --git a/cacheops/tree.py b/cacheops/tree.py
index ef42e30b..cebf9376 100644
--- a/cacheops/tree.py
+++ b/cacheops/tree.py
@@ -1,5 +1,5 @@
 from itertools import product
-from funcy import group_by, join_with, lcat, lmap
+from funcy import group_by, join_with, lcat, lmap, cat
 
 from django.db.models import Subquery
 from django.db.models.query import QuerySet
@@ -23,8 +23,8 @@ def dnfs(qs):
     conditions on joined models and subrequests are ignored.
     __in is converted into = or = or = ...
     """
-    SOME = object()
-    SOME_TREE = [[(None, None, SOME, True)]]
+    SOME = Some()
+    SOME_TREE = {frozenset({(None, None, SOME, True)})}
 
     def negate(term):
         return (term[0], term[1], term[2], not term[3])
@@ -51,37 +51,37 @@ def _dnf(where):
 
             attname = where.lhs.target.attname
             if isinstance(where, Exact):
-                return [[(where.lhs.alias, attname, where.rhs, True)]]
+                return {frozenset({(where.lhs.alias, attname, where.rhs, True)})}
             elif isinstance(where, IsNull):
-                return [[(where.lhs.alias, attname, None, where.rhs)]]
+                return {frozenset({(where.lhs.alias, attname, None, where.rhs)})}
             elif isinstance(where, In) and len(where.rhs) < settings.CACHEOPS_LONG_DISJUNCTION:
-                return [[(where.lhs.alias, attname, v, True)] for v in where.rhs]
+                return {frozenset({(where.lhs.alias, attname, v, True)}) for v in where.rhs}
             else:
                 return SOME_TREE
         elif isinstance(where, NothingNode):
-            return []
+            return set()
         elif isinstance(where, (ExtraWhere, SubqueryConstraint, Exists)):
             return SOME_TREE
         elif len(where) == 0:
-            return [[]]
+            return {frozenset()}
         else:
-            chilren_dnfs = lmap(_dnf, where.children)
+            children_dnfs = lmap(_dnf, where.children)
 
-            if len(chilren_dnfs) == 0:
-                return [[]]
-            elif len(chilren_dnfs) == 1:
-                result = chilren_dnfs[0]
+            if len(children_dnfs) == 0:
+                return {frozenset()}
+            elif len(children_dnfs) == 1:
+                result = children_dnfs[0]
             else:
                 # Just unite children joined with OR
                 if where.connector == OR:
-                    result = lcat(chilren_dnfs)
+                    result = set(cat(children_dnfs))
                 # Use Cartesian product to AND children
                 else:
-                    result = lmap(lcat, product(*chilren_dnfs))
+                    result = {frozenset(cat(conjs)) for conjs in product(*children_dnfs)}
 
             # Negating and expanding brackets
             if where.negated:
-                result = [lmap(negate, p) for p in product(*result)]
+                result = {frozenset(map(negate, conjs)) for conjs in product(*result)}
 
             return result
 
@@ -119,22 +119,23 @@ def add_join_conds(dnf, query):
                     join_exts[join.table_alias, target_col].append((join.parent_alias, parent_col))
 
         if not join_exts:
-            return
+            return dnf
 
-        for conj in dnf:
-            # NOTE: using list comprehension over genexp here since we change the thing we iterate
-            conj.extend([
+        return {
+            conj | {
                 (join_alias, join_col, v, negation)
                 for alias, col, v, negation in conj
-                for (join_alias, join_col) in join_exts[alias, col]
-            ])
+                for join_alias, join_col in join_exts[alias, col]
+            }
+            for conj in dnf
+        }
 
     def query_dnf(query):
         def table_for(alias):
             return alias if alias == main_alias else query.alias_map[alias].table_name
 
         dnf = _dnf(query.where)
-        add_join_conds(dnf, query)
+        dnf = add_join_conds(dnf, query)
 
         # NOTE: we exclude content_type as it never changes and will hold dead invalidation info
         main_alias = query.model._meta.db_table
@@ -156,3 +157,9 @@ def table_for(alias):
         dnfs_.update(join_with(lcat, subqueries))
 
     return dnfs_
+
+
+class Some:
+    def __str__(self):
+        return 'SOME'
+    __repr__ = __str__