diff --git a/src/pages/home.py b/src/pages/home.py
index c7b4095..cdf2c7f 100644
--- a/src/pages/home.py
+++ b/src/pages/home.py
@@ -28,8 +28,6 @@ lf = lf.filter(pl.col("donneesActuelles"))
# Suppression des colonnes inutiles
lf = lf.drop(
[
- "titulaire_siren",
- "acheteur_siren",
"donneesActuelles",
]
)
@@ -231,8 +229,8 @@ def download_data(n_clicks, hidden_columns: list = None):
# Rétablissement des colonnes source et sourceOpenData (voir add_resource_link)
df_to_download = df_to_download.with_columns(
- pl.col("source").str.extract(r'href="(.*?)"').alias("sourceOpenData"),
- pl.col("source").str.extract(r'">(.*?)<').alias("source"),
+ pl.col("source").str.extract(r'href="(.*?)"').alias("sourceFile"),
+ pl.col("source").str.extract(r'">(.*?)<').alias("sourceDataset"),
)
# Les colonnes masquées sont supprimées
diff --git a/src/utils.py b/src/utils.py
index ace70c8..5efe954 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -43,10 +43,10 @@ def split_filter_part(filter_part):
def add_resource_link(lff: pl.LazyFrame) -> pl.LazyFrame:
lff = lff.with_columns(
(
- '' + pl.col("source") + ""
+ '' + pl.col("sourceDataset") + ""
).alias("source")
)
- lff = lff.drop("sourceOpenData")
+ lff = lff.drop(["sourceFile", "sourceDataset"])
return lff