diff options
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 |