add e6data query runner with test cases (#6604)

* add e6data query runner with test cases

* added more test cases

pyproject.toml

@@ -137,6 +137,7 @@ thrift-sasl = ">=0.1.0"
 trino = ">=0.305,<1.0"
 vertica-python = "1.1.1"
 xlrd = "2.0.1"
+e6data-python-connector = "1.1.9"
 
 [tool.poetry.group.ldap3]
 optional = true
@@ -173,4 +174,4 @@ select = ["C9", "E", "F", "W", "I001"]
 max-complexity = 15
 
 [tool.ruff.per-file-ignores]
-"__init__.py" = ["F401"]
+"__init__.py" = ["F401"]

redash/query_runner/e6data.py

@@ -0,0 +1,154 @@
+import logging
+
+from redash.query_runner import (
+    TYPE_BOOLEAN,
+    TYPE_DATE,
+    TYPE_DATETIME,
+    TYPE_FLOAT,
+    TYPE_INTEGER,
+    TYPE_STRING,
+    BaseQueryRunner,
+    register,
+)
+
+try:
+    from e6data_python_connector import Connection
+
+    enabled = True
+
+except ImportError:
+    enabled = False
+
+from redash.utils import json_dumps
+
+logger = logging.getLogger(__name__)
+
+E6DATA_TYPES_MAPPING = {
+    "INT": TYPE_INTEGER,
+    "BYTE": TYPE_INTEGER,
+    "INTEGER": TYPE_INTEGER,
+    "LONG": TYPE_INTEGER,
+    "SHORT": TYPE_INTEGER,
+    "FLOAT": TYPE_FLOAT,
+    "DOUBLE": TYPE_FLOAT,
+    "STRING": TYPE_STRING,
+    "DATETIME": TYPE_DATETIME,
+    "BINARY": TYPE_INTEGER,
+    "ARRAY": TYPE_STRING,
+    "MAP": TYPE_STRING,
+    "STRUCT": TYPE_STRING,
+    "UNION_TYPE": TYPE_STRING,
+    "DECIMAL_TYPE": TYPE_FLOAT,
+    "DATE": TYPE_DATE,
+    "INT96": TYPE_INTEGER,
+    "BOOLEAN": TYPE_BOOLEAN,
+    "CHAR": TYPE_STRING,
+}
+
+
+class e6data(BaseQueryRunner):
+    limit_query = " LIMIT 1000"
+
+    should_annotate_query = False
+
+    def __init__(self, configuration):
+        super().__init__(configuration)
+        self.connection = Connection(
+            host=self.configuration.get("host"),
+            port=self.configuration.get("port"),
+            username=self.configuration.get("username"),
+            database=self.configuration.get("database"),
+            password=self.configuration.get("password"),
+        )
+
+    @classmethod
+    def configuration_schema(cls):
+        return {
+            "type": "object",
+            "properties": {
+                "host": {"type": "string"},
+                "port": {"type": "number"},
+                "username": {"type": "string"},
+                "password": {"type": "string"},
+                "catalog": {"type": "string"},
+                "database": {"type": "string"},
+            },
+            "order": [
+                "host",
+                "port",
+                "username",
+                "password",
+                "catalog",
+                "database",
+            ],
+            "required": ["host", "port", "username", "password", "catalog", "database"],
+            "secret": ["password"],
+        }
+
+    @classmethod
+    def enabled(cls):
+        return enabled
+
+    @classmethod
+    def type(cls):
+        return "e6data"
+
+    def run_query(self, query, user):
+        cursor = None
+        try:
+            cursor = self.connection.cursor(catalog_name=self.configuration.get("catalog"))
+            cursor.execute(query)
+            results = cursor.fetchall()
+            description = cursor.description
+            columns = []
+            for c in description:
+                column_name, column_type = c[0], E6DATA_TYPES_MAPPING.get(c[1], None)
+                columns.append({"name": column_name, "type": column_type})
+            rows = [dict(zip([c["name"] for c in columns], r)) for r in results]
+            data = {"columns": columns, "rows": rows}
+            json_data = json_dumps(data)
+            error = None
+
+        except Exception as error:
+            logger.debug(error)
+            json_data = None
+        finally:
+            if cursor is not None:
+                cursor.clear()
+                cursor.close()
+
+        return json_data, error
+
+    def test_connection(self):
+        self.noop_query = "SELECT 1"
+
+        data, error = self.run_query(self.noop_query, None)
+
+        if error is not None:
+            raise Exception(error)
+
+    def get_schema(self, get_stats=False):
+        tables = self.connection.get_tables(self.configuration.get("catalog"), self.configuration.get("database"))
+
+        schema = list()
+
+        for table_name in tables:
+            columns = self.connection.get_columns(
+                self.configuration.get("catalog"),
+                self.configuration.get("database"),
+                table_name,
+            )
+            columns_with_type = []
+
+            for column in columns:
+                redash_type = E6DATA_TYPES_MAPPING.get(column["fieldType"], None)
+                columns_with_type.append({"name": column["fieldName"], "type": redash_type})
+
+            table_schema = {"name": table_name, "columns": columns_with_type}
+
+            schema.append(table_schema)
+
+        return schema
+
+
+register(e6data)

redash/settings/__init__.py

@@ -337,6 +337,7 @@ def email_server_is_configured():
     "redash.query_runner.google_search_console",
     "redash.query_runner.ignite",
     "redash.query_runner.oracle",
+    "redash.query_runner.e6data",
 ]
 
 enabled_query_runners = array_from_string(