From 175fca6e034971003fe2ab70b7a31abf1a952adb Mon Sep 17 00:00:00 2001 From: admin user Date: Mon, 7 Apr 2025 18:17:30 +0000 Subject: [PATCH] Update block_wrapper.py --- block_wrapper.py | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/block_wrapper.py b/block_wrapper.py index 34f696a..d305535 100644 --- a/block_wrapper.py +++ b/block_wrapper.py @@ -102,18 +102,25 @@ def construct_sql(input_data): try: with open("/app/main.sql", "r") as sql_file: sql_template = sql_file.read() + for key, value in input_data.items(): placeholder = f"${key}" + # Decide the final text (replacement) for SQL: if value is None: replacement = "NULL" - - if isinstance(value, str): - value = f"'{value}'" elif isinstance(value, bool): - value = "TRUE" if value else "FALSE" - sql_template = sql_template.replace(placeholder, str(value)) - logger.info(f"SQL query constructed.") + replacement = "TRUE" if value else "FALSE" + elif isinstance(value, str): + # 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() except Exception as e: logger.error("Error processing SQL template: %s", e)