Skip to content

Commit 9c048ba

Browse files
added the ast class for UPDATE DATABASE
1 parent 75420c6 commit 9c048ba

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed
Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
from mindsdb_sql_parser.ast.base import ASTNode
2+
from mindsdb_sql_parser.utils import indent
3+
4+
5+
class UpdateDatabase(ASTNode):
6+
def __init__(self, name, updated_params, *args, **kwargs):
7+
super().__init__(*args, **kwargs)
8+
self.name = name
9+
self.params = updated_params
10+
11+
def to_tree(self, *args, level=0, **kwargs):
12+
ind = indent(level)
13+
out_str = f'{ind}UpdateDatabase(' \
14+
f'name={self.name.to_string()}, ' \
15+
f'updated_params={self.params})'
16+
return out_str
17+
18+
def get_string(self, *args, **kwargs):
19+
params = self.params.copy()
20+
21+
set_ar = [f'{k}={repr(v)}' for k, v in params.items()]
22+
set_str = ', '.join(set_ar)
23+
24+
out_str = f'UPDATE DATABASE {self.name.to_string()} SET {set_str}'
25+
return out_str

0 commit comments

Comments
 (0)