sqlnbfmt 0.1.6__tar.gz → 0.1.7__tar.gz

{sqlnbfmt-0.1.6/sqlnbfmt.egg-info → sqlnbfmt-0.1.7}/PKG-INFO

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

{sqlnbfmt-0.1.6 → sqlnbfmt-0.1.7}/pyproject.toml

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

{sqlnbfmt-0.1.6 → sqlnbfmt-0.1.7}/sqlnbfmt/formatter.py

@@ -254,7 +254,7 @@ class SQLStringFormatter(ast.NodeTransformer):
                         formatted_sql = '\n'.join(indented_lines)
                         
                         # Reconstruct the string with appropriate quotes
-                        formatted_str = f'"""\n{formatted_sql}\n"""'
+                        formatted_str = f'"""\n{formatted_sql}\n{" " * 4}"""'
                     else:
                         # Ensure newlines are added around the formatted SQL
                         formatted_str = f'"""\n{formatted_sql}\n"""'
@@ -328,11 +328,18 @@ class SQLStringFormatter(ast.NodeTransformer):
 
                     new_node = ast.JoinedStr(values=new_values)
                     if '\n' in formatted_sql:
-                        formatted_fstring = ast.JoinedStr(values=[
-                            ast.Constant(value='\n'),
-                            *new_node.values,
-                            ast.Constant(value='\n')
-                        ])
+                        if in_function:
+                            formatted_fstring = ast.JoinedStr(values=[
+                                ast.Constant(value='\n'),
+                                *new_node.values,
+                                ast.Constant(value=f'\n{" " * 4}')
+                            ])
+                        else:
+                            formatted_fstring = ast.JoinedStr(values=[
+                                ast.Constant(value='\n'),
+                                *new_node.values,
+                                ast.Constant(value='\n')
+                            ])
                     else:
                         formatted_fstring = new_node
                     return formatted_fstring

{sqlnbfmt-0.1.6 → sqlnbfmt-0.1.7/sqlnbfmt.egg-info}/PKG-INFO

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