colin/colibre
1
0
Fork 0
Code Issues 30 Pull Requests Actions Packages Projects Releases 33 Activity

fix(table_sql): gérer *foo* et utiliser isinstance pour les types Polars

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Colin Maudry
2026-04-21 23:37:03 +02:00
parent 95c90e319e
commit 74213d3844
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/utils/table_sql.py
+10 -4
View File
@@ -27,14 +27,19 @@ def filter_query_to_sql(filter_query: str, schema: pl.Schema) -> tuple[str, list
logger.warning(f"Colonne inconnue ignorée : {col_name!r}") logger.warning(f"Colonne inconnue ignorée : {col_name!r}")
continue continue
col_type = str(schema[col_name]) col_type = schema[col_name]
is_numeric = col_type.startswith("Int") or col_type.startswith("Float") is_numeric = isinstance(
col_type, (pl.Int8, pl.Int16, pl.Int32, pl.Int64, pl.Float32, pl.Float64)
)
col_is_date = col_type == pl.Date
quoted_col = f'"{col_name}"' quoted_col = f'"{col_name}"'
if is_numeric: if is_numeric:
try: try:
value = ( value = (
int(raw_value) if col_type.startswith("Int") else float(raw_value) int(raw_value)
if isinstance(col_type, (pl.Int8, pl.Int16, pl.Int32, pl.Int64))
else float(raw_value)
) )
except ValueError: except ValueError:
logger.warning(f"Valeur numérique invalide ignorée : {raw_value!r}") logger.warning(f"Valeur numérique invalide ignorée : {raw_value!r}")
@@ -54,13 +59,14 @@ def filter_query_to_sql(filter_query: str, schema: pl.Schema) -> tuple[str, list
# String / Date : toujours traité comme texte (parité avec Polars) # String / Date : toujours traité comme texte (parité avec Polars)
value = raw_value.strip('"') value = raw_value.strip('"')
col_is_date = col_type == "Date"
if operator == "contains": if operator == "contains":
if value.endswith("*") and not value.startswith("*"): if value.endswith("*") and not value.startswith("*"):
like = value[:-1] + "%" like = value[:-1] + "%"
elif value.startswith("*") and not value.endswith("*"): elif value.startswith("*") and not value.endswith("*"):
like = "%" + value[1:] like = "%" + value[1:]
elif value.startswith("*") and value.endswith("*"):
like = "%" + value[1:-1] + "%"
else: else:
like = "%" + value + "%" like = "%" + value + "%"
target = f"CAST({quoted_col} AS VARCHAR)" if col_is_date else quoted_col target = f"CAST({quoted_col} AS VARCHAR)" if col_is_date else quoted_col