diff options
| author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-09-25 08:20:09 +0000 |
|---|---|---|
| committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-09-25 08:20:09 +0000 |
| commit | 4554ab4c7d6b2bbbaa6f4d0b810bf477d1a505a6 (patch) | |
| tree | 8f4f60a82ab9cd6dcd41397e4ecb2960c332b209 /sqlglot/dataframe/sql | |
| parent | Releasing debian version 18.5.1-1. (diff) | |
| download | sqlglot-4554ab4c7d6b2bbbaa6f4d0b810bf477d1a505a6.tar.xz sqlglot-4554ab4c7d6b2bbbaa6f4d0b810bf477d1a505a6.zip | |
Merging upstream version 18.7.0.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'sqlglot/dataframe/sql')
| -rw-r--r-- | sqlglot/dataframe/sql/column.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/sqlglot/dataframe/sql/column.py b/sqlglot/dataframe/sql/column.py index 3acf494..ca85376 100644 --- a/sqlglot/dataframe/sql/column.py +++ b/sqlglot/dataframe/sql/column.py @@ -212,7 +212,15 @@ class Column: return self.expression.sql(**{"dialect": SparkSession().dialect, **kwargs}) def alias(self, name: str) -> Column: - new_expression = exp.alias_(self.column_expression, name) + from sqlglot.dataframe.sql.session import SparkSession + + dialect = SparkSession().dialect + alias: exp.Expression = sqlglot.maybe_parse(name, dialect=dialect) + new_expression = exp.alias_( + self.column_expression, + alias.this if isinstance(alias, exp.Column) else name, + dialect=dialect, + ) return Column(new_expression) def asc(self) -> Column: |