From c59649ac5b7d4e78038221d32a91c0ba36db3887 Mon Sep 17 00:00:00 2001 From: Beto Dealmeida Date: Fri, 15 Sep 2023 12:13:13 -0700 Subject: [PATCH 1/2] fix: datetime with timezone excel export --- superset/utils/excel.py | 5 +++++ tests/unit_tests/utils/excel_tests.py | 31 +++++++++++++++++++++++++++ 2 files changed, 36 insertions(+) create mode 100644 tests/unit_tests/utils/excel_tests.py diff --git a/superset/utils/excel.py b/superset/utils/excel.py index 1f68031b6497..2119d20926c8 100644 --- a/superset/utils/excel.py +++ b/superset/utils/excel.py @@ -22,6 +22,11 @@ def df_to_excel(df: pd.DataFrame, **kwargs: Any) -> Any: output = io.BytesIO() + + # timezone are not supported + for column in df.select_dtypes(include=["datetimetz"]).columns: + df[column] = df[column].astype(str) + # pylint: disable=abstract-class-instantiated with pd.ExcelWriter(output, engine="xlsxwriter") as writer: df.to_excel(writer, **kwargs) diff --git a/tests/unit_tests/utils/excel_tests.py b/tests/unit_tests/utils/excel_tests.py new file mode 100644 index 000000000000..c15f69a0c62a --- /dev/null +++ b/tests/unit_tests/utils/excel_tests.py @@ -0,0 +1,31 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +# KIND, either express or implied. See the License for the +# specific language governing permissions and limitations +# under the License. + +from datetime import datetime, timezone + +import pandas as pd + +from superset.utils.excel import df_to_excel + + +def test_timezone_conversion() -> None: + """ + Test that columns with timezones are converted to a string. + """ + df = pd.DataFrame({"dt": [datetime(2023, 1, 1, 0, 0, tzinfo=timezone.utc)]}) + contents = df_to_excel(df) + assert pd.read_excel(contents)["dt"][0] == "2023-01-01 00:00:00+00:00" From de73f6fdec6ac0c00cb8cab4f9e4a9b961e8bad7 Mon Sep 17 00:00:00 2001 From: "Michael S. Molina" <70410625+michael-s-molina@users.noreply.github.com> Date: Mon, 18 Sep 2023 19:20:04 -0300 Subject: [PATCH 2/2] Fix typo --- superset/utils/excel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/superset/utils/excel.py b/superset/utils/excel.py index 2119d20926c8..ccbeadee5ece 100644 --- a/superset/utils/excel.py +++ b/superset/utils/excel.py @@ -23,7 +23,7 @@ def df_to_excel(df: pd.DataFrame, **kwargs: Any) -> Any: output = io.BytesIO() - # timezone are not supported + # timezones are not supported for column in df.select_dtypes(include=["datetimetz"]).columns: df[column] = df[column].astype(str)