Merge pull request #241 from vertica/qprof_ui_update_01

QueryProfiler UI Update
vertica · Apr 9, 2024 · 5bde0bc · 5bde0bc
2 parents c8fb0da + 1174b05
commit 5bde0bc
Show file tree

Hide file tree

Showing 3 changed files with 127 additions and 136 deletions.
diff --git a/project/ui/conn.ipynb b/project/ui/conn.ipynb
@@ -254,8 +254,8 @@
     "from IPython.display import display, HTML\n",
     "\n",
     "# Define the URLs for the links\n",
-    "url1 = \"http://127.0.0.1:8889/lab/workspaces/auto-3\"\n",
-    "url2 = \"http://127.0.0.1:8889/voila/render/ui/qprof_main.ipynb\"\n",
+    "url1 = \"/lab/\"\n",
+    "url2 = \"/voila/render/ui/qprof_main.ipynb\"\n",
     "\n",
     "# Define the button labels\n",
     "button_label1 = \"Go to Home Page\"\n",

diff --git a/project/ui/qprof-ui.ipynb b/project/ui/qprof-ui.ipynb
@@ -22,6 +22,9 @@
     "    %store -r transaction_val\n",
     "    %store -r statement_val\n",
     "    %store -r file_val\n",
+    "    %store -r target_schema_val_2\n",
+    "    %store -r key_val_2\n",
+    "    %store -r transaction_statement_list_val\n",
     "\n",
     "    if \"file_val\" in globals():\n",
     "        qprof = QueryProfiler.import_profile(target_schema=target_schema_val,\n",
@@ -31,7 +34,7 @@
     "        )\n",
     "        qprof = QueryProfilerInterface(\n",
     "            target_schema = target_schema_val,\n",
-    "            key_id = key_val,\n",
+    "            key_id = key_val if \"key_val\" in globals() else None,\n",
     "        )\n",
     "    elif \"target_schema_val\" in globals():\n",
     "        qprof = QueryProfilerInterface(\n",
@@ -40,12 +43,23 @@
     "        )\n",
     "    elif \"query_val\" in globals():\n",
     "        qprof = QueryProfilerInterface(\n",
-    "            query_val\n",
+    "            query_val,\n",
+    "            target_schema = target_schema_val_2 if \"target_schema_val_2\" in globals() else None,\n",
+    "            key_id = key_val_2 if \"key_val_2\" in globals() else None,       \n",
     "        )\n",
     "    elif \"transaction_val\" in globals():\n",
     "        qprof = QueryProfilerInterface(\n",
-    "            transactions = (transaction_val, statement_val)\n",
-    "        )"
+    "            transactions = (transaction_val, statement_val),\n",
+    "            target_schema = target_schema_val_2 if \"target_schema_val_2\" in globals() else None,\n",
+    "            key_id = key_val_2 if \"key_val_2\" in globals() else None,\n",
+    "        )\n",
+    "\n",
+    "    elif \"transaction_statement_list_val\" in globals():\n",
+    "        qprof = QueryProfilerInterface(\n",
+    "            transactions = transaction_statement_list_val,\n",
+    "            target_schema = target_schema_val_2 if \"target_schema_val_2\" in globals() else None,\n",
+    "            key_id = key_val_2 if \"key_val_2\" in globals() else None,\n",
+    "        )\n"
    ]
   },
   {
@@ -121,7 +135,7 @@
    "outputs": [],
    "source": [
     "# Define the URLs for the links\n",
-    "url1 = \"http://127.0.0.1:8889/voila/render/ui/qprof_main.ipynb\"\n",
+    "url1 = \"/voila/render/ui/qprof_main.ipynb\"\n",
     "# Define the button labels\n",
     "button_label1 = \"Go Back\"\n",
     "# Define HTML links for button\n",