vortex-cli 4.14.1__tar.gz → 4.15.0__tar.gz

{vortex_cli-4.14.1 → vortex_cli-4.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vortex_cli
-Version: 4.14.1
+Version: 4.15.0
 Home-page: https://github.com/jordanamos/vortex-cli
 Author: Jordan Amos
 Author-email: jordan.amos@gmail.com

{vortex_cli-4.14.1 → vortex_cli-4.15.0}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = vortex_cli
-version = 4.14.1
+version = 4.15.0
 long_description = file: README.md
 long_description_content_type = text/markdown
 url = https://github.com/jordanamos/vortex-cli

{vortex_cli-4.14.1 → vortex_cli-4.15.0}/vortex/cli.py

@@ -536,9 +536,13 @@ def add_db_parser(command_parser: _SubParsersAction[ArgumentParser]) -> Argument
     db_mutex = db_parser.add_mutually_exclusive_group(required=True)
     db_mutex.add_argument(
         "--sql",
+        type=str,
+        nargs="?",
+        const="",
         help=(
-            "Execute a given SQL query. "
-            "Requires <database> to be a database connection id (int)."
+            "The SQL query to execute. "
+            "Requires <database> to be a database connection id (int). "
+            "Interactively prompts for the sql if the value is empty (Non Windows)."
         ),
     )
     db_mutex.add_argument(

{vortex_cli-4.14.1 → vortex_cli-4.15.0}/vortex/commands/db.py

@@ -1,6 +1,10 @@
 from __future__ import annotations
 
 import logging
+import os
+import subprocess
+import sys
+import tempfile
 from typing import Any
 
 import tabulate
@@ -9,7 +13,6 @@ from vortex.models import DatabaseConnection
 from vortex.models import PuakmaServer
 from vortex.workspace import Workspace
 
-
 logger = logging.getLogger("vortex")
 
 GET_SCHEMA_SQL = """\
@@ -90,6 +93,8 @@ def _output_result(
     list_tables: bool = False,
 ) -> int:
     with server:
+        logger.debug("Executing query")
+        logger.debug(sql)
         result = tuple(server.database_designer.execute_query(conn_id, sql, update))
         if schema_table and not result:
             logger.error(f"Table '{schema_table}' not found in database '{db_name}'")
@@ -123,6 +128,21 @@ def _is_connection_cloned(workspace: Workspace, conn_id: int) -> bool:
     return any(conn.id == conn_id for conn in conns)
 
 
+def _read_from_editor() -> str:
+    fd, tmp_sql_file = tempfile.mkstemp(suffix=".sql")
+    os.close(fd)
+    try:
+        subprocess.call(["nano", tmp_sql_file])
+        with open(tmp_sql_file) as f:
+            sql_query = f.read().strip()
+    except Exception:
+        os.remove(tmp_sql_file)
+        raise
+    else:
+        os.remove(tmp_sql_file)
+        return sql_query
+
+
 def db(
     workspace: Workspace,
     server: PuakmaServer,
@@ -134,7 +154,12 @@ def db(
     list_tables: bool = False,
     truncate_cols: bool = False,
 ) -> int:
-    if sql:
+    if sql is not None:
+        if sql.strip() == "":
+            if sys.platform == "win32":
+                sql = None
+            else:
+                sql = _read_from_editor()
         db_name = None
         conn_id = int(database)
         # If we're modifying a database, lets check that its locally
@@ -142,7 +167,7 @@ def db(
         if update and not _is_connection_cloned(workspace, conn_id):
             logger.error(f"No cloned applications with DB Connection ID '{conn_id}'")
             return 1
-        if not update:
+        if sql and not update:
             sql = sql + f" LIMIT {limit_n_results}"
     else:
         db_name = str(database).lower()
@@ -154,8 +179,9 @@ def db(
         elif list_tables:
             sql = LIST_TABLES_SQL % db_name
 
-    if sql is None:
-        raise ValueError("No Query to execute.")
+    if not sql:
+        logger.error("No Query to execute.")
+        return 1
 
     return _output_result(
         server, conn_id, sql, update, truncate_cols, db_name, schema_table, list_tables

{vortex_cli-4.14.1 → vortex_cli-4.15.0}/vortex_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vortex_cli
-Version: 4.14.1
+Version: 4.15.0
 Home-page: https://github.com/jordanamos/vortex-cli
 Author: Jordan Amos
 Author-email: jordan.amos@gmail.com