lf => lff pour éviter un conflit avec le lf global

This commit is contained in:
Colin Maudry
2025-06-24 15:02:53 +02:00
parent 637c35522f
commit 1ab5c639f0
+10 -10
View File
@@ -37,8 +37,8 @@ def split_filter_part(filter_part):
return [None] * 3 return [None] * 3
def add_annuaire_link(df: pl.LazyFrame): def add_annuaire_link(dff: pl.LazyFrame):
df = df.with_columns( dff = dff.with_columns(
pl.when(pl.col("titulaire_typeIdentifiant") == "SIRET") pl.when(pl.col("titulaire_typeIdentifiant") == "SIRET")
.then( .then(
pl.col("titulaire_id") pl.col("titulaire_id")
@@ -49,7 +49,7 @@ def add_annuaire_link(df: pl.LazyFrame):
.otherwise(pl.col("titulaire_id")) .otherwise(pl.col("titulaire_id"))
.alias("titulaire_id") .alias("titulaire_id")
) )
df = df.with_columns( dff = dff.with_columns(
( (
pl.col("acheteur_id") pl.col("acheteur_id")
+ ' <a href="https://annuaire-entreprises.data.gouv.fr/etablissement/' + ' <a href="https://annuaire-entreprises.data.gouv.fr/etablissement/'
@@ -57,28 +57,28 @@ def add_annuaire_link(df: pl.LazyFrame):
+ '" target="_blank">📑</a>' + '" target="_blank">📑</a>'
).alias("acheteur_id") ).alias("acheteur_id")
) )
return df return dff
def booleans_to_strings(lf: pl.LazyFrame) -> pl.LazyFrame: def booleans_to_strings(lff: pl.LazyFrame) -> pl.LazyFrame:
""" """
Convert all boolean columns to string type. Convert all boolean columns to string type.
""" """
lf = lf.with_columns( lff = lff.with_columns(
pl.col(cs.Boolean) pl.col(cs.Boolean)
.cast(pl.String) .cast(pl.String)
.str.replace("true", "oui") .str.replace("true", "oui")
.str.replace("false", "non") .str.replace("false", "non")
) )
return lf return lff
def numbers_to_strings(lf: pl.LazyFrame) -> pl.LazyFrame: def numbers_to_strings(lff: pl.LazyFrame) -> pl.LazyFrame:
""" """
Convert all numeric columns to string type. Convert all numeric columns to string type.
""" """
lf = lf.with_columns(pl.col(pl.Float64, pl.Int16).cast(pl.String).fill_null("")) lff = lff.with_columns(pl.col(pl.Float64, pl.Int16).cast(pl.String).fill_null(""))
return lf return lff
def format_number(number) -> str: def format_number(number) -> str: