{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "28bea13b-67bd-4a0e-8eab-3b8ffd37259e",
   "metadata": {},
   "source": [
    "# BasicTick: Connection Strings\n",
    "Get the Connection strings to basictick HDB, RDB, and Gateway processes"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "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_kdb_1 import *\n",
    "\n",
    "clusters = {\n",
    "    \"rdb_conn\": \"RDB_basictickdb_TFC\",\n",
    "    \"hdb_conn\": \"HDB_basictickdb_TFC\",\n",
    "    \"gw_conn\" : \"GATEWAY_TFC\",\n",
    "}\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3cfe7d89-9f5d-4ceb-ac8c-1f5054a6f15a",
   "metadata": {},
   "outputs": [],
   "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": null,
   "id": "1a9a896d-466e-4321-86c4-a54e76d7997a",
   "metadata": {},
   "outputs": [],
   "source": [
    "print(\"Paste below into q \"+\"=\"*80)\n",
    "\n",
    "for k, v in clusters.items():\n",
    "    try:\n",
    "        conn_str = get_kx_connection_string(client, environmentId=ENV_ID, clusterName=v, userName=KDB_USERNAME, boto_session=session)\n",
    "    except:\n",
    "        conn_str = \"\"\n",
    "        \n",
    "    print(f\"\"\"\n",
    "/ Cluster: {v}\n",
    "{k}:\"{conn_str}\" \"\"\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4775582c-0b50-4f62-af62-4936c29a4c20",
   "metadata": {},
   "outputs": [],
   "source": [
    "print(\"Paste below into SHELL \"+\"=\"*80)\n",
    "\n",
    "for k, v in clusters.items():\n",
    "    try:\n",
    "        conn_str = get_kx_connection_string(client, environmentId=ENV_ID, clusterName=v, userName=KDB_USERNAME, boto_session=session)\n",
    "    except:\n",
    "        conn_str = \"\"\n",
    "        \n",
    "    print(f\"\"\"\n",
    "# Cluster: {v}\n",
    "{k.upper()}=\"{conn_str}\" \"\"\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "86f33240-bb12-49f3-8d9c-5783c25eb182",
   "metadata": {},
   "outputs": [],
   "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
}