Fix mapping

This commit is contained in:
Beto Dealmeida
2025-12-11 10:58:17 -05:00
parent bfb7048e42
commit e253bd2fb3

View File

@@ -305,6 +305,7 @@ class SnowflakeSemanticView(SemanticViewImplementation):
""" """
Execute a query and return the results as a Pandas DataFrame. Execute a query and return the results as a Pandas DataFrame.
""" """
print("AM HERE")
if not metrics and not dimensions: if not metrics and not dimensions:
return DataFrame() return DataFrame()
@@ -321,6 +322,17 @@ class SnowflakeSemanticView(SemanticViewImplementation):
with connect(**connection_parameters) as connection: with connect(**connection_parameters) as connection:
df = connection.cursor().execute(query, parameters).fetch_pandas_all() df = connection.cursor().execute(query, parameters).fetch_pandas_all()
# map column names to dimension/metric names instead of IDs
mapping = {
**{dimension.id: dimension.name for dimension in dimensions},
**{metric.id: metric.name for metric in metrics},
}
print("BETO")
print(df.columns)
print(mapping)
df.rename(columns=mapping, inplace=True)
print(df.columns)
return SemanticResult( return SemanticResult(
requests=[ requests=[
SemanticRequest( SemanticRequest(