diff --git a/src/pages/acheteur.py b/src/pages/acheteur.py index 23bf203..ab43ee7 100644 --- a/src/pages/acheteur.py +++ b/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( - buffer, worksheet="DECP" if annee in ["Toutes les années", None] else annee + write_styled_excel( + 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( diff --git a/src/pages/observatoire.py b/src/pages/observatoire.py index 7dc85f9..e1c7592 100644 --- a/src/pages/observatoire.py +++ b/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") diff --git a/src/pages/tableau.py b/src/pages/tableau.py index 1a272ff..56bdf00 100644 --- a/src/pages/tableau.py +++ b/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") diff --git a/src/pages/titulaire.py b/src/pages/titulaire.py index 3edb8ba..e30022d 100644 --- a/src/pages/titulaire.py +++ b/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( - buffer, worksheet="DECP" if annee in ["Toutes les années", None] else annee + write_styled_excel( + 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(