qodalis-cli-server-plugin-data-explorer-mysql 1.0.0b18__py3-none-any.whl

qodalis_cli_data_explorer_mysql/__init__.py

@@ -0,0 +1,5 @@
+"""Qodalis CLI Data Explorer MySQL plugin — MySQL provider."""
+
+from .mysql_provider import MysqlDataExplorerProvider
+
+__all__ = ["MysqlDataExplorerProvider"]

qodalis_cli_data_explorer_mysql/mysql_provider.py

@@ -0,0 +1,146 @@
+"""MySQL-backed Data Explorer provider."""
+
+from __future__ import annotations
+
+import time
+from urllib.parse import urlparse
+
+import aiomysql
+
+from qodalis_cli_server_abstractions import (
+    DataExplorerExecutionContext,
+    DataExplorerProviderOptions,
+    DataExplorerResult,
+    DataExplorerSchemaColumn,
+    DataExplorerSchemaResult,
+    DataExplorerSchemaTable,
+    IDataExplorerProvider,
+)
+
+
+class MysqlDataExplorerProvider(IDataExplorerProvider):
+    """Executes SQL queries against a MySQL database."""
+
+    def __init__(self, connection_string: str) -> None:
+        parsed = urlparse(connection_string)
+        self._host = parsed.hostname or "localhost"
+        self._port = parsed.port or 3306
+        self._user = parsed.username or "root"
+        self._password = parsed.password or ""
+        self._db = parsed.path.lstrip("/") if parsed.path else None
+
+    async def execute_async(
+        self, context: DataExplorerExecutionContext
+    ) -> DataExplorerResult:
+        """Execute a SQL query against the MySQL database."""
+        start = time.monotonic()
+        conn: aiomysql.Connection | None = None
+        try:
+            conn = await aiomysql.connect(
+                host=self._host,
+                port=self._port,
+                user=self._user,
+                password=self._password,
+                db=self._db,
+            )
+            async with conn.cursor() as cursor:
+                await cursor.execute(context.query, context.parameters or None)
+
+                if cursor.description:
+                    columns = [desc[0] for desc in cursor.description]
+                    rows_raw = await cursor.fetchall()
+                    rows = [list(row) for row in rows_raw]
+                    return DataExplorerResult(
+                        success=True,
+                        source=context.options.name,
+                        language=context.options.language,
+                        default_output_format=context.options.default_output_format,
+                        execution_time=int((time.monotonic() - start) * 1000),
+                        columns=columns,
+                        rows=rows,
+                        row_count=len(rows),
+                        truncated=False,
+                        error=None,
+                    )
+                else:
+                    await conn.commit()
+                    return DataExplorerResult(
+                        success=True,
+                        source=context.options.name,
+                        language=context.options.language,
+                        default_output_format=context.options.default_output_format,
+                        execution_time=int((time.monotonic() - start) * 1000),
+                        columns=["changes"],
+                        rows=[[cursor.rowcount]],
+                        row_count=1,
+                        truncated=False,
+                        error=None,
+                    )
+        except Exception as exc:
+            return DataExplorerResult(
+                success=False,
+                source=context.options.name,
+                language=context.options.language,
+                default_output_format=context.options.default_output_format,
+                execution_time=int((time.monotonic() - start) * 1000),
+                columns=None,
+                rows=[],
+                row_count=0,
+                truncated=False,
+                error=str(exc),
+            )
+        finally:
+            if conn is not None:
+                conn.close()
+
+    async def get_schema_async(
+        self, options: DataExplorerProviderOptions
+    ) -> DataExplorerSchemaResult | None:
+        """Return the database schema (tables, views, and their columns)."""
+        conn: aiomysql.Connection | None = None
+        try:
+            conn = await aiomysql.connect(
+                host=self._host,
+                port=self._port,
+                user=self._user,
+                password=self._password,
+                db=self._db,
+            )
+            async with conn.cursor() as cursor:
+                await cursor.execute(
+                    "SELECT TABLE_NAME, TABLE_TYPE "
+                    "FROM information_schema.tables "
+                    "WHERE TABLE_SCHEMA = DATABASE() "
+                    "ORDER BY TABLE_NAME"
+                )
+                table_rows = await cursor.fetchall()
+
+                tables: list[DataExplorerSchemaTable] = []
+                for table_name, table_type in table_rows:
+                    await cursor.execute(
+                        "SELECT COLUMN_NAME, DATA_TYPE, IS_NULLABLE, COLUMN_KEY "
+                        "FROM information_schema.columns "
+                        "WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = %s "
+                        "ORDER BY ORDINAL_POSITION",
+                        (table_name,),
+                    )
+                    col_rows = await cursor.fetchall()
+                    columns = [
+                        DataExplorerSchemaColumn(
+                            name=col_name,
+                            type=data_type or "TEXT",
+                            nullable=is_nullable == "YES",
+                            primary_key=col_key == "PRI",
+                        )
+                        for col_name, data_type, is_nullable, col_key in col_rows
+                    ]
+                    tables.append(DataExplorerSchemaTable(
+                        name=table_name,
+                        type=table_type,
+                        columns=columns,
+                    ))
+
+                return DataExplorerSchemaResult(source=options.name, tables=tables)
+        finally:
+            if conn is not None:
+                conn.close()

qodalis_cli_server_plugin_data_explorer_mysql-1.0.0b18.dist-info/METADATA

@@ -0,0 +1,21 @@
+Metadata-Version: 2.5
+Name: qodalis-cli-server-plugin-data-explorer-mysql
+Version: 1.0.0b18
+Summary: MySQL Data Explorer provider plugin for Qodalis CLI servers
+Project-URL: Homepage, https://github.com/qodalis-solutions/cli-server-python
+Project-URL: Repository, https://github.com/qodalis-solutions/cli-server-python
+Project-URL: Issues, https://github.com/qodalis-solutions/cli-server-python/issues
+Author-email: Nicolae Lupei <nicolae.lupei@qodalis.com>
+License-Expression: MIT
+Keywords: cli,data-explorer,mysql,plugin,qodalis,sql
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.10
+Requires-Dist: aiomysql>=0.2.0
+Requires-Dist: qodalis-cli-server-abstractions

qodalis_cli_server_plugin_data_explorer_mysql-1.0.0b18.dist-info/RECORD

@@ -0,0 +1,5 @@
+qodalis_cli_data_explorer_mysql/__init__.py,sha256=xNDWVqaUpfSljB5p1PCkviBuqYWk8gcWPToIQYb8NMM,161
+qodalis_cli_data_explorer_mysql/mysql_provider.py,sha256=3XeX0HxL3LFGTG5WV7Dbrh6WH2biPcawF9Hn9M7EZio,5736
+qodalis_cli_server_plugin_data_explorer_mysql-1.0.0b18.dist-info/METADATA,sha256=B-RXCqn_IvB0_e4V-FZ7RkRR73eKR02kw7KNj3uYktQ,1027
+qodalis_cli_server_plugin_data_explorer_mysql-1.0.0b18.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+qodalis_cli_server_plugin_data_explorer_mysql-1.0.0b18.dist-info/RECORD,,

qodalis_cli_server_plugin_data_explorer_mysql-1.0.0b18.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.33.0
+Root-Is-Purelib: true
+Tag: py3-none-any