SNOW-1019480-sql-filter-nonsql-statements: strip comments from stream…

…d queries
snowflakedb · Jan 31, 2024 · 3454440 · 3454440
1 parent c023e5c
commit 3454440
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/snowflake/cli/plugins/sql/manager.py b/src/snowflake/cli/plugins/sql/manager.py
@@ -28,5 +28,10 @@ def execute(
         elif file:
             query = file.read_text()
 
-        single_statement = len(list(split_statements(StringIO(query)))) == 1
-        return single_statement, self._execute_string(query)
+        statements = tuple(
+            statement
+            for statement, _ in split_statements(StringIO(query), remove_comments=True)
+        )
+        single_statement = len(statements) == 1
+
+        return single_statement, self._execute_string("\n".join(statements))