sqlnbfmt 0.1.3__tar.gz → 0.1.5__tar.gz

{sqlnbfmt-0.1.3/sqlnbfmt.egg-info → sqlnbfmt-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlnbfmt
-Version: 0.1.3
+Version: 0.1.5
 Summary: A tool to format SQL code in Jupyter notebooks.
 Author-email: Qing <qingye779@gmail.com>
 License: MIT
@@ -61,14 +61,13 @@ pip install pre-commit
 2. Add to `.pre-commit-config.yaml`:
 ```yaml
 repos:
-  - repo: local
+  - repo: https://github.com/flyersworder/sqlnbfmt
+    rev: v0.1.1
     hooks:
       - id: sqlnbfmt
         name: sqlnbfmt
-        entry: sqlnbfmt
-        language: system
         types: [jupyter]
-        args: [--config, config.yaml, --dialect, mysql]
+        args: [--config, config.yaml, --dialect, postgres]
 ```
 Please fun the following command in your CMD for help:
 

{sqlnbfmt-0.1.3 → sqlnbfmt-0.1.5}/README.md

@@ -45,14 +45,13 @@ pip install pre-commit
 2. Add to `.pre-commit-config.yaml`:
 ```yaml
 repos:
-  - repo: local
+  - repo: https://github.com/flyersworder/sqlnbfmt
+    rev: v0.1.1
     hooks:
       - id: sqlnbfmt
         name: sqlnbfmt
-        entry: sqlnbfmt
-        language: system
         types: [jupyter]
-        args: [--config, config.yaml, --dialect, mysql]
+        args: [--config, config.yaml, --dialect, postgres]
 ```
 Please fun the following command in your CMD for help:
 
@@ -140,4 +139,4 @@ This project is licensed under the MIT License - see the [LICENSE](LICENSE) file
 - All contributors and early adopters who helped shape this tool
 
 ---
-Made with ♥️ by the sqlnbfmt team
+Made with ♥️ by the sqlnbfmt team

{sqlnbfmt-0.1.3 → sqlnbfmt-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sqlnbfmt"
-version = "0.1.3"
+version = "0.1.5"
 description = "A tool to format SQL code in Jupyter notebooks."
 authors = [
   { name = "Qing", email = "qingye779@gmail.com" }
@@ -35,4 +35,7 @@ sqlnbfmt = "sqlnbfmt.formatter:main"
 [tool.pytest.ini_options]
 markers = [
     "integration: mark a test as an integration test."
-]
+]
+
+[tool.setuptools.package-data]
+sqlnbfmt = [".pre-commit-hooks.yaml"]

{sqlnbfmt-0.1.3 → sqlnbfmt-0.1.5}/sqlnbfmt/formatter.py

@@ -223,27 +223,41 @@ class SQLStringFormatter(ast.NodeTransformer):
         return "".join(sql_parts), placeholders
 
     def format_sql_node(
-        self, node: Union[ast.Constant, ast.JoinedStr], force_single_line: bool = False
+        self, 
+        node: Union[ast.Constant, ast.JoinedStr], 
+        force_single_line: bool = False,
+        in_function: bool = False
     ) -> Optional[ast.AST]:
         """Formats SQL code in AST nodes."""
         try:
             if isinstance(node, ast.Constant) and isinstance(node.value, str):
                 if not self.is_likely_sql(node.value):
                     return None
+
                 formatted_sql = format_sql_code(
                     node.value,
                     self.dialect,
                     self.config,
                     force_single_line=force_single_line,
                 )
+
                 if formatted_sql != node.value:
                     self.changed = True
-                    # Reconstruct the string with adjusted triple quotes
+                    if in_function and '\n' in formatted_sql:
+                        # Apply Black-style indentation for function arguments
+                        lines = formatted_sql.split('\n')
+                        indented_lines = [lines[0]] + [
+                            " " * 4 + line if line.strip() else line
+                            for line in lines[1:-1]
+                        ] + [" " * 4 + lines[-1] if lines[-1].strip() else lines[-1]]
+                        formatted_sql = '\n'.join(indented_lines)
+                    
+                    # Reconstruct the string with appropriate quotes
                     if '\n' in formatted_sql:
                         formatted_str = f'"""\n{formatted_sql}\n"""'
                     else:
                         formatted_str = f'"""{formatted_sql}"""'
-                    # Parse the formatted string back into an AST node
+                    
                     formatted_node = ast.parse(formatted_str).body[0].value
                     return formatted_node
 
@@ -273,10 +287,18 @@ class SQLStringFormatter(ast.NodeTransformer):
 
                 if formatted_sql != sql_str:
                     self.changed = True
+                    # Apply Black-style indentation for function arguments
+                    if in_function and '\n' in formatted_sql:
+                        lines = formatted_sql.split('\n')
+                        indented_lines = [lines[0]] + [
+                            " " * 4 + line if line.strip() else line
+                            for line in lines[1:-1]
+                        ] + [" " * 4 + lines[-1] if lines[-1].strip() else lines[-1]]
+                        formatted_sql = '\n'.join(indented_lines)
+
                     # Reconstruct the f-string
                     new_values = []
                     if not placeholders:
-                        # If no placeholders, just create a simple f-string
                         new_values.append(ast.Constant(value=formatted_sql))
                     else:
                         pattern = re.compile('|'.join(re.escape(k) for k in placeholders.keys() if k))
@@ -340,12 +362,13 @@ class SQLStringFormatter(ast.NodeTransformer):
         if any(name in func_name for name in self.config.function_names):
             for idx, arg in enumerate(node.args):
                 if isinstance(arg, (ast.Constant, ast.JoinedStr)):
-                    formatted_node = self.format_sql_node(arg)
+                    # Format with Black-style indentation for function arguments
+                    formatted_node = self.format_sql_node(arg, in_function=True)
                     if formatted_node:
                         node.args[idx] = formatted_node
             for keyword in node.keywords:
                 if isinstance(keyword.value, (ast.Constant, ast.JoinedStr)):
-                    formatted_node = self.format_sql_node(keyword.value)
+                    formatted_node = self.format_sql_node(keyword.value, in_function=True)
                     if formatted_node:
                         keyword.value = formatted_node
         return self.generic_visit(node)

{sqlnbfmt-0.1.3 → sqlnbfmt-0.1.5/sqlnbfmt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlnbfmt
-Version: 0.1.3
+Version: 0.1.5
 Summary: A tool to format SQL code in Jupyter notebooks.
 Author-email: Qing <qingye779@gmail.com>
 License: MIT
@@ -61,14 +61,13 @@ pip install pre-commit
 2. Add to `.pre-commit-config.yaml`:
 ```yaml
 repos:
-  - repo: local
+  - repo: https://github.com/flyersworder/sqlnbfmt
+    rev: v0.1.1
     hooks:
       - id: sqlnbfmt
         name: sqlnbfmt
-        entry: sqlnbfmt
-        language: system
         types: [jupyter]
-        args: [--config, config.yaml, --dialect, mysql]
+        args: [--config, config.yaml, --dialect, postgres]
 ```
 Please fun the following command in your CMD for help: