This is an automated email from the ASF dual-hosted git repository. beto pushed a commit to branch sc-76910 in repository https://gitbox.apache.org/repos/asf/superset.git
commit bd26ad0c68cbceff0a6eaab8c453a2cb56c0af40 Author: Beto Dealmeida <[email protected]> AuthorDate: Tue Nov 28 13:17:58 2023 -0500 fix: alias column when fetching values --- superset/models/helpers.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/superset/models/helpers.py b/superset/models/helpers.py index fc947ff577..df3dd93488 100644 --- a/superset/models/helpers.py +++ b/superset/models/helpers.py @@ -1352,7 +1352,13 @@ class ExploreMixin: # pylint: disable=too-many-public-methods tbl, cte = self.get_from_clause(tp) qry = ( - sa.select([target_col.get_sqla_col(template_processor=tp)]) + sa.select( + # The alias (label) here is important because some dialects will + # automatically add a random alias to the projection because of the + # call to DISTINCT; others will uppercase the column names. This + # gives us a deterministic column name in the dataframe. + [target_col.get_sqla_col(template_processor=tp).label("column_values")] + ) .select_from(tbl) .distinct() ) @@ -1368,7 +1374,7 @@ class ExploreMixin: # pylint: disable=too-many-public-methods sql = self.mutate_query_from_config(sql) df = pd.read_sql_query(sql=sql, con=engine) - return df[denormalized_col_name].to_list() + return df["column_values"].to_list() def get_timestamp_expression( self,
