better timezone support

Author: aersam

bmsdna/table_rendering/table_rendering.py

@@ -153,12 +153,16 @@ def render_into_sheet(
 def create_excel(
     sheets: "Mapping[str, tuple[TableRenderer, list[dict]| pl.DataFrame]|tuple[TableRenderer, list[dict]| pl.DataFrame, SheetOptions]]",
     excel: Path | None,
+    *,
+    workbook_options: dict | None = None,
 ) -> Path: ...
 
 
 @overload
 def create_excel(
     sheets: "Mapping[str, tuple[TableRenderer, list[dict]| pl.DataFrame]|tuple[TableRenderer, list[dict]| pl.DataFrame, SheetOptions]]",
+    *,
+    workbook_options: dict | None = None,
 ) -> Path: ...
 
 
@@ -172,6 +176,8 @@ def create_excel(
 def create_excel(
     sheets: "Mapping[str, tuple[TableRenderer, list[dict]| pl.DataFrame]|tuple[TableRenderer, list[dict]| pl.DataFrame, SheetOptions]]",
     excel: "Path | Workbook | None" = None,
+    *,
+    workbook_options: dict | None = None,
 ) -> Path | None:
     if excel is None:
         excel = Path(os.getenv("TEMP", "/tmp")) / f"{str(uuid4())}.xlsx"
@@ -186,7 +192,13 @@ def create_excel(
             wb = excel
             path = None
         else:
-            wb = Workbook(excel)
+            wb = Workbook(
+                excel,
+                workbook_options
+                or {
+                    "remove_timezone": True,
+                },
+            )
             path = excel
 
         for sheet_name, tpl in sheets.items():

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bmsdna-table-rendering"
-version = "0.3.10"
+version = "0.3.11"
 description = ""
 authors = [{ name = "Adrian Ehrsam", email = "adrian.ehrsam@bmsuisse.ch" }]
 dependencies = [
@@ -39,4 +39,8 @@ dev = [
     "pyright >=1.1.359,<2",
     "ruff >=0.4.1",
 ]
-test = ["pytest >=8.1.1,<9", "openpyxl >=3.1.5,<4"]
+test = [
+    "pytest >=8.1.1,<9",
+    "openpyxl >=3.1.5,<4",
+    "tzdata>=2025.2",
+]

tests/test_merge.py

@@ -1,4 +1,4 @@
-from datetime import date, datetime
+from datetime import date, datetime, timezone
 import os
 import polars as pl
 
@@ -13,11 +13,13 @@ def test_excel():
                 "first_name": "hans",
                 "last_name": "muster",
                 "age": 30,
+                "dt": datetime.now(tz=timezone.utc),
             },
             {
                 "first_name": "max",
                 "last_name": "mustermann",
                 "age": 40,
+                "dt": datetime.now(tz=timezone.utc),
             },
         ]
     )