From 0365f767928da067c23367bb9cd7e98395f5302e Mon Sep 17 00:00:00 2001 From: Beto Dealmeida Date: Fri, 13 Aug 2021 18:21:20 -0700 Subject: [PATCH] fix: pivot columns with ints for name (#16259) (cherry picked from commit 9b2dffeb1dbda8a3565605fc2d02fa12065f89cf) --- superset/charts/post_processing.py | 2 +- superset/reports/commands/execute.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/superset/charts/post_processing.py b/superset/charts/post_processing.py index 5265557b633a..b80487e7a08b 100644 --- a/superset/charts/post_processing.py +++ b/superset/charts/post_processing.py @@ -177,7 +177,7 @@ def pivot_table_v2( # pylint: disable=too-many-branches df = df.stack(0).unstack().reindex(level=-1, columns=metrics) # flatten column names - df.columns = [" ".join(column) for column in df.columns] + df.columns = [" ".join(str(name) for name in column) for column in df.columns] return df diff --git a/superset/reports/commands/execute.py b/superset/reports/commands/execute.py index 347646a7cb96..91e4315b6be3 100644 --- a/superset/reports/commands/execute.py +++ b/superset/reports/commands/execute.py @@ -182,6 +182,7 @@ def _get_screenshot(self) -> bytes: screenshot: Optional[BaseScreenshot] = None if self._report_schedule.chart: url = self._get_url(standalone="true") + logger.info("Screenshotting chart at %s", url) screenshot = ChartScreenshot( url, self._report_schedule.chart.digest, @@ -190,6 +191,7 @@ def _get_screenshot(self) -> bytes: ) else: url = self._get_url() + logger.info("Screenshotting dashboard at %s", url) screenshot = DashboardScreenshot( url, self._report_schedule.dashboard.digest, @@ -235,6 +237,7 @@ def _get_csv_data(self) -> bytes: ) from ex try: + logger.info("Getting chart from %s", url) csv_data = get_chart_csv_data(url, auth_cookies) except SoftTimeLimitExceeded: raise ReportScheduleCsvTimeout()