Skip to content

Commit 5e75f27

Browse files
added method for updating database
1 parent a408514 commit 5e75f27

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

mindsdb_sdk/databases.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from typing import Dict, List, Union
22

3-
from mindsdb_sql_parser.ast.mindsdb import CreateDatabase
3+
from mindsdb_sql_parser.ast.mindsdb import AlterDatabase, CreateDatabase
44
from mindsdb_sql_parser.ast import DropDatabase, Identifier
55

66
from mindsdb_sdk.utils.objects_collection import CollectionBase
@@ -175,3 +175,19 @@ def get(self, name: str) -> Database:
175175
if name not in databases:
176176
raise AttributeError("Database doesn't exist")
177177
return databases[name]
178+
179+
def update(self, name: str, connection_args: Dict) -> Database:
180+
"""
181+
Update integration connection parameters
182+
183+
:param name: name of integration
184+
:param connection_args: new connection parameters
185+
:return: updated Database object
186+
"""
187+
ast_query = AlterDatabase(
188+
name=Identifier(name),
189+
altered_params={
190+
"parameters": connection_args},
191+
)
192+
self.api.sql_query(ast_query.to_string())
193+
return self.get(name)

0 commit comments

Comments
 (0)