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

mindsdb/integrations/handlers/scylla_handler/__about__.py

@@ -0,0 +1,9 @@
+__title__ = 'MindsDB Scylla handler'
+__package_name__ = 'mindsdb_scylla_handler'
+__version__ = '0.0.1'
+__description__ = "MindsDB handler for Scylla"
+__author__ = 'MindsDB Inc'
+__github__ = 'https://github.com/mindsdb/mindsdb'
+__pypi__ = 'https://pypi.org/project/mindsdb/'
+__license__ = 'MIT'
+__copyright__ = 'Copyright 2022- mindsdb'

mindsdb/integrations/handlers/scylla_handler/__init__.py

@@ -0,0 +1,21 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+
+from .__about__ import __version__ as version, __description__ as description
+from .connection_args import connection_args
+try:
+    from .scylla_handler import ScyllaHandler as Handler
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+
+
+title = 'ScyllaDB'
+name = 'scylladb'
+type = HANDLER_TYPE.DATA
+icon_path = 'icon.svg'
+
+__all__ = [
+    'Handler', 'version', 'name', 'type', 'title', 'description',
+    'connection_args', 'import_error', 'icon_path'
+]

mindsdb/integrations/handlers/scylla_handler/connection_args.py

@@ -0,0 +1,50 @@
+from collections import OrderedDict
+
+from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
+
+
+connection_args = OrderedDict(
+    user={
+        'type': ARG_TYPE.STR,
+        'description': 'User name',
+        'required': True,
+        'label': 'User'
+    },
+    password={
+        'type': ARG_TYPE.PWD,
+        'description': 'Password',
+        'required': True,
+        'label': 'Password',
+        'secret': True
+    },
+    protocol_version={
+        'type': ARG_TYPE.INT,
+        'description': 'is not required, and default to 4.',
+        'required': False,
+        'label': 'Protocol version'
+    },
+    host={
+        'type': ARG_TYPE.STR,
+        'description': ' is the host name or IP address of the ScyllaDB.',
+        'required': True,
+        'label': 'Host'
+    },
+    port={
+        'type': ARG_TYPE.INT,
+        'description': 'Server port',
+        'required': True,
+        'label': 'Port'
+    },
+    keyspace={
+        'type': ARG_TYPE.STR,
+        'description': 'is the keyspace to connect to. It is a top level container for tables.',
+        'required': True,
+        'label': 'Keyspace'
+    },
+    secure_connect_bundle={
+        'type': ARG_TYPE.STR,
+        'description': 'Path or URL to the secure connect bundle',
+        'required': True,
+        'label': 'Host'
+    }
+)

mindsdb/integrations/handlers/scylla_handler/icon.svg

@@ -0,0 +1,17 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="39" viewBox="0 0 40 39" fill="none">
+  <g clip-path="url(#clip0_155_1865)">
+    <path d="M19.8792 0.238098C10.8226 0.238098 4.36224 5.94166 4.36224 13.9867C4.36224 17.8591 8.40753 36.6809 8.40753 36.6809C8.67923 37.7916 9.76602 38.5721 10.9132 38.3619C11.1849 38.3319 11.3962 38.2418 11.6075 38.1218H11.7887C12.1811 38.3619 12.6641 38.452 13.1472 38.422C13.5396 38.3919 13.9019 38.2118 14.2038 38.0017L14.6868 37.9717C15.0792 38.3019 15.6226 38.452 16.166 38.422C16.7396 38.3919 17.2226 38.1518 17.5849 37.7916L18.4604 37.7315C18.8528 38.1218 19.3962 38.422 20 38.422C20.6641 38.422 21.2377 38.1818 21.6604 37.7315L22.566 37.7916C22.8981 38.1218 23.3509 38.3319 23.8641 38.3919C24.4679 38.452 25.0113 38.3019 25.4641 37.9417L26.0075 37.9717C26.2792 38.1518 26.5811 38.3019 26.9132 38.3619C27.3962 38.452 27.849 38.3619 28.2717 38.1818C28.4528 38.2719 28.6339 38.3319 28.8453 38.3619C29.9924 38.542 31.1094 37.7916 31.3509 36.6809C31.3509 36.6809 35.3962 17.8291 35.3962 13.9867C35.366 5.94166 28.9358 0.238098 19.8792 0.238098Z" fill="#57D1E5"/>
+    <path d="M14.8679 19.4803C18.686 19.4803 21.7811 16.4026 21.7811 12.606C21.7811 8.80942 18.686 5.73169 14.8679 5.73169C11.0499 5.73169 7.95471 8.80942 7.95471 12.606C7.95471 16.4026 11.0499 19.4803 14.8679 19.4803Z" fill="white"/>
+    <path d="M14.8679 19.8406C10.8528 19.8406 7.56226 16.5985 7.56226 12.576C7.56226 8.5535 10.8226 5.34149 14.8679 5.34149C18.883 5.34149 22.1736 8.58352 22.1736 12.606C22.1736 16.6285 18.883 19.8406 14.8679 19.8406ZM14.8679 6.09196C11.2453 6.09196 8.31697 9.00378 8.31697 12.606C8.31697 16.2083 11.2453 19.1201 14.8679 19.1201C18.4906 19.1201 21.4189 16.2083 21.4189 12.606C21.4189 9.00378 18.4604 6.09196 14.8679 6.09196Z" fill="#3A2D55"/>
+    <path d="M16.2869 8.10336C15.0793 8.34351 14.868 10.9551 15.5623 11.6756C16.0454 12.1559 16.9812 12.306 16.9812 12.6362C16.9812 12.9664 16.0152 13.1165 15.5623 13.5968C14.8378 14.3172 15.0793 16.9289 16.2869 17.169C17.7057 17.4692 19.8793 15.7581 19.8793 12.6662C19.8793 9.57427 17.7057 7.80317 16.2869 8.10336Z" fill="#3A2D55"/>
+    <path d="M11.4565 21.9717C11.6981 22.6922 12.483 24.043 13.4188 23.8029C13.8716 23.6828 14.1433 23.0824 14.1433 22.1218" fill="white"/>
+    <path d="M17.7358 21.3714C17.9773 22.332 18.2189 23.0525 18.9434 23.0525C19.6679 23.0525 20.634 22.0919 20.3623 20.4409" fill="white"/>
+    <path d="M23.7434 18.0994C23.4415 17.7692 23.0188 17.5291 22.566 17.439H22.5358C22.3245 17.439 22.1434 17.5891 22.1434 17.7692C22.1434 17.9794 22.2943 18.1595 22.4754 18.1595C22.6868 18.1595 22.8679 18.2195 23.049 18.3396C21.2075 19.5704 19.1547 20.4709 17.0113 20.9812C14.7773 21.5216 12.4528 21.6116 10.1585 21.2814C9.97732 21.2514 9.82638 21.3715 9.79619 21.5516C9.766 21.7317 9.88675 21.8818 10.0679 21.9118C10.4 21.9719 10.7622 22.0019 11.1245 22.0319V22.0619C11.4264 22.9625 12.2113 24.1632 13.2075 24.1632C13.2981 24.1632 13.4188 24.1632 13.5396 24.1332C13.9018 24.0432 14.5358 23.6529 14.5358 22.0919C14.5358 22.0619 14.5358 22.0619 14.5358 22.0319C15.4113 21.9719 16.2868 21.8218 17.1622 21.6417C17.2528 21.6417 17.3434 21.6116 17.4037 21.5816C17.6151 22.3921 17.9169 23.3827 18.9132 23.3827C19.3358 23.3827 19.7886 23.1426 20.1207 22.7523C20.4226 22.4221 20.8754 21.6717 20.7245 20.4109C21.7207 19.9606 22.6868 19.4203 23.5622 18.8199C23.6528 18.97 23.7132 19.1201 23.7434 19.3002C23.7735 19.4203 23.8943 19.5103 24.0151 19.5103C24.166 19.5103 24.2566 19.3902 24.2566 19.2402C24.2264 18.8499 24.0452 18.4296 23.7434 18.0994ZM13.7509 22.122C13.7509 22.8424 13.5698 23.3527 13.3283 23.4128C12.7547 23.5629 12.1811 22.7223 11.9094 22.0919C12.5434 22.122 13.1471 22.122 13.7509 22.122C13.7811 22.122 13.7509 22.122 13.7509 22.122ZM19.5471 22.3021C19.366 22.5122 19.1245 22.6623 18.9132 22.6623C18.5811 22.6623 18.3698 22.3621 18.1283 21.4015C18.7622 21.2214 19.3962 21.0113 20 20.7411C20.0302 21.4015 19.8792 21.9418 19.5471 22.3021Z" fill="#3A2D55"/>
+    <path d="M31.4113 3.87056C28.5434 1.31896 24.4981 -0.0919189 20 -0.0919189C15.5019 -0.0919189 11.4868 1.31896 8.58865 3.87056C5.69054 6.42215 4.12073 10.0544 4.12073 14.0469C4.12073 17.8893 8.01507 36.0507 8.16601 36.8312C8.43771 38.0319 9.49431 38.8424 10.6717 38.8424C10.8226 38.8424 10.9434 38.8424 11.0943 38.8124C11.3358 38.7824 11.5773 38.6923 11.7887 38.5722C12.2717 38.7824 12.8151 38.8724 13.3585 38.7824C13.7509 38.7223 14.1434 38.5422 14.4453 38.2721C14.717 38.4822 15.049 38.6623 15.4113 38.7524C15.9849 38.9025 16.5887 38.8424 17.1019 38.6023C17.4641 38.4221 17.7962 38.182 18.0377 37.8818C18.2188 38.122 18.4302 38.3021 18.7019 38.4822C18.9132 38.6023 19.1245 38.6923 19.366 38.7824C19.6075 38.8724 19.8188 38.8724 20.0604 38.8724C20.3019 38.8724 20.5434 38.8424 20.7547 38.7824C20.9962 38.7223 21.2075 38.6323 21.4188 38.4822C21.6302 38.3621 21.8113 38.212 21.9622 38.0319C21.9924 38.0019 22.0528 37.9419 22.083 37.9118C22.4754 38.3921 23.0792 38.7524 23.7132 38.8424C24.3471 38.9325 25.0113 38.7824 25.5245 38.4221C25.5849 38.3921 25.6151 38.3621 25.6754 38.3021C25.9773 38.5422 26.3396 38.7223 26.7019 38.7824C26.9736 38.8424 27.2453 38.8424 27.517 38.8124C27.7585 38.7824 28 38.7223 28.2113 38.6323C28.4226 38.7223 28.6038 38.7824 28.8453 38.8124C30.2038 39.0225 31.5019 38.152 31.7736 36.8312C31.9245 36.0507 35.8188 17.8893 35.8188 14.0469C35.8792 10.0544 34.3094 6.45217 31.4113 3.87056ZM31.1094 36.6811C30.8981 37.5816 30.0528 38.182 29.117 38.0919C29.2981 37.9118 29.449 37.7317 29.5698 37.4916C29.6905 37.2514 29.8113 36.9813 29.8415 36.7411L29.9622 36.0507L30.4151 33.319C30.566 32.3884 30.717 31.4878 30.8075 30.5572C30.8075 30.4372 30.7471 30.3171 30.6264 30.2871C30.5056 30.2571 30.3547 30.3471 30.3245 30.4672C30.1132 31.3677 29.9321 32.2983 29.7509 33.1989L29.2075 35.9306L29.0868 36.621C29.0566 36.8011 28.966 36.9813 28.8754 37.1614C28.7849 37.3415 28.6641 37.4916 28.5132 37.6116C28.3622 37.7317 28.2113 37.8518 28.0302 37.9419C27.849 38.0019 27.6679 38.0619 27.4868 38.0919C27.3056 38.122 27.0943 38.0919 26.9132 38.0619C26.6717 38.0319 26.4302 37.9118 26.2188 37.7617C26.4 37.4916 26.5509 37.2214 26.6113 36.8912C26.6415 36.8312 26.6415 36.7411 26.6415 36.651C26.6415 36.621 26.6415 36.561 26.6415 36.531V36.4409L26.6717 36.0507L26.7622 35.2402L26.9132 33.6191C27.0037 32.5385 27.1245 31.4578 27.1849 30.3771C27.1849 30.2571 27.0943 30.137 26.9736 30.107C26.8528 30.0769 26.7019 30.167 26.6717 30.3171C26.5207 31.3677 26.3698 32.4484 26.2188 33.5291L26.0075 35.1501L25.917 35.9606C25.8868 36.2308 25.8566 36.531 25.7962 36.7111C25.7056 37.1313 25.4037 37.5216 25.0415 37.7617C24.6792 38.0019 24.1962 38.0919 23.7736 38.0319C23.3207 37.9719 22.9283 37.7317 22.6264 37.3715C22.566 37.2814 22.5056 37.1914 22.4453 37.1013C22.4754 36.9813 22.5056 36.8612 22.5358 36.7711C22.5962 36.531 22.5962 36.2908 22.5962 36.1107V34.91C22.5962 34.1295 22.6264 33.319 22.6264 32.5385C22.6264 31.758 22.6264 30.9475 22.6264 30.167C22.6264 30.0469 22.5358 29.9268 22.3849 29.9268C22.2339 29.9268 22.1132 30.0169 22.1132 30.167C22.083 30.9475 22.0226 31.758 21.9924 32.5385C21.9622 33.319 21.932 34.1295 21.8717 34.91L21.8113 36.1107C21.8113 36.2008 21.8113 36.3208 21.8113 36.4109C21.8113 36.5009 21.8113 36.561 21.7811 36.651C21.7509 36.8011 21.6905 36.9813 21.6 37.1013C21.2981 37.7017 20.6339 38.0919 19.9698 38.0619C19.8188 38.0619 19.6377 38.0619 19.4868 38.0019C19.3358 37.9719 19.1849 37.8818 19.0339 37.8218C18.732 37.6417 18.5207 37.4015 18.3396 37.1313C18.249 36.9813 18.2188 36.8312 18.1585 36.6811C18.1283 36.531 18.1283 36.3509 18.0981 36.1407L18.0377 34.94C18.0075 34.1595 17.9773 33.349 17.917 32.5685C17.8868 31.788 17.8264 30.9775 17.7962 30.197C17.7962 30.0769 17.6754 29.9569 17.5547 29.9569C17.4037 29.9569 17.3132 30.0769 17.3132 30.197C17.3132 30.9775 17.3132 31.788 17.3132 32.5685C17.3132 33.349 17.3132 34.1595 17.3434 34.94V36.1407C17.3434 36.3208 17.3434 36.561 17.4037 36.8011C17.4339 36.9212 17.4641 37.0413 17.4943 37.1313C17.283 37.4616 17.0113 37.7317 16.649 37.8818C16.2868 38.0619 15.8641 38.0919 15.4717 37.9719C15.0792 37.8818 14.717 37.6417 14.4754 37.3115C14.3547 37.1614 14.2339 36.9813 14.1736 36.7711C14.1434 36.6811 14.1132 36.561 14.1132 36.4709C14.1132 36.4409 14.1132 36.3509 14.083 36.2908L14.0528 36.1107L13.6905 33.1989C13.5698 32.2083 13.4188 31.2477 13.2679 30.2871C13.2679 30.167 13.1471 30.0769 12.9962 30.0769C12.8453 30.0769 12.7547 30.197 12.7547 30.3471C12.8151 31.3377 12.9056 32.2983 12.9962 33.2889L13.2981 36.2308V36.4109C13.2981 36.4709 13.2981 36.531 13.3283 36.621C13.3283 36.7711 13.3887 36.9212 13.4188 37.0413C13.4792 37.2814 13.6302 37.5216 13.7509 37.7317C13.5396 37.8818 13.2981 38.0019 13.0264 38.0319C12.6641 38.0919 12.2717 38.0319 11.9094 37.8818C11.5773 37.7017 11.2754 37.4616 11.0943 37.1013C11.0037 36.9212 10.9434 36.7711 10.883 36.561L10.9434 35.9006L10.4 33.1989C10.2188 32.2983 10.0377 31.3978 9.82639 30.4972C9.7962 30.3771 9.67545 30.2871 9.55469 30.3171C9.43394 30.3171 9.31318 30.4672 9.34337 30.6173C9.46412 31.5479 9.61507 32.4484 9.73582 33.349L10.1887 36.0807L10.3094 36.7711C10.3396 37.0413 10.4604 37.3115 10.5811 37.5216C10.7019 37.7317 10.8528 37.9419 11.0339 38.122C11.0037 38.122 11.0037 38.122 10.9736 38.122C10.0075 38.2721 9.10186 37.6717 8.89054 36.7411C8.86035 36.561 4.84526 17.8893 4.84526 14.107C4.84526 10.3246 6.35469 6.90245 9.07167 4.47093C11.7887 2.03941 15.683 0.718588 19.9698 0.718588C24.2566 0.718588 28.1509 2.06943 30.8679 4.47093C33.5849 6.87244 35.0943 10.2946 35.0943 14.107C35.1547 17.8293 31.1396 36.501 31.1094 36.6811Z" fill="#3A2D55"/>
+  </g>
+  <defs>
+    <clipPath id="clip0_155_1865">
+      <rect width="40" height="39" fill="white"/>
+    </clipPath>
+  </defs>
+</svg>

mindsdb/integrations/handlers/scylla_handler/requirements.txt

@@ -0,0 +1 @@
+scylla-driver

mindsdb/integrations/handlers/scylla_handler/scylla_handler.py

@@ -0,0 +1,205 @@
+import tempfile
+
+import pandas as pd
+import requests
+
+from cassandra.cluster import Cluster
+from cassandra.auth import PlainTextAuthProvider
+from cassandra.util import Date
+
+from mindsdb_sql_parser import parse_sql
+from mindsdb_sql_parser.ast.base import ASTNode
+from mindsdb_sql_parser import ast
+from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
+
+from mindsdb.integrations.libs.base import DatabaseHandler
+from mindsdb.integrations.libs.response import (
+    HandlerStatusResponse as StatusResponse,
+    HandlerResponse as Response,
+    RESPONSE_TYPE
+)
+from mindsdb.utilities import log
+
+logger = log.getLogger(__name__)
+
+class ScyllaHandler(DatabaseHandler):
+    """
+    This handler handles connection and execution of the Scylla statements.
+    """
+    name = 'scylla'
+
+    def __init__(self, name=None, **kwargs):
+        super().__init__(name)
+        self.parser = parse_sql
+        self.connection_args = kwargs.get('connection_data')
+        self.session = None
+        self.is_connected = False
+
+    def download_secure_bundle(self, url, max_size=10*1024*1024):
+        """
+        Downloads the secure bundle from a given URL and stores it in a temporary file.
+        
+        :param url: URL of the secure bundle to be downloaded.
+        :param max_size: Maximum allowable size of the bundle in bytes. Defaults to 10MB.
+        :return: Path to the downloaded secure bundle saved as a temporary file.
+        :raises ValueError: If the secure bundle size exceeds the allowed `max_size`.
+        
+        TODO:
+        - Find a way to periodically clean up or delete the temporary files 
+        after they have been used to prevent filling up storage over time.
+        """
+        response = requests.get(url, stream=True, timeout=10)
+        response.raise_for_status()
+
+        content_length = int(response.headers.get('content-length', 0))
+        if content_length > max_size:
+            raise ValueError("Secure bundle is larger than the allowed size!")
+
+        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+            size_downloaded = 0
+            for chunk in response.iter_content(chunk_size=8192):
+                size_downloaded += len(chunk)
+                if size_downloaded > max_size:
+                    raise ValueError("Secure bundle is larger than the allowed size!")
+                temp_file.write(chunk)
+        return temp_file.name
+
+    def connect(self):
+        """
+        Handles the connection to a Scylla keystore.
+        """
+        if self.is_connected is True:
+            return self.session
+
+        auth_provider = None
+        if any(key in self.connection_args for key in ('user', 'password')):
+            if all(key in self.connection_args for key in ('user', 'password')):
+                auth_provider = PlainTextAuthProvider(
+                    username=self.connection_args['user'], password=self.connection_args['password']
+                )
+            else:
+                raise ValueError("If authentication is required, both 'user' and 'password' must be provided!") 
+
+        connection_props = {
+            'auth_provider': auth_provider
+        }
+        connection_props['protocol_version'] = self.connection_args.get('protocol_version', 4)
+        secure_connect_bundle = self.connection_args.get('secure_connect_bundle')
+
+        if secure_connect_bundle:
+            # Check if the secure bundle is a URL
+            if secure_connect_bundle.startswith(('http://', 'https://')):
+                secure_connect_bundle = self.download_secure_bundle(secure_connect_bundle)
+            connection_props['cloud'] = {
+                'secure_connect_bundle': secure_connect_bundle
+            }
+        else:
+            connection_props['contact_points'] = [self.connection_args['host']]
+            connection_props['port'] = int(self.connection_args['port'])
+
+        cluster = Cluster(**connection_props)
+        session = cluster.connect(self.connection_args.get('keyspace'))
+
+        self.is_connected = True
+        self.session = session
+        return self.session
+
+    def check_connection(self) -> StatusResponse:
+        """
+        Check the connection of the Scylla database
+        :return: success status and error message if error occurs
+        """
+        response = StatusResponse(False)
+
+        try:
+            session = self.connect()
+            # TODO: change the healthcheck
+            session.execute('SELECT release_version FROM system.local').one()
+            response.success = True
+        except Exception as e:
+            logger.error(f'Error connecting to Scylla {self.connection_args["keyspace"]}, {e}!')
+            response.error_message = e
+
+        if response.success is False and self.is_connected is True:
+            self.is_connected = False
+
+        return response
+
+    def prepare_response(self, resp):
+        # replace cassandra types
+        data = []
+        for row in resp:
+            row2 = {}
+            for k, v in row._asdict().items():
+                if isinstance(v, Date):
+                    v = v.date()
+                row2[k] = v
+            data.append(row2)
+        return data
+
+    def native_query(self, query: str) -> Response:
+        """
+        Receive SQL query and runs it
+        :param query: The SQL query to run in MySQL
+        :return: returns the records from the current recordset
+        """
+        session = self.connect()
+        try:
+            resp = session.execute(query).all()
+            resp = self.prepare_response(resp)
+            if resp:
+                response = Response(
+                    RESPONSE_TYPE.TABLE,
+                    pd.DataFrame(
+                        resp
+                    )
+                )
+            else:
+                response = Response(RESPONSE_TYPE.OK)
+        except Exception as e:
+            logger.error(f'Error running query: {query} on {self.connection_args["keyspace"]}!')
+            response = Response(
+                RESPONSE_TYPE.ERROR,
+                error_message=str(e)
+            )
+        return response
+
+    def query(self, query: ASTNode) -> Response:
+        """
+        Retrieve the data from the SQL statement.
+        """
+
+        # remove table alias because Cassandra Query Language doesn't support it
+        if isinstance(query, ast.Select):
+            if isinstance(query.from_table, ast.Identifier) and query.from_table.alias is not None:
+                query.from_table.alias = None
+
+            # remove table name from fields
+            table_name = query.from_table.parts[-1]
+
+            for target in query.targets:
+                if isinstance(target, ast.Identifier):
+                    if target.parts[0] == table_name:
+                        target.parts.pop(0)
+
+        renderer = SqlalchemyRender('mysql')
+        query_str = renderer.get_string(query, with_failback=True)
+        return self.native_query(query_str)
+
+    def get_tables(self) -> Response:
+        """
+        Get a list with all of the tabels in MySQL
+        """
+        q = "DESCRIBE TABLES;"
+        result = self.native_query(q)
+        df = result.data_frame
+        result.data_frame = df.rename(columns={df.columns[0]: 'table_name'})
+        return result
+
+    def get_columns(self, table_name) -> Response:
+        """
+        Show details about the table
+        """
+        q = f"DESCRIBE {table_name};"
+        result = self.native_query(q)
+        return result

mindsdb/integrations/handlers/scylla_handler/tests/test_scylla_handler.py

@@ -0,0 +1,38 @@
+import unittest
+from mindsdb.integrations.handlers.scylla_handler.scylla_handler import ScyllaHandler
+from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE
+
+
+class ScyllaHandlerTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.kwargs = {
+            "connection_data": {
+                "host": "",
+                "port": "9042",
+                "user": "",
+                "password": "",
+                "keyspace": "test_data"
+            }
+        }
+        cls.handler = ScyllaHandler('test_scylla_handler', **cls.kwargs)
+
+    def test_0_connect(self):
+        self.handler.check_connection()
+
+    def test_1_native_query_show_keyspaces(self):
+        dbs = self.handler.native_query("DESC KEYSPACES;")
+        assert dbs['type'] is not RESPONSE_TYPE.ERROR
+
+    def test_2_get_tables(self):
+        tbls = self.handler.get_tables()
+        assert tbls['type'] is not RESPONSE_TYPE.ERROR
+
+    def test_3_describe_table(self):
+        described = self.handler.get_columns("home_rentals")
+        assert described['type'] is RESPONSE_TYPE.TABLE
+
+    def test_4_select_query(self):
+        query = "SELECT * FROM home_rentals WHERE 'id'='3712'"
+        result = self.handler.query(query)
+        assert result['type'] is RESPONSE_TYPE.TABLE

mindsdb/integrations/handlers/sendinblue_handler/__about__.py

@@ -0,0 +1,9 @@
+__title__ = "MindsDB Sendinblue handler"
+__package_name__ = "mindsdb_sendinblue_handler"
+__version__ = "0.0.1"
+__description__ = "MindsDB handler for Sendinblue"
+__author__ = "Minura Punchihewa"
+__github__ = "https://github.com/mindsdb/mindsdb"
+__pypi__ = "https://pypi.org/project/mindsdb/"
+__license__ = "MIT"
+__copyright__ = "Copyright 2023 - mindsdb"

mindsdb/integrations/handlers/sendinblue_handler/__init__.py

@@ -0,0 +1,27 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+
+from .__about__ import __version__ as version, __description__ as description
+
+try:
+    from .sendinblue_handler import SendinblueHandler as Handler
+
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+
+title = "Sendinblue"
+name = "sendinblue"
+type = HANDLER_TYPE.DATA
+icon_path = "icon.svg"
+
+__all__ = [
+    "Handler",
+    "version",
+    "name",
+    "type",
+    "title",
+    "description",
+    "import_error",
+    "icon_path",
+]

mindsdb/integrations/handlers/sendinblue_handler/icon.svg

@@ -0,0 +1,3 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="40" viewBox="0 0 40 40" fill="none">
+  <path d="M34.3313 28.4355C33.3389 30.1496 31.7151 31.4125 29.8207 31.9538C30.0913 31.0066 30.2266 30.0594 30.2266 29.067C30.2266 24.2407 26.8888 20.1811 22.4233 19.0535C23.3705 18.1514 24.4982 17.4748 25.7611 17.1591C27.7007 16.6178 29.8207 16.8884 31.5347 17.9258C35.1883 20.0909 36.4512 24.782 34.3313 28.4355ZM19.9876 36.7802C18.0029 36.7802 16.1085 35.9682 14.7102 34.6151C15.6574 34.3895 16.5595 34.0287 17.3714 33.5325C21.5212 31.0968 23.3705 26.1802 22.0624 21.7147C25.265 22.6169 27.6105 25.5939 27.6105 29.067C27.6105 33.3521 24.1824 36.7802 19.9876 36.7802ZM5.64387 28.4355C4.65154 26.7215 4.38091 24.6466 4.87707 22.7522C5.55366 23.4288 6.32046 24.0603 7.13237 24.5564C8.66597 25.4585 10.4251 25.9547 12.2293 25.9547C15.0259 25.9547 17.6872 24.8271 19.6267 22.7973C20.4386 26.0449 19.0403 29.5632 16.0634 31.2772C14.8906 31.9538 13.5825 32.3146 12.2744 32.3146C9.52298 32.2695 6.99705 30.8262 5.64387 28.4355ZM5.64387 11.7914C6.6362 10.0774 8.26002 8.81444 10.1545 8.27317C9.88383 9.2204 9.74852 10.1676 9.74852 11.16C9.74852 15.9863 13.0864 20.0458 17.5518 21.1735C15.1612 23.4739 11.4625 24.0603 8.44044 22.3011C6.68131 21.2637 5.41834 19.6399 4.87707 17.6552C4.3358 15.6705 4.60643 13.5506 5.64387 11.7914ZM19.9876 3.44683C21.9722 3.44683 23.8667 4.25874 25.265 5.61192C24.3177 5.83745 23.4156 6.1983 22.6037 6.69446C20.2131 8.09275 18.5442 10.3029 17.8225 12.9642C17.3263 14.7684 17.3714 16.708 17.8676 18.5122C14.6651 17.6101 12.3196 14.6331 12.3196 11.16C12.3647 6.91999 15.7927 3.44683 19.9876 3.44683ZM34.3313 11.7914C35.3236 13.5506 35.5942 15.5803 35.0981 17.4748C34.4215 16.7531 33.6547 16.1667 32.8428 15.6705C30.4522 14.2723 27.7007 13.9114 25.0394 14.6331C23.2352 15.1293 21.5663 16.1216 20.3033 17.4748C19.9876 16.2118 19.9876 14.9037 20.3484 13.6408C20.8897 11.6561 22.1527 9.9872 23.9118 8.99487C25.6709 7.95743 27.7458 7.68679 29.6854 8.22806C31.67 8.76934 33.2938 10.0323 34.3313 11.7914ZM36.6317 10.4834C35.2785 8.09275 33.0683 6.37873 30.407 5.65703C29.7756 5.47661 29.1441 5.38639 28.5126 5.34129C26.6181 2.54472 23.4156 0.830688 20.0327 0.830688C16.5144 0.830688 13.4021 2.63493 11.5528 5.34129C8.16981 5.61192 5.1026 7.50637 3.38857 10.4834C2.03539 12.874 1.67455 15.6255 2.35114 18.2867C2.53156 18.9182 2.75709 19.5046 3.02773 20.0909C1.53923 23.1581 1.67455 26.7666 3.38857 29.7436C5.14771 32.8108 8.26002 34.6151 11.5076 34.8857C13.4021 37.6823 16.6046 39.3963 19.9876 39.3963C23.5058 39.3963 26.6181 37.5921 28.4675 34.8857C31.8504 34.6151 34.9176 32.7206 36.6317 29.7436C38.3908 26.6764 38.3908 23.0679 36.9925 20.0909C38.4359 17.0688 38.3006 13.4153 36.6317 10.4834Z" fill="#0092FF"/>
+</svg>

mindsdb/integrations/handlers/sendinblue_handler/requirements.txt

@@ -0,0 +1 @@
+sib_api_v3_sdk

mindsdb/integrations/handlers/sendinblue_handler/sendinblue_handler.py

@@ -0,0 +1,94 @@
+import sib_api_v3_sdk
+
+from mindsdb.integrations.handlers.sendinblue_handler.sendinblue_tables import EmailCampaignsTable
+from mindsdb.integrations.libs.api_handler import APIHandler
+from mindsdb.integrations.libs.response import (
+    HandlerStatusResponse as StatusResponse,
+)
+
+from mindsdb.utilities import log
+from mindsdb_sql_parser import parse_sql
+
+logger = log.getLogger(__name__)
+
+class SendinblueHandler(APIHandler):
+    """
+    The Sendinblue handler implementation.
+    """
+
+    name = 'sendinblue'
+
+    def __init__(self, name: str, **kwargs):
+        """
+        Initialize the handler.
+        Args:
+            name (str): name of particular handler instance
+            **kwargs: arbitrary keyword arguments.
+        """
+        super().__init__(name)
+
+        connection_data = kwargs.get("connection_data", {})
+        self.connection_data = connection_data
+        self.kwargs = kwargs
+
+        self.connection = None
+        self.is_connected = False
+
+        email_campaigns_data = EmailCampaignsTable(self)
+        self._register_table("email_campaigns", email_campaigns_data)
+
+    def connect(self):
+        """
+        Set up the connection required by the handler.
+        Returns
+        -------
+        StatusResponse
+            connection object
+        """
+        if self.is_connected is True:
+            return self.connection
+
+        configuration = sib_api_v3_sdk.Configuration()
+        configuration.api_key['api-key'] = self.connection_data['api_key']
+
+        self.connection = sib_api_v3_sdk.ApiClient(configuration)
+
+        self.is_connected = True
+
+        return self.connection
+
+    def check_connection(self) -> StatusResponse:
+        """
+        Check connection to the handler.
+        Returns:
+            HandlerStatusResponse
+        """
+
+        response = StatusResponse(False)
+
+        try:
+            connection = self.connect()
+            api_instance = sib_api_v3_sdk.AccountApi(connection)
+            api_instance.get_account()
+            response.success = True
+        except Exception as e:
+            logger.error(f'Error connecting to Sendinblue!')
+            response.error_message = str(e)
+
+        self.is_connected = response.success
+
+        return response
+
+    def native_query(self, query: str) -> StatusResponse:
+        """Receive and process a raw query.
+        Parameters
+        ----------
+        query : str
+            query in a native format
+        Returns
+        -------
+        StatusResponse
+            Request status
+        """
+        ast = parse_sql(query)
+        return self.query(ast)