diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-01-30 17:08:37 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-01-30 17:08:37 +0000 |
commit | be1cb18ea28222fca384a5459a024b7e9af5cadb (patch) | |
tree | 4698c9069380a7c30ceb51129f93f6c8662315e4 /sqlglot/dataframe/sql/dataframe.py | |
parent | Releasing debian version 10.5.6-1. (diff) | |
download | sqlglot-be1cb18ea28222fca384a5459a024b7e9af5cadb.tar.xz sqlglot-be1cb18ea28222fca384a5459a024b7e9af5cadb.zip |
Merging upstream version 10.5.10.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'sqlglot/dataframe/sql/dataframe.py')
-rw-r--r-- | sqlglot/dataframe/sql/dataframe.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/sqlglot/dataframe/sql/dataframe.py b/sqlglot/dataframe/sql/dataframe.py index a17bb9d..65a37f5 100644 --- a/sqlglot/dataframe/sql/dataframe.py +++ b/sqlglot/dataframe/sql/dataframe.py @@ -111,16 +111,13 @@ class DataFrame: return DataFrameNaFunctions(self) def _replace_cte_names_with_hashes(self, expression: exp.Select): - expression = expression.copy() - ctes = expression.ctes replacement_mapping = {} - for cte in ctes: + for cte in expression.ctes: old_name_id = cte.args["alias"].this new_hashed_id = exp.to_identifier( self._create_hash_from_expression(cte.this), quoted=old_name_id.args["quoted"] ) replacement_mapping[old_name_id] = new_hashed_id - cte.set("alias", exp.TableAlias(this=new_hashed_id)) expression = expression.transform(replace_id_value, replacement_mapping) return expression @@ -183,7 +180,7 @@ class DataFrame: expression = df.expression hint_expression = expression.args.get("hint") or exp.Hint(expressions=[]) for hint in df.pending_partition_hints: - hint_expression.args.get("expressions").append(hint) + hint_expression.append("expressions", hint) df.pending_hints.remove(hint) join_aliases = { @@ -209,7 +206,7 @@ class DataFrame: sequence_id_expression.set("this", matching_cte.args["alias"].this) df.pending_hints.remove(hint) break - hint_expression.args.get("expressions").append(hint) + hint_expression.append("expressions", hint) if hint_expression.expressions: expression.set("hint", hint_expression) return df |