sqlnbfmt 0.3.1__tar.gz → 0.3.2__tar.gz

{sqlnbfmt-0.3.1/sqlnbfmt.egg-info → sqlnbfmt-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlnbfmt
-Version: 0.3.1
+Version: 0.3.2
 Summary: A tool to format SQL code in Jupyter notebooks.
 Author-email: Qing <qingye779@gmail.com>
 License: MIT

{sqlnbfmt-0.3.1 → sqlnbfmt-0.3.2}/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-0.3.1 → sqlnbfmt-0.3.2}/sqlnbfmt/formatter.py

@@ -10,7 +10,7 @@ from typing import Set, Optional, Dict, Union, Tuple
 
 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:

{sqlnbfmt-0.3.1 → sqlnbfmt-0.3.2/sqlnbfmt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlnbfmt
-Version: 0.3.1
+Version: 0.3.2
 Summary: A tool to format SQL code in Jupyter notebooks.
 Author-email: Qing <qingye779@gmail.com>
 License: MIT