fixed type hints, docstrings and imports

Author: MinuraPunchihewa

mindsdb_sql_parser/ast/mindsdb/update_database.py

@@ -1,9 +1,18 @@
 from mindsdb_sql_parser.ast.base import ASTNode
+from mindsdb_sql_parser.ast.select import Identifier
 from mindsdb_sql_parser.utils import indent
 
 
 class UpdateDatabase(ASTNode):
-    def __init__(self, name, updated_params, *args, **kwargs):
+    """
+    Update a database.
+    """
+    def __init__(self, name: Identifier, updated_params: dict, *args, **kwargs):
+        """
+        Args:
+            name: Identifier -- name of the database to update.
+            params: dict -- parameters to update in the database.
+        """
         super().__init__(*args, **kwargs)
         self.name = name
         self.params = updated_params
@@ -22,4 +31,4 @@ def get_string(self, *args, **kwargs):
         set_str = ', '.join(set_ar)
 
         out_str = f'UPDATE DATABASE {self.name.to_string()} SET {set_str}'
-        return out_str
+        return out_str

mindsdb_sql_parser/parser.py

@@ -78,6 +78,7 @@ class MindsDBParser(Parser):
        'delete',
        'evaluate',
        'drop_database',
+       'update_database',
        'drop_view',
        'drop_table',
        'create_table',