fix comment

Author: 3nids

pum/utils/execute_sql.py

@@ -28,51 +28,50 @@ def execute_sql(
         RuntimeError: If the SQL execution fails.
     """
     cursor = conn.cursor()
-    try:
-        if isinstance(sql, Path):
-            logger.debug(
-                f"Executing SQL from file: {sql} with parameters: {parameters}",
-            )
-            with open(sql) as file:
-                sql_content = file.read()
-                # Remove SQL comments
-                def remove_sql_comments(sql):
-                    # Remove multiline comments (/* ... */)
-                    sql = re.sub(r'/\*.*?\*/', '', sql, flags=re.DOTALL)
-                    # Remove single-line comments (-- ...)
-                    sql = re.sub(r'--.*', '', sql)
-                    return sql
-                sql_content = remove_sql_comments(sql_content)
-                if parameters:
-                    for key, value in parameters.items():
-                        sql_content = sql_content.replace(f"{{{{ {key} }}}}", str(value))
-                    sql_code = sql_content
-                else:
-                    sql_code = sql_content
+    if isinstance(sql, Path):
+        logger.debug(
+            f"Executing SQL from file: {sql} with parameters: {parameters}",
+        )
+        with open(sql) as file:
+            sql_content = file.read()
+            # Remove SQL comments
+            def remove_sql_comments(sql):
+                # Remove multiline comments (/* ... */)
+                sql = re.sub(r'/\*.*?\*/', '', sql, flags=re.DOTALL)
+                # Remove single-line comments (-- ...)
+                sql = re.sub(r'--(?=(?:[^\'"]|\'[^\']*\'|"[^"]*")*$).*', '', sql)
+                return sql
+            sql_content = remove_sql_comments(sql_content)
+            if parameters:
+                for key, value in parameters.items():
+                    sql_content = sql_content.replace(f"{{{{ {key} }}}}", str(value))
+                sql_code = sql_content
+            else:
+                sql_code = sql_content
 
-                def split_sql_statements(sql):
-                    pattern = r'(?:[^;\'"]|\'[^\']*\'|"[^"]*")*;'
-                    matches = re.finditer(pattern, sql, re.DOTALL)
-                    statements = []
-                    last_end = 0
-                    for match in matches:
-                        end = match.end()
-                        statements.append(sql[last_end : end - 1].strip())
-                        last_end = end
-                    if last_end < len(sql):
-                        statements.append(sql[last_end:].strip())
-                    return [stmt for stmt in statements if stmt]
+            def split_sql_statements(sql):
+                pattern = r'(?:[^;\'"]|\'[^\']*\'|"[^"]*")*;'
+                matches = re.finditer(pattern, sql, re.DOTALL)
+                statements = []
+                last_end = 0
+                for match in matches:
+                    end = match.end()
+                    statements.append(sql[last_end : end - 1].strip())
+                    last_end = end
+                if last_end < len(sql):
+                    statements.append(sql[last_end:].strip())
+                return [stmt for stmt in statements if stmt]
 
-                sql_code = split_sql_statements(sql_code)
-        else:
-            sql_code = [sql]
+            sql_code = split_sql_statements(sql_code)
+    else:
+        sql_code = [sql]
 
-        for statement in sql_code:
+    for statement in sql_code:
+        try:
             cursor.execute(statement)
-
-    except SyntaxError as e:
-        logger.debug(f"Error executing SQL: {sql_code}")
-        raise PumSqlException(f"SQL execution failed for the following code: {sql} {e}") from e
+        except SyntaxError as e:
+            logger.debug(f"Error executing SQL: {statement}")
+            raise PumSqlException(f"SQL execution failed for the following code: {sql} {e}") from e
     if commit:
         conn.commit()