Fix splitting of BEGIN TRANSACTION statements (fixes #826).

Author: andialbrecht

CHANGELOG

@@ -1,7 +1,9 @@
 Development Version
 -------------------
 
-Nothing yet.
+Bug Fixes
+
+* Fix splitting of BEGIN TRANSACTION statements (issue826).
 
 
 Release 0.5.4 (Nov 28, 2025)

sqlparse/engine/statement_splitter.py

@@ -61,6 +61,17 @@ def _change_splitlevel(self, ttype, value):
                 return 1
             return 0
 
+        # Issue826: If we see a transaction keyword after BEGIN,
+        # it's a transaction statement, not a block.
+        if self._seen_begin and \
+                (ttype is T.Keyword or ttype is T.Name) and \
+                unified in ('TRANSACTION', 'WORK', 'TRAN',
+                            'DISTRIBUTED', 'DEFERRED',
+                            'IMMEDIATE', 'EXCLUSIVE'):
+            self._begin_depth = max(0, self._begin_depth - 1)
+            self._seen_begin = False
+            return 0
+
         # BEGIN and CASE/WHEN both end with END
         if unified == 'END':
             if not self._in_case:
@@ -121,7 +132,7 @@ def process(self, stream):
                     self.consume_ws = True
             elif ttype is T.Keyword and value.split()[0] == 'GO':
                 self.consume_ws = True
-            elif (ttype not in (T.Whitespace, T.Comment.Single,
+            elif (ttype not in (T.Whitespace, T.Newline, T.Comment.Single,
                                 T.Comment.Multiline)
                   and not (ttype is T.Keyword and value.upper() == 'BEGIN')):
                 # Reset _seen_begin if we see a non-whitespace, non-comment

tests/test_split.py

@@ -254,3 +254,33 @@ def test_split_begin_end_procedure():  # issue809
     stmts = sqlparse.split(sql)
     assert len(stmts) == 1
     assert 'CREATE OR REPLACE PROCEDURE' in stmts[0]
+
+
+def test_split_begin_transaction():  # issue826
+    # BEGIN TRANSACTION should not be treated as a block start
+    sql = """BEGIN TRANSACTION;
+DELETE FROM "schema"."table_a" USING "table_a_temp" WHERE "schema"."table_a"."id" = "table_a_temp"."id";
+INSERT INTO "schema"."table_a" SELECT * FROM "table_a_temp";
+END TRANSACTION;"""
+    stmts = sqlparse.split(sql)
+    assert len(stmts) == 4
+    assert stmts[0] == 'BEGIN TRANSACTION;'
+    assert stmts[1].startswith('DELETE')
+    assert stmts[2].startswith('INSERT')
+    assert stmts[3] == 'END TRANSACTION;'
+
+
+def test_split_begin_transaction_formatted():  # issue826
+    # Test with formatted SQL (newlines between BEGIN and TRANSACTION)
+    sql = """BEGIN
+TRANSACTION;
+DELETE FROM "schema"."table_a" USING "table_a_temp" WHERE "schema"."table_a"."id" = "table_a_temp"."id";
+INSERT INTO "schema"."table_a" SELECT * FROM "table_a_temp";
+END
+TRANSACTION;"""
+    stmts = sqlparse.split(sql)
+    assert len(stmts) == 4
+    assert stmts[0] == 'BEGIN\nTRANSACTION;'
+    assert stmts[1].startswith('DELETE')
+    assert stmts[2].startswith('INSERT')
+    assert stmts[3] == 'END\nTRANSACTION;'