Merge pull request #211 from mindsdb/feature/update_database

Added the Update Operation for Databases

Author: MinuraPunchihewa

mindsdb_sdk/__about__.py

@@ -1,6 +1,6 @@
 __title__ = 'mindsdb_sdk'
 __package_name__ = 'mindsdb_sdk'
-__version__ = '3.4.8'
+__version__ = '3.4.9'
 __description__ = "MindsDB Python SDK, provides an SDK to use a remote mindsdb instance"
 __email__ = "[email protected]"
 __author__ = 'MindsDB Inc'

mindsdb_sdk/databases.py

@@ -1,6 +1,6 @@
 from typing import Dict, List, Union
 
-from mindsdb_sql_parser.ast.mindsdb import CreateDatabase
+from mindsdb_sql_parser.ast.mindsdb import AlterDatabase, CreateDatabase
 from mindsdb_sql_parser.ast import DropDatabase, Identifier
 
 from mindsdb_sdk.utils.objects_collection import CollectionBase
@@ -175,3 +175,19 @@ def get(self, name: str) -> Database:
         if name not in databases:
             raise AttributeError("Database doesn't exist")
         return databases[name]
+
+    def update(self, name: str, connection_args: Dict) -> Database:
+        """
+        Update integration connection parameters
+
+        :param name: name of integration
+        :param connection_args: new connection parameters
+        :return: updated Database object
+        """
+        ast_query = AlterDatabase(
+            name=Identifier(name),
+            altered_params={
+                "parameters": connection_args},
+        )
+        self.api.sql_query(ast_query.to_string())
+        return self.get(name)

requirements.txt

@@ -1,6 +1,6 @@
 requests
 pandas >= 1.3.5
-mindsdb-sql-parser ~= 0.0.0
+mindsdb-sql-parser == 0.10.0
 docstring-parser >= 0.7.3
 tenacity >= 8.0.1
 openai >= 1.74.1