{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "28bea13b-67bd-4a0e-8eab-3b8ffd37259e",
   "metadata": {},
   "source": [
    "# TAQ: Connection Strings\n",
    "Get the Connection strings Clusters"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "0d5f1d4a-ed45-44e3-bf75-9bdb75fcddbb",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import boto3\n",
    "import json\n",
    "import datetime\n",
    "\n",
    "from managed_kx import *\n",
    "from env_2 import *\n",
    "\n",
    "CLUSTER_NAMES=[ \n",
    "#    \"cluster_welcomedb\", \n",
    "#    \"cluster_create_delete_db\",\n",
    "    \"RDB_welcomedb\",\n",
    "    \"HDB_welcomedb\",\n",
    "]\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "3cfe7d89-9f5d-4ceb-ac8c-1f5054a6f15a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Using variables ...\n"
     ]
    }
   ],
   "source": [
    "# triggers credential get\n",
    "session=None\n",
    "\n",
    "try:\n",
    "    # aws: use ada for credentials\n",
    "    subprocess.call([\"which\", \"ada\"])\n",
    "    os.system(f\"ada credentials update --account={ACCOUNT_ID} --provider=isengard --role=Admin --once\")\n",
    "except: \n",
    "    None\n",
    "\n",
    "if AWS_ACCESS_KEY_ID is None:\n",
    "    print(\"Using Defaults ...\")\n",
    "    # create AWS session: using access variables\n",
    "    session = boto3.Session()\n",
    "else:\n",
    "    print(\"Using variables ...\")\n",
    "    session = boto3.Session(\n",
    "        aws_access_key_id=AWS_ACCESS_KEY_ID,\n",
    "        aws_secret_access_key=AWS_SECRET_ACCESS_KEY,\n",
    "        aws_session_token=AWS_SESSION_TOKEN\n",
    "    )\n",
    "\n",
    "# create finspace client\n",
    "client = session.client(service_name='finspace', endpoint_url=ENDPOINT_URL)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "1d9b2d5a-19e9-44f9-8c38-ee8600453e87",
   "metadata": {},
   "outputs": [],
   "source": [
    "# get the connection strings\n",
    "conns = {}\n",
    "\n",
    "for c in CLUSTER_NAMES:\n",
    "    try:\n",
    "        conn_str = get_kx_connection_string(client, environmentId=ENV_ID, clusterName=c, userName=KDB_USERNAME, boto_session=session)\n",
    "        conns[c] = conn_str\n",
    "    except:\n",
    "        continue"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "1a9a896d-466e-4321-86c4-a54e76d7997a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Paste below into q ================================================================================\n",
      "\n",
      "/ Cluster: RDB_welcomedb\n",
      "hdb_conn:\":tcps://vpce-0a299a1f762f351b0-uiaa28cm.vpce-svc-061a7d8630e15dd1a.us-east-1.vpce.amazonaws.com:443:bob:Host=vpce-0a299a1f762f351b0-uiaa28cm.vpce-svc-061a7d8630e15dd1a.us-east-1.vpce.amazonaws.com&Port=5000&User=bob&Action=finspace%3AConnectKxCluster&X-Amz-Security-Token=IQoJb3JpZ2luX2VjEAAaCXVzLWVhc3QtMSJHMEUCIFsng7qckpYKerjid4SvfGE7H2aVheI3rTcTfgeTfAuYAiEAqILYzbtZ4yZzztL34JHz2klpVYJstJgd8L56y8MGvFkq9gIISRAAGgw2MTI4NDEzODM1OTQiDIPI1lwNvLgPVam9sCrTAl9WOpF83o5eXfBz3j%2Fukl6nQgZ4M%2Bw%2B%2FF%2BpzbJtRnknvzz9AC4KF0MLUHHJTKAAImJm%2BRkDiedNAu2q6%2F0mL4ZH95Q9DCYf35fkTjTRAQqLWXt0yB58ZneCOy5lHjGc2Zlkjdf7hDKCH%2BY7eIN29Db9zaGTz5Q7zzXFdorwtVY5BnMfiq%2FSNtyswEsWGn1zqMZLUsDVOSPHrG0dP8pQNhdDBodNeln3NYNT%2BfYzzdv8p22GMpHFP4PZw0ff%2FEO2ESie0T1DoLrxKUMLwjPxjjlfEd5sNNWcz5y7ZdiTqh3wveg62puWb3JTgd8uX5jbOW%2Bymo4tX%2BUjr3y8tXSpKXwobRsu60ZZAIm7UIXzYxtro6JdvO2qLQGxPcfwzPd9pajNny8ZqNeRLu%2FbzYEMDlPJK90J%2B%2FnCyyXjyW4fLo%2FCboNd5m%2FXX10ycz5lOwFY8ANJPzDWoP2jBjq%2FAekGWQRmxmgnZb5HN8fCwozjN3xjD4THeg86PrCMdoNkT4BJfFHUWasSLtEUO2ONZAx4gbaUlE7m80VSRSs01R0NlorImsm%2B3rwyf0I5hbSl4tY3PDTIT4xpf%2Fq3XT7iGZ%2B%2FqGmnaO%2F%2FxNETdZiYvJUCXG6w%2FnAxTZb33x9QE32ixtuC%2FwbNTccIYgmsGFsK5%2FrXuT%2FkKWJIPVd9FCddLT9aez%2BknrxmNF3sdFSkQYKA4Hi8Ss%2BT72OdAWSQoOpz&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Date=20230606T152718Z&X-Amz-SignedHeaders=host&X-Amz-Expires=900&X-Amz-Credential=ASIAY5MBRM2VBG2QPJ5U%2F20230606%2Fus-east-1%2Ffinspace-apricot%2Faws4_request&X-Amz-Signature=3feac70098d0efe6469463fdd25722fb72c478bd336c6b17251e475b2956d9ed\"\n",
      "\n",
      "\n",
      "/ Cluster: HDB_welcomedb\n",
      "hdb_conn:\":tcps://vpce-059110fcb074ef975-uyzzkvgm.vpce-svc-099e64025966b40d1.us-east-1.vpce.amazonaws.com:443:bob:Host=vpce-059110fcb074ef975-uyzzkvgm.vpce-svc-099e64025966b40d1.us-east-1.vpce.amazonaws.com&Port=5000&User=bob&Action=finspace%3AConnectKxCluster&X-Amz-Security-Token=IQoJb3JpZ2luX2VjEAAaCXVzLWVhc3QtMSJGMEQCIG6rFprwAbhCItQpvQ1vubSHPrszMu0xRsbBpXWgq8l5AiBPBm71NYAJA%2FjcyNWP3YWXov%2Fbgk%2BZKsY9zEFE%2BYI9Dyr3AghJEAAaDDYxMjg0MTM4MzU5NCIMt5VPtythEsjeIWGYKtQCdLXBQPsLmlQlrf7GLqbVtBh2D4O0fpzlIORQC2t8VfY1RqE%2BtUNvpNb%2BgVFXmYfFNijRjjRAEOOJJSE0TQLnbvCLkr3PdXD%2BxO7nNqM9Dti%2BaKTByjPB%2Bn11MblJlLMQUR1mRhmIza1eR4oRCdbt8kl2ctT%2BCdXP1t4WTj94yU7V0TWHrKyYFCnQ%2Bdmjejl5T2vGFa760l5SIQ0noYFJXpuA%2Fc%2BwEzUbhjU7mE6jb5lEyhBiGPVCcVVYJUDrkPdAqmUdLacAwK%2B5N8IgjE5XwOsBVntnkMESzqSxS%2BYuamFBT2GaPo5cKJdyU3aAlbeYKNJOF3OJzJ0vHIo4sA9N8i8%2FIBtQXpyHXpU5bfjDPSIGpx7dDRDzzsJrAy490h9APALriUftnITgw3Qubg3mgE9NPex4%2BFxxbZ393AcN%2BJnJ8zltME91xYLUqM%2FGP%2FZzEcSDITDWoP2jBjrAAZYbbs3N2CnEIqpn0KivICj%2BbrWJc52E2pcity0jV3eJtCXThDWv60A%2FXuc4wnMbEsDr2b2E1JPX89OlO1lY6I2gYT%2FcqYcqK1UzomgyY3nVgl9tTxqwWP8wBYvf%2BltwAX6DwWrHEfjW1S2zZ%2BxdJgptdgj1pYU5Om2dc%2B8taSIfgwhWHiyEXnyteFj63P8dEH0rT0dGAAugD%2FxP7VZvvfiMcOwd9h1n8xkeNUzhhbEB1WsXNmy4jI8qIuSVWgW5Mg%3D%3D&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Date=20230606T152718Z&X-Amz-SignedHeaders=host&X-Amz-Expires=900&X-Amz-Credential=ASIAY5MBRM2VG6Y2ED7L%2F20230606%2Fus-east-1%2Ffinspace-apricot%2Faws4_request&X-Amz-Signature=cfe9d04492f850878dc7a40a03513d6db7142174209b9172975798ffbfe8593c\"\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print(\"Paste below into q \"+\"=\"*80)\n",
    "\n",
    "for cluster, conn_str in conns.items():\n",
    "    print(f\"\"\"\n",
    "/ Cluster: {cluster}\n",
    "hdb_conn:\"{conn_str}\"\n",
    "\"\"\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "86f33240-bb12-49f3-8d9c-5783c25eb182",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Last Run: 2023-06-06 15:27:18.986132\n"
     ]
    }
   ],
   "source": [
    "print( f\"Last Run: {datetime.datetime.now()}\" )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d71daf7b-510b-4c0d-b285-360eb1c19533",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}