MindsDB 1.2.9__py3-none-any.whl → 24.12.4.0__py3-none-any.whl

mindsdb/integrations/handlers/vitess_handler/tests/test_vitess_handler.py

@@ -0,0 +1,56 @@
+import unittest
+from mindsdb.integrations.handlers.vitess_handler.vitess_handler import VitessHandler
+from mindsdb.integrations.libs.response import RESPONSE_TYPE
+
+class VitessHandlerTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.kwargs = {
+            "connection_data": {
+                "host": "localhost",
+                "port": 33577 ,
+                "user": "root",
+                "password": "",
+                "database": "vitess",
+            }
+        }
+        cls.handler = VitessHandler('test_vitess_handler', **cls.kwargs)
+
+    def test_0_check_connection(self):
+        assert self.handler.check_connection()
+
+    def test_1_connect(self):
+        assert self.handler.connect()
+    
+    def test_2_create_table(self):
+        query = "CREATE Table IF NOT EXISTS Lover(name varchar(101));"
+        result = self.handler.query(query)
+        assert result.type is not RESPONSE_TYPE.ERROR 
+
+    def test_3_insert(self):
+        query = "INSERT INTO LOVER VALUES('Shiv Shakti');"
+        result = self.handler.query(query)
+        assert result.type is not RESPONSE_TYPE.ERROR 
+
+    def test_4_native_query_select(self):
+        query = "SELECT * FROM LOVER;"
+        result = self.handler.query(query)
+        assert result.type is RESPONSE_TYPE.TABLE 
+
+    def test_5_get_tables(self):
+        tables = self.handler.get_tables()
+        assert tables.type is  RESPONSE_TYPE.TABLE
+
+    def test_6_get_columns(self):
+        columns = self.handler.get_columns('LOVER')
+        
+        query = "DROP Table IF  EXISTS Lover;"
+        result = self.handler.query(query)
+        assert columns.type is not RESPONSE_TYPE.ERROR
+
+
+if __name__ == '__main__':
+    unittest.main()
+
+
+    

mindsdb/integrations/handlers/vitess_handler/vitess_handler.py

@@ -0,0 +1,48 @@
+from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
+from collections import OrderedDict
+
+
+from mindsdb.integrations.handlers.mysql_handler import Handler as MysqlHandler
+
+
+class VitessHandler(MysqlHandler):
+    """
+    This handler handles connection and execution of the Vitess statements.
+    """
+    name = 'vitess'
+
+    def __init__(self, name, **kwargs):
+        super().__init__(name, **kwargs)
+
+
+
+connection_args = OrderedDict(
+    user={
+        'type': ARG_TYPE.STR,
+        'description': 'The user name used to authenticate with the Vitess server.'
+    },
+    password={
+        'type': ARG_TYPE.STR,
+        'description': 'The password to authenticate the user with the Vitess server.'
+    },
+    database={
+        'type': ARG_TYPE.STR,
+        'description': 'The database name to use when connecting with the Vitess server.'
+    },
+    host={
+        'type': ARG_TYPE.STR,
+        'description': 'The host name or IP address of the Vitess server.'
+    },
+    port={
+        'type': ARG_TYPE.INT,
+        'description': 'The TCP/IP port of the Vitess server. Must be an integer.'
+    }
+)
+
+connection_args_example = OrderedDict(
+    host='127.0.0.1',
+    port=5432,
+    user='root',
+    password='<???>',
+    database='database'
+)

mindsdb/integrations/handlers/weaviate_handler/__about__.py

@@ -0,0 +1,9 @@
+__title__ = "MindsDB Weaviate handler"
+__package_name__ = "mindsdb_weaviate_handler"
+__version__ = "0.0.1"
+__description__ = "MindsDB handler for weaviate"
+__author__ = "Abhijit Pal"
+__github__ = "https://github.com/mindsdb/mindsdb"
+__pypi__ = "https://pypi.org/project/mindsdb/"
+__license__ = "MIT"
+__copyright__ = "Copyright 2023 - mindsdb"

mindsdb/integrations/handlers/weaviate_handler/__init__.py

@@ -0,0 +1,29 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+
+from .__about__ import __description__ as description
+from .__about__ import __version__ as version
+from .connection_args import connection_args, connection_args_example
+try:
+    from .weaviate_handler import WeaviateDBHandler as Handler
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+
+title = "Weaviate"
+name = "weaviate"
+type = HANDLER_TYPE.DATA
+icon_path = "icon.svg"
+
+__all__ = [
+    "Handler",
+    "version",
+    "name",
+    "type",
+    "title",
+    "description",
+    "connection_args",
+    "connection_args_example",
+    "import_error",
+    "icon_path",
+]

mindsdb/integrations/handlers/weaviate_handler/connection_args.py

@@ -0,0 +1,29 @@
+from collections import OrderedDict
+
+from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
+
+
+connection_args = OrderedDict(
+    weaviate_url={
+        "type": ARG_TYPE.STR,
+        "description": "weaviate url/ local endpoint",
+        "required": False,
+    },
+    weaviate_api_key={
+        "type": ARG_TYPE.STR,
+        "description": "weaviate API KEY",
+        "required": False,
+        "secret": True
+    },
+    persistence_directory={
+        "type": ARG_TYPE.STR,
+        "description": "persistence directory for weaviate",
+        "required": False,
+    },
+)
+
+connection_args_example = OrderedDict(
+    weaviate_url="http://localhost:8080",
+    weaviate_api_key="<api_key>",
+    persistence_directory="db_path",
+)