refactor: use write_styled_excel in all download callbacks #83

src/pages/acheteur.py

@@ -34,6 +34,7 @@ from src.utils.table import (
     get_default_hidden_columns,
     prepare_table_data,
     sort_table_data,
+    write_styled_excel,
 )
 from src.utils.tracking import track_search
 
@@ -397,8 +398,10 @@ def download_acheteur_data(
     df_to_download = pl.DataFrame(data)
 
     def to_bytes(buffer):
-        df_to_download.write_excel(
+        write_styled_excel(
-            buffer, worksheet="DECP" if annee in ["Toutes les années", None] else annee
+            df_to_download,
+            buffer,
+            worksheet="DECP" if annee in ["Toutes les années", None] else annee,
         )
 
     date = datetime.datetime.now().strftime("%Y-%m-%d_%H:%M:%S")
@@ -439,7 +442,7 @@ def download_filtered_acheteur_data(
         lff = sort_table_data(lff, sort_by)
 
     def to_bytes(buffer):
-        lff.collect(engine="streaming").write_excel(buffer, worksheet="DECP")
+        write_styled_excel(lff.collect(engine="streaming"), buffer)
 
     date = datetime.datetime.now().strftime("%Y-%m-%d_%H:%M:%S")
     return dcc.send_bytes(

src/pages/observatoire.py

@@ -40,7 +40,12 @@ from src.utils.data import (
 )
 from src.utils.frontend import get_enum_values_as_dict
 from src.utils.seo import META_CONTENT
-from src.utils.table import COLUMNS, get_default_hidden_columns, prepare_table_data
+from src.utils.table import (
+    COLUMNS,
+    get_default_hidden_columns,
+    prepare_table_data,
+    write_styled_excel,
+)
 
 NAME = "Observatoire"
 
@@ -811,7 +816,7 @@ def download_observatoire(_n_clicks, filter_params, hidden_columns):
         dff = dff.drop(hidden_columns)
 
     def to_bytes(buffer):
-        dff.write_excel(buffer, worksheet="DECP")
+        write_styled_excel(dff, buffer)
 
     date = datetime.now().strftime("%Y-%m-%d_%H:%M:%S")
     return dcc.send_bytes(to_bytes, filename=f"decp_observatoire_{date}.xlsx")

src/pages/tableau.py

@@ -30,6 +30,7 @@ from src.utils.table import (
     invert_columns,
     prepare_table_data,
     sort_table_data,
+    write_styled_excel,
 )
 from src.utils.tracking import track_search
 
@@ -342,7 +343,7 @@ def download_data(n_clicks, filter_query, sort_by, hidden_columns: list | None =
         lff = sort_table_data(lff, sort_by)
 
     def to_bytes(buffer):
-        lff.collect(engine="streaming").write_excel(buffer, worksheet="DECP")
+        write_styled_excel(lff.collect(engine="streaming"), buffer)
 
     date = datetime.now().strftime("%Y-%m-%d_%H:%M:%S")
     return dcc.send_bytes(to_bytes, filename=f"decp_{date}.xlsx")

src/pages/titulaire.py

@@ -33,6 +33,7 @@ from src.utils.table import (
     get_default_hidden_columns,
     prepare_table_data,
     sort_table_data,
+    write_styled_excel,
 )
 from src.utils.tracking import track_search
 
@@ -418,8 +419,10 @@ def download_titulaire_data(
     df_to_download = pl.DataFrame(data)
 
     def to_bytes(buffer):
-        df_to_download.write_excel(
+        write_styled_excel(
-            buffer, worksheet="DECP" if annee in ["Toutes les années", None] else annee
+            df_to_download,
+            buffer,
+            worksheet="DECP" if annee in ["Toutes les années", None] else annee,
         )
 
     date = datetime.datetime.now().strftime("%Y-%m-%d_%H:%M:%S")
@@ -460,7 +463,7 @@ def download_filtered_titulaire_data(
         lff = sort_table_data(lff, sort_by)
 
     def to_bytes(buffer):
-        lff.collect(engine="streaming").write_excel(buffer, worksheet="DECP")
+        write_styled_excel(lff.collect(engine="streaming"), buffer)
 
     date = datetime.datetime.now().strftime("%Y-%m-%d_%H:%M:%S")
     return dcc.send_bytes(