Fix handling of IF EXISTS statements in BEGIN...END blocks (fixes #812).

Author: andialbrecht

CHANGELOG

@@ -25,6 +25,8 @@ Bug Fixes
 * Remove shebang from cli.py and remove executable flag (pr818).
 * Fix strip_comments not removing all comments when input contains only
   comments (issue801, pr803 by stropysh).
+* Fix splitting statements with IF EXISTS/IF NOT EXISTS inside BEGIN...END
+  blocks (issue812).
 
 
 Release 0.5.3 (Dez 10, 2024)

sqlparse/keywords.py

@@ -70,6 +70,7 @@
     (r'((LEFT\s+|RIGHT\s+|FULL\s+)?(INNER\s+|OUTER\s+|STRAIGHT\s+)?'
      r'|(CROSS\s+|NATURAL\s+)?)?JOIN\b', tokens.Keyword),
     (r'END(\s+IF|\s+LOOP|\s+WHILE)?\b', tokens.Keyword),
+    (r'IF\s+(NOT\s+)?EXISTS\b', tokens.Keyword),
     (r'NOT\s+NULL\b', tokens.Keyword),
     (r'(ASC|DESC)(\s+NULLS\s+(FIRST|LAST))?\b', tokens.Keyword.Order),
     (r'(ASC|DESC)\b', tokens.Keyword.Order),

tests/test_split.py

@@ -208,3 +208,17 @@ def test_split_go(sql, num):  # issue762
 def test_split_multiple_case_in_begin(load_file):  # issue784
     stmts = sqlparse.split(load_file('multiple_case_in_begin.sql'))
     assert len(stmts) == 1
+
+
+def test_split_if_exists_in_begin_end():  # issue812
+    # IF EXISTS should not be confused with control flow IF
+    sql = """CREATE TASK t1 AS
+BEGIN
+    CREATE OR REPLACE TABLE temp1;
+    DROP TABLE IF EXISTS temp1;
+END;
+EXECUTE TASK t1;"""
+    stmts = sqlparse.split(sql)
+    assert len(stmts) == 2
+    assert 'CREATE TASK' in stmts[0]
+    assert 'EXECUTE TASK' in stmts[1]