MindsDB 1.3.1__py3-none-any.whl → 25.1.2.0__py3-none-any.whl

mindsdb/integrations/handlers/sqlite_handler/sqlite_handler.py

@@ -0,0 +1,184 @@
+import os
+from typing import Optional
+
+import pandas as pd
+import sqlite3
+
+from mindsdb_sql_parser import parse_sql
+from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
+from mindsdb.integrations.libs.base import DatabaseHandler
+
+from mindsdb_sql_parser.ast.base import ASTNode
+
+from mindsdb.utilities import log
+from mindsdb.integrations.libs.response import (
+    HandlerStatusResponse as StatusResponse,
+    HandlerResponse as Response,
+    RESPONSE_TYPE
+)
+
+
+logger = log.getLogger(__name__)
+
+class SQLiteHandler(DatabaseHandler):
+    """
+    This handler handles connection and execution of the SQLite statements.
+    """
+
+    name = 'sqlite'
+
+    def __init__(self, name: str, connection_data: Optional[dict], **kwargs):
+        """
+        Initialize the handler.
+        Args:
+            name (str): name of particular handler instance
+            connection_data (dict): parameters for connecting to the database
+            **kwargs: arbitrary keyword arguments.
+        """
+        super().__init__(name)
+        self.parser = parse_sql
+        self.dialect = 'sqlite'
+        self.connection_data = connection_data
+        self.kwargs = kwargs
+
+        self.connection = None
+        self.is_connected = False
+
+    def __del__(self):
+        if self.is_connected is True:
+            self.disconnect()
+
+    def connect(self) -> StatusResponse:
+        """
+        Set up the connection required by the handler.
+        Returns:
+            HandlerStatusResponse
+        """
+
+        if self.is_connected is True:
+            return self.connection
+
+        self.connection = sqlite3.connect(self.connection_data['db_file'])
+        self.is_connected = True
+
+        return self.connection
+
+    def disconnect(self):
+        """
+        Close any existing connections.
+        """
+
+        if self.is_connected is False:
+            return
+
+        self.connection.close()
+        self.is_connected = False
+        return self.is_connected
+
+    def check_connection(self) -> StatusResponse:
+        """
+        Check connection to the handler.
+        Returns:
+            HandlerStatusResponse
+        """
+
+        response = StatusResponse(False)
+        need_to_close = self.is_connected is False
+
+        try:
+            if not os.path.isfile(self.connection_data['db_file']):
+                raise FileNotFoundError(f"File '{self.connection_data['db_file']}' not found. Use ':memory:' to create an in-memory database if you don't have a file.")
+            self.connect()
+            response.success = True
+        except Exception as e:
+            logger.error(f'Error connecting to SQLite {self.connection_data["db_file"]}, {e}!')
+            response.error_message = str(e)
+        finally:
+            if response.success is True and need_to_close:
+                self.disconnect()
+            if response.success is False and self.is_connected is True:
+                self.is_connected = False
+
+        return response
+
+    def native_query(self, query: str) -> StatusResponse:
+        """
+        Receive raw query and act upon it somehow.
+        Args:
+            query (str): query in native format
+        Returns:
+            HandlerResponse
+        """
+
+        need_to_close = self.is_connected is False
+
+        connection = self.connect()
+        cursor = connection.cursor()
+
+        try:
+            cursor.execute(query)
+            result = cursor.fetchall()
+            if result:
+                response = Response(
+                    RESPONSE_TYPE.TABLE,
+                    data_frame=pd.DataFrame(
+                        result,
+                        columns=[x[0] for x in cursor.description]
+                    )
+                )
+            else:
+                connection.commit()
+                response = Response(RESPONSE_TYPE.OK)
+        except Exception as e:
+            logger.error(f'Error running query: {query} on {self.connection_data["db_file"]}!')
+            response = Response(
+                RESPONSE_TYPE.ERROR,
+                error_message=str(e)
+            )
+
+        cursor.close()
+        if need_to_close is True:
+            self.disconnect()
+
+        return response
+
+    def query(self, query: ASTNode) -> StatusResponse:
+        """
+        Receive query as AST (abstract syntax tree) and act upon it somehow.
+        Args:
+            query (ASTNode): sql query represented as AST. May be any kind
+                of query: SELECT, INTSERT, DELETE, etc
+        Returns:
+            HandlerResponse
+        """
+        renderer = SqlalchemyRender('sqlite')
+        query_str = renderer.get_string(query, with_failback=True)
+        return self.native_query(query_str)
+
+    def get_tables(self) -> StatusResponse:
+        """
+        Return list of entities that will be accessible as tables.
+        Returns:
+            HandlerResponse
+        """
+
+        query = "SELECT name from sqlite_master where type= 'table';"
+        result = self.native_query(query)
+        df = result.data_frame
+        result.data_frame = df.rename(columns={df.columns[0]: 'table_name'})
+        return result
+
+    def get_columns(self, table_name: str) -> StatusResponse:
+        """
+        Returns a list of entity columns.
+        Args:
+            table_name (str): name of one of tables returned by self.get_tables()
+        Returns:
+            HandlerResponse
+        """
+
+        query = f"PRAGMA table_info([{table_name}]);"
+        result = self.native_query(query)
+        df = result.data_frame
+        result.data_frame = df.rename(columns={'name': 'column_name', 'type': 'data_type'})
+        return result

mindsdb/integrations/handlers/sqlite_handler/tests/test_sqlite_handler.py

@@ -0,0 +1,32 @@
+import unittest
+from mindsdb.integrations.handlers.sqlite_handler.sqlite_handler import SQLiteHandler
+from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE
+
+
+class SQLiteHandlerTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.kwargs = {
+            "db_file": "chinook.db",
+        }
+        cls.handler = SQLiteHandler('test_sqlite_handler', cls.kwargs)
+
+    def test_0_check_connection(self):
+        assert self.handler.check_connection()
+
+    def test_1_native_query_select(self):
+        query = "SELECT * FROM customers"
+        result = self.handler.native_query(query)
+        assert result.type is RESPONSE_TYPE.TABLE
+
+    def test_2_get_tables(self):
+        tables = self.handler.get_tables()
+        assert tables.type is not RESPONSE_TYPE.ERROR
+
+    def test_4_get_columns(self):
+        columns = self.handler.get_columns('customers')
+        assert columns.type is not RESPONSE_TYPE.ERROR
+
+
+if __name__ == '__main__':
+    unittest.main()

mindsdb/integrations/handlers/sqreamdb_handler/__about__.py

@@ -0,0 +1,9 @@
+__title__ = 'MindsDB SQreamDB handler'
+__package_name__ = 'mindsdb_sqreamdb_handler'
+__version__ = '0.0.1'
+__description__ = "MindsDB handler for SQreamDB"
+__author__ = 'Parthiv Makwana'
+__github__ = 'https://github.com/mindsdb/mindsdb'
+__pypi__ = 'https://pypi.org/project/mindsdb/'
+__license__ = 'MIT'
+__copyright__ = 'Copyright 2023- mindsdb'

mindsdb/integrations/handlers/sqreamdb_handler/__init__.py

@@ -0,0 +1,20 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+
+from .__about__ import __version__ as version, __description__ as description
+from .connection_args import connection_args, connection_args_example
+try:
+    from .sqreamdb_handler import SQreamDBHandler as Handler
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+
+title = 'SQreamDB'
+name = 'sqreamdb'
+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/sqreamdb_handler/connection_args.py

@@ -0,0 +1,48 @@
+from collections import OrderedDict
+
+from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
+
+
+connection_args = OrderedDict(
+    host={
+        'type': ARG_TYPE.STR,
+        'description': 'The host name or IP address of the SQreamDB server/database.'
+    },
+    user={
+        'type': ARG_TYPE.STR,
+        'description': 'The user name used to authenticate with the SQreamDB server.'
+    },
+    password={
+        'type': ARG_TYPE.PWD,
+        'description': 'The password to authenticate the user with the SQreamDB server.',
+        'secret': True
+    },
+    port={
+        'type': ARG_TYPE.INT,
+        'description': 'Specify port to connect SQreamDB server'
+    }, 
+    database={
+        'type': ARG_TYPE.STR,
+        'description': 'Specify database name  to connect SQreamDB server'
+    },
+    service={
+        'type': ARG_TYPE.STR,
+        'description': 'Optional: service queue (default: "sqream")'
+    },
+    use_ssl={
+        'type': ARG_TYPE.BOOL,
+        'description': 'use SSL connection (default: False)'
+    },
+    clustered={
+        'type': ARG_TYPE.BOOL,
+        'description': 'Optional:  Connect through load balancer, or direct to worker (Default: false - direct to worker)'
+    },
+)
+
+connection_args_example = OrderedDict(
+    host='127.0.0.1',
+    port=5000,
+    password='sqream',
+    user='master',
+    database='sqream'
+)

mindsdb/integrations/handlers/sqreamdb_handler/icon.svg

@@ -0,0 +1,21 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="40" viewBox="0 0 40 40" fill="none">
+  <g clip-path="url(#clip0_155_2068)">
+    <path d="M3.46641 24.9984C3.75397 24.9885 4.10256 24.974 4.4515 24.9653C5.06453 24.9497 5.67767 24.9379 6.29081 24.9243C6.56687 24.9182 6.84245 24.9115 7.11921 24.9048C7.45004 24.8976 7.78091 24.8908 8.11182 24.8844C8.18153 24.883 8.26111 24.8955 8.29808 24.8133C8.33869 24.7235 8.27614 24.6573 8.23189 24.5906C8.22616 24.5828 8.21996 24.5756 8.21334 24.5687C7.82211 24.1142 7.43087 23.6598 7.03964 23.2055C6.69375 22.8049 6.34586 22.4063 5.99997 22.0054C5.87708 21.8631 5.75689 21.718 5.63694 21.5728C5.54862 21.4535 5.43587 21.3586 5.3078 21.2958C5.17974 21.233 5.03995 21.204 4.89974 21.2112C4.78166 21.2181 4.66359 21.2259 4.54528 21.2291C4.25091 21.2372 3.95655 21.2428 3.66218 21.2494L2.77897 21.2692C2.4846 21.2758 2.19012 21.2839 1.89575 21.2889C1.40691 21.2973 0.917941 21.3029 0.428623 21.3116C0.325689 21.3134 0.22252 21.3209 0.119938 21.3306C0.00714523 21.3414 -0.0291223 21.4128 0.022286 21.5213C0.0420528 21.5623 0.0664195 21.6004 0.0948211 21.6348C0.173811 21.731 0.255501 21.8247 0.336604 21.919C0.725727 22.3711 1.11481 22.8232 1.50385 23.2754C1.75608 23.5675 2.0109 23.8569 2.26312 24.149C2.38589 24.2913 2.50737 24.4353 2.62592 24.5818C2.72392 24.7214 2.85189 24.8325 2.99848 24.9051C3.14508 24.9778 3.30579 25.0098 3.46641 24.9984Z" fill="#396AB3"/>
+    <path d="M14.7472 24.7161C14.6885 24.6427 14.6473 24.5863 14.6021 24.5333C14.3304 24.2158 14.0582 23.8987 13.7855 23.5821C13.3374 23.0631 12.8884 22.5449 12.4386 22.0275C12.3602 21.9363 12.2649 21.8645 12.1591 21.8166C12.0533 21.7688 11.9394 21.7462 11.825 21.7502C11.7098 21.7515 11.5948 21.7642 11.4796 21.7674C11.1852 21.7755 10.8906 21.7814 10.5962 21.7881L9.70421 21.8081L8.8298 21.8277L7.95551 21.8476C7.66102 21.8544 7.36642 21.8583 7.07218 21.8706C7.01748 21.8729 6.93063 21.8963 6.9183 21.9347C6.90117 21.9861 6.93802 22.0617 6.95715 22.1248C6.96807 22.1486 6.98333 22.1697 7.00199 22.1868C7.28031 22.5113 7.55879 22.8355 7.83743 23.1595C8.18352 23.5602 8.53011 23.9605 8.87722 24.3602C8.99811 24.4999 9.1129 24.6463 9.23719 24.7824C9.27403 24.8191 9.32166 24.8401 9.37147 24.8418C9.6325 24.841 9.89353 24.8311 10.1546 24.8251C10.4551 24.8181 10.7557 24.8113 11.0563 24.8048L11.9305 24.7848C12.228 24.7781 12.5254 24.7703 12.8225 24.7653C13.3359 24.7568 13.8492 24.7512 14.3625 24.7417C14.4835 24.7392 14.6033 24.726 14.7472 24.7161Z" fill="#396AB3"/>
+    <path d="M21.0388 24.5566C20.9013 24.3963 20.7797 24.2539 20.657 24.1122C20.5173 23.9506 20.3759 23.7906 20.2366 23.6288C19.8727 23.2061 19.5133 22.7778 19.1427 22.3618C19.0749 22.2847 18.993 22.224 18.9023 22.1837C18.8117 22.1434 18.7143 22.1244 18.6165 22.1278C18.0581 22.1401 17.4997 22.1529 16.9413 22.1663C16.6469 22.1731 16.3524 22.1803 16.0582 22.1862C15.6787 22.1938 15.2992 22.2009 14.9197 22.2073C14.5494 22.214 14.1791 22.221 13.8087 22.2282C13.7814 22.2281 13.7543 22.2322 13.728 22.2404C13.7089 22.2434 13.6908 22.2519 13.6756 22.265C13.6605 22.2782 13.6487 22.2954 13.6416 22.3151C13.6344 22.3347 13.6321 22.3561 13.635 22.377C13.6378 22.398 13.6456 22.4177 13.6576 22.4343C13.6885 22.4875 13.7245 22.5368 13.765 22.5816C13.998 22.8523 14.2333 23.1206 14.4662 23.3915C14.8133 23.7953 15.1595 24.2001 15.5047 24.6059C15.5262 24.6334 15.5533 24.655 15.5839 24.6688C15.6145 24.6826 15.6477 24.6883 15.6807 24.6854C15.8931 24.6758 16.1054 24.6705 16.3177 24.6651C16.6121 24.6576 16.9065 24.6514 17.2008 24.6449L18.0928 24.6249C18.3902 24.6182 18.6876 24.6102 18.9848 24.6052C19.5071 24.5966 20.0294 24.5914 20.5518 24.5816C20.7063 24.579 20.8601 24.5662 21.0388 24.5566Z" fill="#396AB3"/>
+    <path d="M27.3403 24.436C27.3306 24.4182 27.32 24.401 27.3084 24.3847C27.0643 24.0979 26.8201 23.8113 26.5755 23.5249C26.3096 23.2149 26.0458 22.9026 25.7741 22.599C25.7036 22.5133 25.6174 22.4449 25.5211 22.398C25.4248 22.3511 25.3206 22.3269 25.2152 22.3269C24.939 22.3313 24.6627 22.3397 24.3865 22.346C24.0829 22.353 23.7794 22.3598 23.4758 22.3663L22.5832 22.3861C22.2919 22.3926 22.0002 22.4002 21.7089 22.4056C21.2838 22.4134 20.8585 22.4192 20.4334 22.4282C20.385 22.4276 20.3369 22.4363 20.2913 22.4538C20.277 22.4577 20.2637 22.4653 20.2527 22.476C20.2417 22.4866 20.2332 22.5001 20.2279 22.5151C20.2226 22.5302 20.2207 22.5464 20.2224 22.5624C20.2241 22.5784 20.2293 22.5937 20.2375 22.607C20.2537 22.6426 20.2741 22.6757 20.2982 22.7055C20.6693 23.1395 21.041 23.573 21.4132 24.006C21.546 24.161 21.6786 24.3163 21.8079 24.4745C21.8292 24.5043 21.8567 24.5282 21.8882 24.544C21.9196 24.5598 21.9541 24.567 21.9887 24.5652C22.4772 24.5494 22.9659 24.5376 23.4545 24.5252C23.743 24.5179 24.0314 24.5113 24.3198 24.5053C24.6842 24.4978 25.0485 24.4908 25.4129 24.4845C25.8837 24.4765 26.3545 24.4717 26.8254 24.4613C26.9918 24.4586 27.1582 24.445 27.3403 24.436Z" fill="#396AB3"/>
+    <path d="M33.6528 24.2887C33.551 24.1732 33.4641 24.077 33.3795 23.9783C33.043 23.5862 32.7112 23.1893 32.3694 22.8027C32.2934 22.7113 32.2 22.6391 32.0959 22.5913C31.9917 22.5434 31.8792 22.521 31.7663 22.5256C31.5054 22.5357 31.2444 22.5397 30.9834 22.5458C30.6798 22.5529 30.3763 22.5595 30.0727 22.5658C29.7844 22.5723 29.496 22.5803 29.2077 22.5853C28.7249 22.5936 28.242 22.5993 27.759 22.6076C27.5007 22.6122 27.2425 22.6162 26.9854 22.6301C26.9567 22.6305 26.9284 22.6377 26.9025 22.6513C26.8766 22.6648 26.8537 22.6844 26.8355 22.7086C26.8307 22.7399 26.8317 22.772 26.8385 22.8029C26.8454 22.8337 26.8578 22.8628 26.8752 22.8883C26.9194 22.9516 26.969 23.0101 27.0234 23.0632C27.3979 23.4982 27.7714 23.9343 28.1485 24.3665C28.1833 24.4014 28.2285 24.4212 28.2757 24.4222C28.4243 24.4232 28.5728 24.4105 28.7217 24.4064C29.0222 24.3983 29.3228 24.3925 29.6234 24.3857L30.5154 24.3657C30.8068 24.3592 31.0981 24.3513 31.3896 24.3463C31.8723 24.338 32.3552 24.3323 32.8382 24.3236C33.0809 24.3194 33.3236 24.3108 33.5659 24.3041C33.5951 24.3007 33.6241 24.2956 33.6528 24.2887Z" fill="#396AB3"/>
+    <path d="M39.9998 24.0986C39.9901 24.0807 39.9794 24.0636 39.9677 24.0472C39.6617 23.6921 39.3556 23.337 39.0493 22.9821C38.9307 22.8411 38.7702 22.7509 38.5963 22.7274C38.4638 22.7169 38.3307 22.7164 38.1981 22.7259C37.9158 22.7313 37.6337 22.7388 37.3514 22.7453L36.4318 22.7659C36.1465 22.7723 35.8612 22.7802 35.5759 22.7852C35.1235 22.7933 34.671 22.7981 34.2188 22.8069C33.9668 22.8115 33.715 22.8198 33.4633 22.8269C33.445 22.8269 33.4268 22.8289 33.409 22.8331C33.3914 22.8351 33.3744 22.8417 33.3597 22.8526C33.345 22.8634 33.333 22.878 33.3247 22.8952C33.3164 22.9123 33.3121 22.9314 33.3121 22.9508C33.3122 22.9702 33.3167 22.9893 33.3251 23.0063C33.3396 23.0465 33.3605 23.0834 33.387 23.1154C33.6535 23.4291 33.925 23.7377 34.1874 24.0552C34.2345 24.1195 34.2958 24.1697 34.3654 24.2011C34.4351 24.2325 34.5109 24.2441 34.5858 24.2348C35.1916 24.1994 35.7994 24.1998 36.4064 24.1862C36.6881 24.1799 36.9709 24.1723 37.2533 24.1667C37.6418 24.159 38.0304 24.152 38.419 24.1456C38.865 24.1381 39.3116 24.1318 39.758 24.1228C39.8332 24.1215 39.9083 24.1083 39.9998 24.0986Z" fill="#396AB3"/>
+    <path fill-rule="evenodd" clip-rule="evenodd" d="M2.35458 19.7718H0.248127C0.125357 19.7718 0.12301 19.7708 0.142493 19.6405C0.178526 19.4016 0.219136 19.1635 0.2547 18.9245C0.266437 18.8417 0.312563 18.83 0.377939 18.8303C0.934158 18.8319 1.49038 18.8312 2.0466 18.8312C2.8642 18.8312 3.68192 18.833 4.49964 18.8286C4.60868 18.8258 4.71694 18.8078 4.82182 18.7751C4.89799 18.7552 4.96688 18.7106 5.01946 18.6471C5.07203 18.5837 5.10583 18.5043 5.11642 18.4195C5.13884 18.2891 5.15163 18.1565 5.1791 18.0274C5.24506 17.7136 5.13039 17.6033 4.82617 17.5607C4.75343 17.5544 4.68039 17.5531 4.6075 17.5569C3.6987 17.5569 2.7899 17.5634 1.8811 17.5522C1.64489 17.5454 1.40995 17.5119 1.18005 17.4522C1.06591 17.4349 0.958427 17.3832 0.869624 17.3029C0.78082 17.2226 0.714205 17.1169 0.677234 16.9976C0.642553 16.8606 0.631802 16.7178 0.645544 16.5764C0.663267 16.3691 0.712093 16.1651 0.743079 15.9589C0.765999 15.7782 0.834861 15.6082 0.941828 15.4681C1.04879 15.3281 1.18957 15.2237 1.34812 15.1669C1.5665 15.0735 1.79863 15.0245 2.0331 15.0223C3.46431 15.0146 4.89537 15.0086 6.32628 15.004C6.43708 15.004 6.44881 15.0208 6.42956 15.1377C6.39024 15.3762 6.34952 15.6142 6.31325 15.8532C6.30339 15.9184 6.27675 15.9472 6.21724 15.943C6.18696 15.9411 6.15644 15.943 6.12604 15.943C4.78865 15.943 3.45129 15.9434 2.11397 15.944C2.02904 15.945 1.94429 15.9529 1.86045 15.9678C1.81735 15.9707 1.77665 15.9904 1.74577 16.0234C1.71489 16.0564 1.69589 16.1004 1.69225 16.1474C1.68204 16.2296 1.66502 16.3107 1.65141 16.3922C1.64419 16.4144 1.64185 16.4382 1.6446 16.4616C1.64735 16.485 1.65511 16.5073 1.66723 16.5268C1.67936 16.5463 1.69553 16.5625 1.71441 16.5739C1.7333 16.5853 1.75438 16.5916 1.77594 16.5925C1.85676 16.6071 1.9385 16.6149 2.02042 16.6156C3.01126 16.6174 4.00222 16.6107 4.99295 16.6224C5.20504 16.6292 5.4154 16.6664 5.61865 16.733C6.07299 16.8731 6.23672 17.2308 6.19377 17.7044C6.16172 18.0144 6.11205 18.3219 6.04506 18.6251C6.00028 18.8839 5.88493 19.122 5.71417 19.3078C5.54341 19.4937 5.32522 19.6188 5.08825 19.6666C4.8522 19.7275 4.61092 19.7611 4.36842 19.7667C3.69812 19.7804 3.02629 19.7718 2.35458 19.7718Z" fill="#396AB3"/>
+    <path fill-rule="evenodd" clip-rule="evenodd" d="M25.84 16.9763C25.8079 17.1766 25.7791 17.3556 25.7501 17.5346C25.7367 17.5925 25.7271 17.6514 25.7214 17.7108C25.7188 17.8172 25.6698 17.8402 25.5805 17.84C24.1851 17.8376 22.7898 17.8372 21.3946 17.8388C21.2322 17.8388 21.2325 17.8145 21.2034 18.0065C21.1629 18.2743 21.1167 18.5412 21.0704 18.8228C21.1218 18.8262 21.1626 18.8314 21.2032 18.8314C22.5986 18.8318 23.9939 18.8318 25.3892 18.8314C25.5478 18.8314 25.5479 18.8321 25.5183 19.0063C25.4812 19.2283 25.4417 19.4498 25.4102 19.6727C25.3993 19.7497 25.3694 19.7775 25.3032 19.772C25.282 19.7702 25.2607 19.772 25.2393 19.772H20.1423C20.1241 19.7709 20.1058 19.7709 20.0876 19.772C20.0068 19.782 19.9905 19.7455 20.0045 19.6613C20.1323 18.8976 20.2581 18.1334 20.3821 17.3688C20.5055 16.6144 20.6301 15.8598 20.7498 15.1048C20.7629 15.0214 20.7967 15.0022 20.8646 15.003C21.0713 15.0051 21.2779 15.0036 21.4846 15.0036H25.9898C26.0228 14.9984 26.0565 14.9997 26.0891 15.0076C26.1144 15.0189 26.1507 15.0589 26.1478 15.0794C26.1097 15.3446 26.0671 15.6093 26.02 15.8734C26.015 15.9006 25.9746 15.9286 25.9455 15.94C25.9127 15.947 25.8791 15.9481 25.8459 15.9433H21.6783C21.5433 15.9433 21.5433 15.9433 21.519 16.0912C21.4755 16.3557 21.4338 16.6207 21.3873 16.8845C21.3756 16.953 21.3939 16.9782 21.4545 16.9755C21.4849 16.9742 21.5154 16.9755 21.5457 16.9755H25.84V16.9763Z" fill="#396AB3"/>
+    <path fill-rule="evenodd" clip-rule="evenodd" d="M16.931 18.0992L18.7738 19.7386L18.7637 19.7664C18.7293 19.7684 18.6949 19.7717 18.6604 19.7717C18.2563 19.7717 17.8519 19.7677 17.4479 19.7748C17.4 19.7772 17.3522 19.769 17.3073 19.7507C17.2624 19.7323 17.2214 19.7042 17.1867 19.668C16.8873 19.3721 16.5868 19.0776 16.2851 18.7847C16.0633 18.5682 15.8424 18.3503 15.6161 18.1394C15.5714 18.1014 15.5167 18.08 15.46 18.0785C15.1806 18.0726 14.9009 18.0785 14.6213 18.0742C14.5406 18.073 14.5113 18.1066 14.4981 18.1917C14.4213 18.6822 14.339 19.1716 14.2604 19.6617C14.2481 19.7387 14.2212 19.7759 14.1418 19.7739C13.9261 19.7681 13.7103 19.7705 13.4945 19.7726C13.4234 19.7732 13.3983 19.7505 13.4124 19.666C13.6277 18.367 13.8415 17.0677 14.0537 15.7681C14.088 15.5592 14.1225 15.3501 14.1517 15.1403C14.1653 15.0427 14.2004 15.0013 14.2987 15.0015C15.5567 15.0049 16.8148 15.0015 18.0729 15.0069C18.2579 15.0094 18.4422 15.033 18.6227 15.0774C19.0804 15.1868 19.3146 15.5157 19.2728 16.025C19.2441 16.3585 19.1909 16.6889 19.1138 17.0131C18.9675 17.6381 18.5631 17.9567 17.9964 18.0335C17.7265 18.0701 17.4517 18.0639 17.179 18.0753C17.1004 18.0787 17.0215 18.0753 16.9428 18.0753L16.931 18.0992ZM16.4498 15.9433C15.9576 15.9433 15.4653 15.9452 14.9733 15.9413C14.8927 15.9413 14.86 15.9715 14.8471 16.0553C14.7935 16.4051 14.7354 16.7543 14.6796 17.1039C14.659 17.2335 14.6598 17.2345 14.7766 17.2345C15.7519 17.2345 16.7272 17.2339 17.7025 17.2326C17.7996 17.2311 17.896 17.2159 17.9895 17.1873C18.0674 17.1685 18.1382 17.1239 18.1922 17.0596C18.2463 16.9953 18.2809 16.9145 18.2915 16.828C18.3171 16.668 18.3437 16.5081 18.3714 16.3484C18.382 16.3091 18.3849 16.2678 18.3797 16.2272C18.3746 16.1867 18.3616 16.1478 18.3416 16.1131C18.3216 16.0785 18.2951 16.0489 18.2638 16.0264C18.2325 16.0039 18.1971 15.9889 18.1601 15.9825C18.0749 15.9593 17.9875 15.9467 17.8996 15.9449C17.4165 15.9417 16.9335 15.9435 16.4503 15.9437L16.4498 15.9433Z" fill="#396AB3"/>
+    <path fill-rule="evenodd" clip-rule="evenodd" d="M11.6172 20.2788C11.578 20.2835 11.5492 20.29 11.5204 20.29C11.3137 20.2909 11.107 20.2884 10.9005 20.2916C10.8871 20.2941 10.8734 20.2936 10.8602 20.2903C10.847 20.2869 10.8345 20.2808 10.8235 20.2722C10.8124 20.2635 10.8031 20.2526 10.7959 20.24C10.7887 20.2275 10.7839 20.2135 10.7817 20.1988C10.7574 20.0995 10.7223 20.0031 10.7009 19.9037C10.6796 19.8042 10.6338 19.7692 10.5358 19.7697C9.73942 19.7745 8.94306 19.7761 8.14682 19.7697C7.89042 19.7792 7.63588 19.7189 7.4061 19.5942C7.302 19.5348 7.21333 19.4476 7.14835 19.3407C7.08338 19.2338 7.04421 19.1106 7.0345 18.9827C6.99929 18.6988 7.06596 18.4287 7.1095 18.1555C7.21056 17.5214 7.31525 16.8881 7.42006 16.2548C7.46097 15.9708 7.57228 15.7045 7.74189 15.4848C7.89268 15.2946 8.09592 15.1631 8.321 15.1101C8.54485 15.0508 8.77366 15.0168 9.00386 15.0085C9.81524 14.9964 10.627 14.9986 11.4384 15.0065C11.7218 14.9917 12.0044 15.0524 12.2617 15.1833C12.3737 15.2446 12.4681 15.3383 12.5348 15.4541C12.6015 15.57 12.638 15.7037 12.6401 15.8406C12.6453 16.0963 12.6212 16.3517 12.5682 16.6008C12.4637 17.2342 12.3623 17.8681 12.2564 18.5013C12.2178 18.7683 12.1209 19.0212 11.9739 19.2386C11.8582 19.398 11.7072 19.5226 11.5353 19.6007C11.4649 19.635 11.4523 19.6715 11.4745 19.7438C11.5122 19.8666 11.5462 19.9907 11.5801 20.1147C11.5938 20.1646 11.6031 20.2158 11.6172 20.2788ZM11.1962 18.7439C11.2581 18.7084 11.3106 18.656 11.3484 18.5918C11.3863 18.5276 11.4084 18.4538 11.4124 18.3775C11.485 17.9335 11.5569 17.4894 11.6282 17.045C11.669 16.7871 11.7134 16.5294 11.7414 16.2696C11.747 16.2389 11.7469 16.2073 11.741 16.1766C11.7351 16.146 11.7235 16.117 11.7071 16.0914C11.6906 16.0658 11.6695 16.0442 11.6452 16.0277C11.6208 16.0113 11.5937 16.0004 11.5654 15.9958C11.4696 15.9646 11.3704 15.9477 11.2705 15.9457C10.4683 15.9418 9.6663 15.9418 8.86443 15.9457C8.77337 15.9468 8.68276 15.9598 8.59447 15.9842C8.50851 16.0021 8.4301 16.0497 8.3708 16.1201C8.3115 16.1904 8.27446 16.2798 8.26513 16.3749C8.25046 16.4764 8.23344 16.5775 8.21701 16.6788C8.12311 17.2563 8.03297 17.8355 7.93297 18.4123C7.92048 18.4577 7.9177 18.5057 7.92484 18.5525C7.93198 18.5993 7.94885 18.6437 7.97416 18.6823C7.99947 18.7208 8.03254 18.7525 8.07082 18.7749C8.1091 18.7973 8.15156 18.8098 8.19494 18.8114C8.24259 18.8211 8.29078 18.8273 8.33919 18.83C8.98943 18.8312 9.6397 18.8312 10.29 18.83C10.3254 18.8277 10.3607 18.8234 10.3956 18.8171C10.3326 18.594 10.2738 18.3856 10.2149 18.1772C10.184 18.068 10.1914 18.0545 10.2946 18.0538C10.4824 18.0528 10.6713 18.0538 10.8597 18.0538C11.0021 18.0538 11.0033 18.0538 11.0468 18.2067C11.0966 18.3807 11.1439 18.5547 11.1962 18.7439Z" fill="#396AB3"/>
+    <path fill-rule="evenodd" clip-rule="evenodd" d="M29.6869 18.2552C29.1307 18.2552 28.5745 18.2566 28.0183 18.2537C27.9327 18.2537 27.8979 18.2794 27.883 18.3783C27.8171 18.8136 27.7407 19.2468 27.6717 19.6815C27.6611 19.7474 27.6352 19.7726 27.5762 19.7721C27.3543 19.7706 27.1325 19.7706 26.9107 19.7721C26.8384 19.7721 26.822 19.7367 26.8338 19.6657C26.9569 18.918 27.0777 18.1699 27.202 17.4225C27.26 17.0735 27.3057 16.7202 27.3886 16.378C27.5677 15.6373 28.0185 15.2005 28.706 15.064C28.894 15.0273 29.0846 15.008 29.2757 15.0062C29.8804 14.999 30.4853 15.0021 31.09 15.0042C31.3726 14.9935 31.6549 15.033 31.926 15.1211C32.1306 15.1867 32.3094 15.3248 32.4343 15.5137C32.5593 15.7026 32.6232 15.9315 32.616 16.1645C32.6068 16.4896 32.5685 16.8129 32.5017 17.1298C32.3665 17.9768 32.224 18.8212 32.0883 19.6675C32.0756 19.7479 32.0464 19.7736 31.9764 19.7727C31.7637 19.7699 31.5509 19.7694 31.3382 19.7727C31.2524 19.7741 31.236 19.7372 31.2507 19.6504C31.3231 19.2232 31.3889 18.7946 31.4626 18.3671C31.4786 18.2747 31.4563 18.2527 31.3744 18.2531C30.8115 18.2566 30.2491 18.2552 29.6869 18.2552ZM29.8395 17.3754C30.3984 17.3754 30.9572 17.3737 31.5163 17.3773C31.597 17.3773 31.6301 17.3469 31.6421 17.2631C31.6793 17.0045 31.7212 16.7467 31.7604 16.4884C31.7785 16.4 31.7679 16.3075 31.7304 16.2268C31.6929 16.146 31.6309 16.082 31.555 16.0458C31.4671 16.0017 31.3728 15.9747 31.2764 15.966C31.0345 15.9497 30.7914 15.9438 30.5487 15.9448C29.9841 15.9472 29.4195 15.9489 28.855 15.9652C28.473 15.9762 28.2628 16.177 28.1801 16.5853C28.1364 16.8019 28.1053 17.0216 28.0713 17.2408C28.0506 17.3744 28.0531 17.3752 28.1808 17.3753L29.8395 17.3754Z" fill="#396AB3"/>
+    <path fill-rule="evenodd" clip-rule="evenodd" d="M34.4782 16.6181C34.4438 16.8208 34.4131 16.9957 34.3843 17.171C34.2721 17.857 34.1603 18.543 34.0488 19.229C34.0206 19.4021 33.9901 19.5747 33.964 19.7482C33.9638 19.7606 33.9612 19.7728 33.9563 19.784C33.9515 19.7952 33.9445 19.8052 33.9359 19.8132C33.9272 19.8212 33.9171 19.827 33.9062 19.8304C33.8953 19.8338 33.8838 19.8346 33.8726 19.8327C33.6629 19.8315 33.4532 19.8309 33.2435 19.8327C33.1723 19.8335 33.1568 19.8055 33.1696 19.73C33.2691 19.1397 33.3654 18.5481 33.4621 17.957C33.5972 17.1307 33.7322 16.3043 33.8669 15.4777C33.9114 15.2059 34.0341 15.059 34.2687 15.0157C34.3391 14.9986 34.4126 15.005 34.4796 15.0342C34.5466 15.0635 34.604 15.1141 34.6442 15.1795C35.0094 15.7595 35.3709 16.3421 35.734 16.9236C35.9361 17.2473 36.1374 17.5715 36.3416 17.8936C36.4148 18.0091 36.4182 18.0067 36.506 17.9103C36.9385 17.4385 37.3711 16.9667 37.8037 16.4949C38.2069 16.0573 38.6115 15.6212 39.0175 15.1867C39.0755 15.1176 39.149 15.0661 39.2306 15.0373C39.3123 15.0085 39.3994 15.0032 39.4834 15.0222C39.5224 15.0302 39.5596 15.0467 39.5926 15.0707C39.6257 15.0947 39.654 15.1257 39.6759 15.1619C39.6978 15.1981 39.7128 15.2387 39.7201 15.2814C39.7273 15.324 39.7267 15.3678 39.7182 15.4102C39.6164 16.0376 39.5143 16.6649 39.4118 17.292C39.2796 18.1022 39.1462 18.9119 39.0166 19.7225C39.0035 19.8045 38.9747 19.8355 38.8993 19.834C38.7048 19.8299 38.5102 19.8332 38.3156 19.8324C38.2087 19.8324 38.2019 19.8206 38.2209 19.7041C38.3834 18.7078 38.5456 17.7115 38.7074 16.7151C38.7103 16.6968 38.71 16.6781 38.7127 16.6329C38.6711 16.674 38.6443 16.698 38.62 16.7246C37.9339 17.4738 37.248 18.2233 36.5624 18.973C36.4935 19.0584 36.3998 19.1153 36.2971 19.1339C36.2249 19.1473 36.1507 19.1361 36.0845 19.1018C36.0183 19.0676 35.9634 19.0119 35.9271 18.9423C35.4634 18.1907 34.9945 17.4428 34.5273 16.6938C34.5169 16.677 34.5062 16.6612 34.4782 16.6181Z" fill="#396AB3"/>
+  </g>
+  <defs>
+    <clipPath id="clip0_155_2068">
+      <rect width="40" height="40" fill="white"/>
+    </clipPath>
+  </defs>
+</svg>

mindsdb/integrations/handlers/sqreamdb_handler/requirements.txt

@@ -0,0 +1,2 @@
+pysqream>=3.2.5
+pysqream_sqlalchemy>=0.8

mindsdb/integrations/handlers/sqreamdb_handler/sqreamdb_handler.py

@@ -0,0 +1,154 @@
+from typing import Optional
+
+from mindsdb_sql_parser.ast.base import ASTNode
+
+from mindsdb.integrations.libs.base import DatabaseHandler
+from mindsdb.utilities import log
+from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
+from mindsdb.integrations.libs.response import (
+    HandlerStatusResponse as StatusResponse,
+    HandlerResponse as Response,
+    RESPONSE_TYPE
+)
+
+import pandas as pd
+import pysqream as db
+
+from pysqream_sqlalchemy.dialect import SqreamDialect 
+
+logger = log.getLogger(__name__)
+
+
+class SQreamDBHandler(DatabaseHandler):
+
+
+    name= 'sqreamdb'
+
+    def __init__(self, name: str, connection_data: Optional[dict], **kwargs):
+        """ Initialize the handler
+        Args:
+            name (str): name of particular handler instance
+            connection_data (dict): parameters for connecting to the database
+            **kwargs: arbitrary keyword arguments.
+        """
+        super().__init__(name)
+        
+        
+        self.connection_data = connection_data
+
+        self.connection = None
+        self.is_connected = False
+          
+ 
+    def connect(self):
+        """
+        Handles the connection to a YugabyteSQL database insance.
+        """
+        if self.is_connected is True:
+            return self.connection
+        
+        args={
+            "database": self.connection_data.get('database'),
+            "host": self.connection_data.get('host'),
+            "port": self.connection_data.get('port'),
+            "username": self.connection_data.get('user'),
+            "password": self.connection_data.get('password'),
+            "clustered":self.connection_data.get('clustered',False),
+            "use_ssl":self.connection_data.get('use_ssl',False), 
+            "service":self.connection_data.get('service','sqream')
+        }
+
+
+
+        connection = db.connect(**args)
+
+        self.is_connected = True
+        self.connection = connection
+        return self.connection
+
+    def check_connection(self) -> StatusResponse:
+        """
+        Check the connection of the SQreamDB  database
+        :return: success status and error message if error occurs
+        """
+        response = StatusResponse(False)
+        need_to_close = self.is_connected is False
+
+        try:
+            connection = self.connect()
+            with connection.cursor() as cur:
+                cur.execute('select 1;')
+            response.success = True
+        except db.Error as e:
+            logger.error(f'Error connecting to SQreamDB {self.database}, {e}!')
+            response.error_message = e
+
+        if response.success is True and need_to_close:
+            self.disconnect()
+        if response.success is False and self.is_connected is True:
+            self.is_connected = False
+
+        return response
+
+    def native_query(self, query: str) -> StatusResponse:
+        """Receive raw query and act upon it somehow.
+        Args:
+            query (Any): query in native format (str for sql databases,
+                dict for mongo, etc)
+        Returns:
+            HandlerResponse
+        """
+        need_to_close = self.is_connected is False
+        conn = self.connect()
+        with conn.cursor() as cur:
+            try:
+                cur.execute(query)
+                   
+                if cur.rowcount >0 and query.upper().startswith('SELECT') :
+                    result = cur.fetchall() 
+                    response = Response(
+                        RESPONSE_TYPE.TABLE,
+                        data_frame=pd.DataFrame(
+                            result,
+                            columns=[x[0] for x in cur.description]
+                        )
+                    )
+                else:
+                    response = Response(RESPONSE_TYPE.OK)
+                self.connection.commit()
+            except Exception as e:
+                logger.error(f'Error running query: {query} on {self.database}!')
+                response = Response(
+                    RESPONSE_TYPE.ERROR,
+                    error_message=str(e)
+                )
+                self.connection.rollback()
+
+        if need_to_close is True:
+            self.disconnect()
+
+        return response
+
+    def query(self, query: ASTNode) -> Response:
+        """
+        Retrieve the data from the SQL statement
+        """
+        renderer = SqlalchemyRender(SqreamDialect)
+        query_str = renderer.get_string(query, with_failback=True)
+        return self.native_query(query_str)
+
+    def get_tables(self) -> Response:
+        """
+        List all tables in SQreamDB stored in 'sqream_catalog' 
+        """
+
+        query = "SELECT table_name FROM sqream_catalog.tables"
+
+        return self.query(query)
+
+    def get_columns(self, table_name):
+        query = f"""SELECT column_name, type_name 
+        FROM sqream_catalog.columns 
+        WHERE table_name = '{table_name}';
+        """
+        return self.query(query)

mindsdb/integrations/handlers/sqreamdb_handler/tests/test_sqreamdb_handler.py

@@ -0,0 +1,45 @@
+import unittest
+from mindsdb.integrations.handlers.sqreamdb_handler.sqreamdb_handler import SQreamDBHandler
+from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE
+
+
+class SQreamDBHandlerTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.kwargs = {
+            "connection_data": {
+                "host": "127.0.0.1",
+                "port": "5000",
+                "user": "sqream",
+                "password": "sqream",
+                "database": "master"
+            }
+        }
+        cls.handler = SQreamDBHandler('test_sqreamdb_handler', **cls.kwargs)
+
+    def test_0_connect(self):
+        self.handler.connect()
+
+    def test_1_drop_table(self):
+        res = self.handler.query("DROP TABLE IF EXISTS LOVE")
+        assert res.type is not RESPONSE_TYPE.ERROR
+
+    def test_2_create_table(self):
+        res = self.handler.query("CREATE TABLE IF NOT EXISTS LOVE (LOVER varchar(20))")
+        assert res.type is not RESPONSE_TYPE.ERROR
+
+    def test_3_get_tables(self):
+        tables = self.handler.get_tables()
+        assert tables.type is not RESPONSE_TYPE.ERROR
+ 
+    def test_4_select_query(self):
+        query = "SELECT * FROM AUTHORS"
+        result = self.handler.native_query(query)
+        assert result.type is RESPONSE_TYPE.TABLE
+
+    def test_5_check_connection(self):
+         self.handler.check_connection()
+
+        
+if __name__ == '__main__':
+    unittest.main()

mindsdb/integrations/handlers/stabilityai_handler/__about__.py

@@ -0,0 +1,9 @@
+__title__ = "MindsDB Stability AI handler"
+__package_name__ = "mindsdb_stabilityai_handler"
+__version__ = "0.0.1"
+__description__ = "MindsDB handler for Stability AI"
+__author__ = "Abhilash K R"
+__github__ = "https://github.com/mindsdb/mindsdb"
+__pypi__ = "https://pypi.org/project/mindsdb/"
+__license__ = "MIT"
+__copyright__ = "Copyright 2023 - mindsdb"

mindsdb/integrations/handlers/stabilityai_handler/__init__.py

@@ -0,0 +1,19 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+
+from .__about__ import __version__ as version, __description__ as description
+try:
+    from .stabilityai_handler import StabilityAIHandler as Handler
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+
+title = 'Stability AI'
+name = 'stabilityai'
+type = HANDLER_TYPE.ML
+icon_path = 'icon.svg'
+permanent = False
+
+__all__ = [
+    'Handler', 'version', 'name', 'type', 'title', 'description', 'import_error', 'icon_path'
+]

mindsdb/integrations/handlers/stabilityai_handler/icon.svg

@@ -0,0 +1,10 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="40" viewBox="0 0 40 40" fill="none">
+  <path d="M11.3154 36.1953C19.0462 36.1953 24.0744 32.1007 24.0744 25.9412C24.0744 21.1644 21.0162 18.1291 15.548 16.872L12.0372 15.819C8.95744 15.1276 7.15981 14.2974 7.58335 12.1765C7.9357 10.4121 8.98982 9.41601 11.4411 9.41601C19.2336 9.41601 22.1208 12.1765 22.1208 12.1765V5.53709C22.1208 5.53709 19.3099 3 11.4411 3C4.02447 3 0 6.81622 0 12.7244C0 17.5011 2.78896 20.2791 8.44569 21.599C8.84158 21.698 9.04393 21.7488 9.05272 21.7516C9.91315 22.0179 11.0742 22.3697 12.5359 22.8071C15.4271 23.4984 16.1706 24.2322 16.1706 26.4321C16.1706 28.4434 14.0809 29.5866 11.3154 29.5866C3.34874 29.5866 0 25.5564 0 25.5564V32.9123C0 32.9123 2.09414 36.1953 11.3154 36.1953Z" fill="url(#paint0_linear_155_2082)"/>
+  <path d="M35.2254 35.6979C37.9621 35.6979 40 33.7182 40 31.0397C40 28.303 38.0203 26.3815 35.2254 26.3815C32.4887 26.3815 30.5089 28.303 30.5089 31.0397C30.5089 33.7764 32.4887 35.6979 35.2254 35.6979Z" fill="#E80000"/>
+  <defs>
+    <linearGradient id="paint0_linear_155_2082" x1="1203.72" y1="3" x2="1203.72" y2="3322.53" gradientUnits="userSpaceOnUse">
+      <stop stop-color="#9D39FF"/>
+      <stop offset="1" stop-color="#A380FF"/>
+    </linearGradient>
+  </defs>
+</svg>

mindsdb/integrations/handlers/stabilityai_handler/requirements.txt

@@ -0,0 +1,2 @@
+stability-sdk
+pillow