{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "e13084db-c69f-4e70-aa79-acfd2159093a",
   "metadata": {},
   "source": [
    "# Get Cluster Details"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "a44654e8-d728-4fb4-a6c5-a15325f7c440",
   "metadata": {},
   "outputs": [],
   "source": [
    "CLUSTER_NAME=\"hdb-cluster-welcomedb\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "3698017d-42ed-4280-afa1-723f19ae398c",
   "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"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "ed241efe-523e-4d2f-8e1b-94a09adf42e7",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Using variables ...\n"
     ]
    }
   ],
   "source": [
    "# triggers credential get\n",
    "session=None\n",
    "\n",
    "try:\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": "markdown",
   "id": "6b7294c7-d9a3-4fb2-b671-12ead3e17e83",
   "metadata": {},
   "source": [
    "# Cluster Details"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "ccb6b824-4af8-447a-8b0a-ecd4d8136081",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'status': 'RUNNING',\n",
       " 'clusterName': 'hdb-cluster-welcomedb',\n",
       " 'clusterType': 'HDB',\n",
       " 'databases': [{'databaseName': 'welcomedb',\n",
       "   'cacheConfigurations': [{'cacheType': 'CACHE_1000', 'dbPaths': ['/']}],\n",
       "   'changesetId': 'isRIOfpd2lyrudnX5paQQA'}],\n",
       " 'cacheStorageConfigurations': [{'type': 'CACHE_1000', 'size': 1200}],\n",
       " 'capacityConfiguration': {'nodeType': 'kx.s.xlarge', 'nodeCount': 2},\n",
       " 'releaseLabel': '1.0',\n",
       " 'vpcConfiguration': {'vpcId': 'vpc-0e702dec545865b11',\n",
       "  'securityGroupIds': ['sg-018111774e795682d'],\n",
       "  'subnetIds': ['subnet-0f97cae6600859c17'],\n",
       "  'ipAddressType': 'IP_V4'},\n",
       " 'initializationScript': 'code/init.q',\n",
       " 'commandLineArguments': [{'key': 's', 'value': '4'},\n",
       "  {'key': 'dbname', 'value': 'welcomedb'},\n",
       "  {'key': 'codebase', 'value': 'code'}],\n",
       " 'code': {'s3Bucket': 'kdb-demo-612841383594-kms', 's3Key': 'code/code.zip'},\n",
       " 'lastModifiedTimestamp': datetime.datetime(2023, 6, 6, 11, 44, 19, 914000, tzinfo=tzlocal()),\n",
       " 'azMode': 'SINGLE',\n",
       " 'availabilityZoneId': 'use1-az4',\n",
       " 'createdTimestamp': datetime.datetime(2023, 6, 6, 11, 26, 51, 517000, tzinfo=tzlocal())}"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "Last Run: 2023-06-06 13:27:42.605223\n"
     ]
    }
   ],
   "source": [
    "try:\n",
    "    resp=client.get_kx_cluster(environmentId=ENV_ID, clusterName=CLUSTER_NAME)\n",
    "    resp.pop('ResponseMetadata')\n",
    "except client.exceptions.ResourceNotFoundException:\n",
    "    resp=f\"Cluster: {CLUSTER_NAME} Not Found\"\n",
    "\n",
    "display(resp)\n",
    "\n",
    "print()\n",
    "print( f\"Last Run: {datetime.datetime.now()}\" )"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "b6f0fcc9-e777-4506-8378-7f0daef05fe8",
   "metadata": {},
   "source": [
    "# Connection String for Cluster"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "d5f21dfa-727b-42eb-8ca4-d8495894dad6",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Connection String:\n",
      ":tcps://vpce-0aed0eaa4e0ae7d31-fg77afb9.vpce-svc-039dfe1c1935d800c.us-east-1.vpce.amazonaws.com:443:bob:Host=vpce-0aed0eaa4e0ae7d31-fg77afb9.vpce-svc-039dfe1c1935d800c.us-east-1.vpce.amazonaws.com&Port=5000&User=bob&Action=finspace%3AConnectKxCluster&X-Amz-Security-Token=IQoJb3JpZ2luX2VjEP7%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FwEaCXVzLWVhc3QtMSJHMEUCIQC%2BHZzJsIXfIDi2DaDTyOslyZQjQtcHExfT%2B5SNEC7aPwIgUdDOKjw4y%2FSGIXkf1Rd1Wzm%2BeN0VihxJA%2BN83do%2FviAq%2BAIIRxAAGgw2MTI4NDEzODM1OTQiDI8lIuEkQdjIRMhqgirVAi9H3rWIUsxFlC0xnhYp5EM97hRJIZuCWBURveSQFt8S48dUsIELJNvwrkWYLWI1wlDrYpAvyXQAoVs2bZO8L82%2BMoU4O2o%2FJMgs%2Fs1vuzhHU1VFCUon6C2BgXOgWJGiFtsWIGhrGDyKqQpxnTP1ZycgnrhhpkVKj%2FZPWNqwjVqgTlLh6LRtTCRZ6nVDRdqg9g5K6%2FJBKVtCnh%2BiHV%2BqQEgjLbNj1nwFattGm2ipcB5vXwajX9E3Ez5vevvbs3Ch7tmHXzVgdOd4JDUeTo%2BA4h4Q17EIyBjpe%2Bum94dgQAOFzAcYY57%2Fd4aDqehc2Ah5HsBEmi2GBritQPnHyZWFLZCZVs3%2Bgi9hAo%2Fy1mTxZljRFeeVKOv8QXotDgJTPOnPKE08kBNmlquUcIIjg4U8IWxsc2Bc%2BKqGLnRbhV81WCZXg5zob2wCZxFlVWLU7EPQIhL5p64sMNDo%2FKMGOr8B5%2FGqSSsHfe20nYgMIpzPq8gurOtolBEdmrUnQ7336b7kP9VlR%2Bkia%2B%2FCKrbdGQJEWPZxLTYY8MJuqrFH546LA6wBxf4TXHsSM3xQ7LPZ2ejKit5fxFgRoZfzx4MgMrza9Q9gd6cBG9D84rHNOUjA16BG%2F3SIFgRag8vZo%2B1f8tFBIkzffZHqztxDgxX%2B2MCJFXnrGwYsj4E%2BUVUMSyfJd7gY46Je2Pp%2FNqMqsAHoJ0JHn6cfgRmEiSTYyxHjlSg%3D&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Date=20230606T132744Z&X-Amz-SignedHeaders=host&X-Amz-Expires=900&X-Amz-Credential=ASIAY5MBRM2VDDZCWGAP%2F20230606%2Fus-east-1%2Ffinspace-apricot%2Faws4_request&X-Amz-Signature=8a5bb6ecbffacfd93462d43ad762ca7565f32afae643b3a737eaaa62eb75afa8\n"
     ]
    }
   ],
   "source": [
    "conn_str = get_kx_connection_string(client, environmentId=ENV_ID, clusterName=CLUSTER_NAME, userName=KDB_USERNAME, boto_session=session)\n",
    "print(f\"Connection String:\")\n",
    "print(conn_str)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "876590d3-743d-4dda-aeab-074015f12f0e",
   "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
}