fix: separate multi-statement SQL with newlines instead of concatenating (v0.3.2)

Closes #3 — multiple SQL statements in a single cell were being
concatenated on one line after the semicolon. Switched from
sqlglot.parse_one() to sqlglot.parse() so each statement is formatted
independently and joined with ";\n".

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: flyersworder

CHANGELOG.md

@@ -5,6 +5,20 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.3.2] - 2026-03-16
+
+### Fixed
+
+- Multiple SQL statements in a single cell are no longer concatenated on one line — each statement now starts on its own line after the `;` separator ([#3](https://github.com/flyersworder/sqlnbfmt/issues/3))
+
+### Changed
+
+- SQL parsing switched from `sqlglot.parse_one()` to `sqlglot.parse()` to correctly handle multi-statement inputs
+
+### Added
+
+- `multi_statement` eval fixture case covering `DROP TABLE` + `CREATE TABLE` sequences
+
 ## [0.3.1] - 2026-03-15
 
 ### Fixed

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sqlnbfmt"
-version = "0.3.1"
+version = "0.3.2"
 description = "A tool to format SQL code in Jupyter notebooks."
 authors = [
   { name = "Qing", email = "qingye779@gmail.com" }

sqlnbfmt/formatter.py

@@ -10,7 +10,7 @@
 
 import nbformat
 import yaml
-from sqlglot import parse_one, errors
+from sqlglot import parse, errors
 
 DEFAULT_SQL_KEYWORDS = frozenset(
     {
@@ -206,10 +206,18 @@ def format_sql_code(
         temp_sql = temp_sql.strip()
 
         # Parse and format SQL
-        parsed = parse_one(temp_sql, read=dialect)
-        formatted_sql = parsed.sql(
-            pretty=not force_single_line, indent=config.indent_width, dialect=dialect
-        )
+        statements = parse(temp_sql, read=dialect)
+        formatted_parts = []
+        for stmt in statements:
+            formatted_parts.append(
+                stmt.sql(
+                    pretty=not force_single_line,
+                    indent=config.indent_width,
+                    dialect=dialect,
+                )
+            )
+        separator = "; " if force_single_line else ";\n"
+        formatted_sql = separator.join(formatted_parts)
 
         # Apply formatting based on context
         if is_magic_command and not is_cell_magic:

tests/eval/generate_fixtures.py

@@ -223,6 +223,20 @@ def non_sql_unchanged():
     return nb
 
 
+def multi_statement():
+    """Multiple SQL statements separated by semicolons — each on its own line."""
+    nb = nbf.new_notebook()
+    nb.cells = [
+        nbf.new_code_cell(
+            "%%sql\n"
+            "DROP TABLE IF EXISTS Contract; DROP TABLE IF EXISTS Permanent; "
+            "CREATE TABLE Employee (eid INT, name VARCHAR(80) NOT NULL, PRIMARY KEY (eid)); "
+            "CREATE TABLE Department (did INT, name VARCHAR(30) NOT NULL, PRIMARY KEY (did))"
+        ),
+    ]
+    return nb
+
+
 def skip_hint():
     """Cell with # sqlnbfmt: skip — must remain untouched."""
     nb = nbf.new_notebook()
@@ -259,6 +273,7 @@ def skip_hint():
     "in_function_sql": in_function_sql,
     "idempotency": idempotency,
     "non_sql_unchanged": non_sql_unchanged,
+    "multi_statement": multi_statement,
     "skip_hint": skip_hint,
 }
 

tests/eval/multi_statement/expected.ipynb

@@ -0,0 +1,29 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "70bf528e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%%sql\n",
+    "DROP TABLE IF EXISTS Contract;\n",
+    "DROP TABLE IF EXISTS Permanent;\n",
+    "CREATE TABLE Employee (\n",
+    "  eid INT,\n",
+    "  name VARCHAR(80) NOT NULL,\n",
+    "  PRIMARY KEY (eid)\n",
+    ");\n",
+    "CREATE TABLE Department (\n",
+    "  did INT,\n",
+    "  name VARCHAR(30) NOT NULL,\n",
+    "  PRIMARY KEY (did)\n",
+    ")"
+   ]
+  }
+ ],
+ "metadata": {},
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

tests/eval/multi_statement/input.ipynb

@@ -0,0 +1,18 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "70bf528e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%%sql\n",
+    "DROP TABLE IF EXISTS Contract; DROP TABLE IF EXISTS Permanent; CREATE TABLE Employee (eid INT, name VARCHAR(80) NOT NULL, PRIMARY KEY (eid)); CREATE TABLE Department (did INT, name VARCHAR(30) NOT NULL, PRIMARY KEY (did))"
+   ]
+  }
+ ],
+ "metadata": {},
+ "nbformat": 4,
+ "nbformat_minor": 5
+}