Add Support for Parameterized Queries issue fixed

Author: pankaj-bind

mindsdb_sql_parser/ast/select/parameter.py

@@ -1,16 +1,18 @@
 from mindsdb_sql_parser.ast.base import ASTNode
+from mindsdb_sql_parser.utils import indent
 
 
 class Parameter(ASTNode):
     def __init__(self, value, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.value = value
 
-    def __repr__(self):
-        return f'Parameter({self.value})'
-
     def to_tree(self, *args, level=0, **kwargs):
-        return '\t' * level + f'Parameter({repr(self.value)})'
+        alias_str = f', alias={self.alias.to_tree()}' if self.alias else ''
+        return indent(level) + f'Parameter(value={repr(self.value)}{alias_str})'
 
     def get_string(self, *args, **kwargs):
-        return ':' + str(self.value)
+        return str(self.value)
+
+    def __repr__(self):
+        return f'Parameter({repr(self.value)})'

tests/test_base_sql/test_parameters.py

@@ -0,0 +1,72 @@
+from mindsdb_sql_parser import parse_sql
+from mindsdb_sql_parser.ast import *
+
+class TestParameters:
+    def test_select_with_parameter_in_where(self):
+        sql = "SELECT * FROM tbl WHERE col = ?"
+        ast = parse_sql(sql)
+        expected_ast = Select(
+            targets=[Star()],
+            from_table=Identifier('tbl'),
+            where=BinaryOperation(op='=', args=[
+                Identifier('col'),
+                Parameter('?')
+            ])
+        )
+        assert ast.to_tree() == expected_ast.to_tree()
+        assert str(ast) == str(expected_ast)
+
+    def test_select_multiple_parameters(self):
+        sql = "SELECT * FROM tbl WHERE col1 > ? AND col2 = ?"
+        ast = parse_sql(sql)
+        expected_ast = Select(
+            targets=[Star()],
+            from_table=Identifier('tbl'),
+            where=BinaryOperation(op='and', args=[
+                BinaryOperation(op='>', args=[
+                    Identifier('col1'),
+                    Parameter('?')
+                ]),
+                BinaryOperation(op='=', args=[
+                    Identifier('col2'),
+                    Parameter('?')
+                ])
+            ])
+        )
+        assert ast.to_tree() == expected_ast.to_tree()
+        assert str(ast) == str(expected_ast)
+
+    def test_insert_with_parameters(self):
+        sql = "INSERT INTO tbl_name(a, c) VALUES (?, ?)"
+        ast = parse_sql(sql)
+        expected_ast = Insert(
+            table=Identifier('tbl_name'),
+            columns=[Identifier('a'), Identifier('c')],
+            values=[
+                [Parameter('?'), Parameter('?')]
+            ]
+        )
+        assert ast.to_tree() == expected_ast.to_tree()
+        assert str(ast) == str(expected_ast)
+
+    def test_insert_with_multiple_parameter_rows(self):
+        sql = "INSERT INTO tbl_name VALUES (?, ?), (?, ?)"
+        ast = parse_sql(sql)
+        expected_ast = Insert(
+            table=Identifier('tbl_name'),
+            values=[
+                [Parameter('?'), Parameter('?')],
+                [Parameter('?'), Parameter('?')]
+            ]
+        )
+        assert ast.to_tree() == expected_ast.to_tree()
+        assert str(ast) == str(expected_ast)
+        
+    def test_select_parameter_as_target(self):
+        sql = "SELECT ?"
+        ast = parse_sql(sql)
+        expected_ast = Select(
+            targets=[Parameter('?')]
+        )
+        assert ast.to_tree() == expected_ast.to_tree()
+        assert str(ast) == str(expected_ast)