From 770d50092f847ef7e79a895c72a680f7626a1164 Mon Sep 17 00:00:00 2001 From: Eric Masseran Date: Fri, 6 Aug 2021 04:11:09 -0400 Subject: Dispatch independent ctes on compound select Fix issue in :class:`_sql.CTE` where new :meth:`_sql.HasCTE.add_cte` method added in version 1.4.21 / :ticket:`6752` failed to function correctly for "compound select" structures such as :func:`_sql.union`, :func:`_sql.union_all`, :func:`_sql.except`, etc. Pull request courtesy Eric Masseran. Fixes: #6752 Closes: #6849 Pull-request: https://github.com/sqlalchemy/sqlalchemy/pull/6849 Pull-request-sha: 1c4b4d72b2789cf89ff5043ca964ebdd6c9a6617 Change-Id: I49a16a4fc2af8299502011f3a02d8a2ad93255e3 --- lib/sqlalchemy/sql/compiler.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'lib/sqlalchemy/sql/compiler.py') diff --git a/lib/sqlalchemy/sql/compiler.py b/lib/sqlalchemy/sql/compiler.py index a81507acb..1e090ef10 100644 --- a/lib/sqlalchemy/sql/compiler.py +++ b/lib/sqlalchemy/sql/compiler.py @@ -1787,6 +1787,8 @@ class SQLCompiler(Compiled): if toplevel and not self.compile_state: self.compile_state = compile_state + compound_stmt = compile_state.statement + entry = self._default_stack_entry if toplevel else self.stack[-1] need_result_map = toplevel or ( not compound_index @@ -1807,6 +1809,10 @@ class SQLCompiler(Compiled): } ) + if compound_stmt._independent_ctes: + for cte in compound_stmt._independent_ctes: + cte._compiler_dispatch(self, **kwargs) + keyword = self.compound_keywords.get(cs.keyword) text = (" " + keyword + " ").join( -- cgit v1.2.1