Update block_wrapper.py
This commit is contained in:
parent
f2d318832f
commit
175fca6e03
@ -102,18 +102,25 @@ def construct_sql(input_data):
|
|||||||
try:
|
try:
|
||||||
with open("/app/main.sql", "r") as sql_file:
|
with open("/app/main.sql", "r") as sql_file:
|
||||||
sql_template = sql_file.read()
|
sql_template = sql_file.read()
|
||||||
|
|
||||||
for key, value in input_data.items():
|
for key, value in input_data.items():
|
||||||
placeholder = f"${key}"
|
placeholder = f"${key}"
|
||||||
|
|
||||||
|
# Decide the final text (replacement) for SQL:
|
||||||
if value is None:
|
if value is None:
|
||||||
replacement = "NULL"
|
replacement = "NULL"
|
||||||
|
|
||||||
if isinstance(value, str):
|
|
||||||
value = f"'{value}'"
|
|
||||||
elif isinstance(value, bool):
|
elif isinstance(value, bool):
|
||||||
value = "TRUE" if value else "FALSE"
|
replacement = "TRUE" if value else "FALSE"
|
||||||
sql_template = sql_template.replace(placeholder, str(value))
|
elif isinstance(value, str):
|
||||||
logger.info(f"SQL query constructed.")
|
# Escape quotes if needed, or just do naive single quote
|
||||||
|
replacement = f"'{value}'"
|
||||||
|
else:
|
||||||
|
replacement = str(value)
|
||||||
|
|
||||||
|
# Replace the placeholder with our final replacement string
|
||||||
|
sql_template = sql_template.replace(placeholder, replacement)
|
||||||
|
|
||||||
|
logger.info("SQL query constructed.")
|
||||||
return sql_template.strip()
|
return sql_template.strip()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Error processing SQL template: %s", e)
|
logger.error("Error processing SQL template: %s", e)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user