Accounts APIs - Add headers and conditions checks.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Enter intended project's APImetrics API key here:\n",
    "API_KEY = input(\"Enter your APImetrics API key for the intended project: \")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "from apimetrics_api import APImetricsAPI \n",
    "\n",
    "# An instance of the class that calls the APImetrics API\n",
    "CLIENT = APImetricsAPI(API_KEY)\n",
    "\n",
    "def find_header(header_key, call):\n",
    "    header_key = header_key.lower()\n",
    "    return next(\n",
    "        (header for header in call['request']['headers'] if header['key'].lower() == header_key),\n",
    "        None,\n",
    "    )\n",
    "\n",
    "def find_condition(source, variable_path, conditions):\n",
    "    return next(\n",
    "        (condition for condition in conditions if condition['source'] == source and condition['variable_path'].lower() == variable_path.lower()),\n",
    "        None\n",
    "    )\n",
    "\n",
    "HSBC = False\n",
    "RBS = True\n",
    "\n",
    "calls = CLIENT.get_calls()\n",
    "for call in calls['results']:\n",
    "    if \"api_expert:exclude\" not in call['meta']['tags']:\n",
    "        \n",
    "        print(f\"API: {call['meta']['name']}...\", end='')\n",
    "        needs_save = False\n",
    "        conditions = CLIENT.get_call_conditions(call['id'])\n",
    "        cond_array = conditions['conditions']\n",
    "        \n",
    "        if 'ob_v:3.1' in call['meta']['tags']:\n",
    "            has_header = find_header(\"x-fapi-interaction-id\", call)\n",
    "            if not has_header:\n",
    "                needs_save = True\n",
    "                request = call['request']\n",
    "                headers = request['headers'] or []\n",
    "                headers.append(dict(key='x-fapi-interaction-id', value='%%GUID%%'))\n",
    "                CLIENT.update_call(call['id'], dict(request=dict(headers=headers)))\n",
    "                print(' ! ', end='')\n",
    "            else:\n",
    "                print(' . ', end='')\n",
    "\n",
    "            has_cond = find_condition('HEADERS', 'x-fapi-interaction-id', cond_array)\n",
    "            if not has_cond:\n",
    "                cond_array.append({\n",
    "                    'source': 'HEADERS',\n",
    "                    'variable_path': 'x-fapi-interaction-id',\n",
    "                    'condition': 'EXISTS',\n",
    "                    'val': '',\n",
    "                    'test_result_on_true': None,\n",
    "                    'test_result_on_false': 'CONTENT_ERROR',\n",
    "                    'variable_name': 'X_FAPI_INTERACTION_ID',\n",
    "                })\n",
    "                needs_save = True\n",
    "                print(' O', end='')\n",
    "            else:\n",
    "                print(' o', end='')\n",
    "\n",
    "        if RBS:\n",
    "            has_cond = find_condition('HEADERS', 'X-Rbs-Interaction-Id', cond_array)\n",
    "            if not has_cond:\n",
    "                cond_array.append({\n",
    "                    'source': 'HEADERS',\n",
    "                    'variable_path': 'X-Rbs-Interaction-Id',\n",
    "                    'condition': 'EXISTS',\n",
    "                    'val': '',\n",
    "                    'test_result_on_true': None,\n",
    "                    'test_result_on_false': None,\n",
    "                    'variable_name': 'X_RBS_INTERACTION_ID',\n",
    "                })\n",
    "                needs_save = True\n",
    "                print(' O', end='')\n",
    "            else:\n",
    "                print(' o', end='')\n",
    "\n",
    "        if HSBC:\n",
    "            has_cond = find_condition('HEADERS', 'x-amzn-RequestId', cond_array)\n",
    "            if not has_cond:\n",
    "                cond_array.append({\n",
    "                    'source': 'HEADERS',\n",
    "                    'variable_path': 'x-amzn-RequestId',\n",
    "                    'condition': 'EXISTS',\n",
    "                    'val': '',\n",
    "                    'test_result_on_true': None,\n",
    "                    'test_result_on_false': None,\n",
    "                    'variable_name': 'X_AMZN_REQUEST_ID',\n",
    "                })\n",
    "                needs_save = True\n",
    "                print(' O', end='')\n",
    "            else:\n",
    "                print(' o', end='')\n",
    "\n",
    "\n",
    "            has_cond = find_condition('HEADERS', 'x-amz-apigw-id', cond_array)\n",
    "            if not has_cond:\n",
    "                cond_array.append({\n",
    "                    'source': 'HEADERS',\n",
    "                    'variable_path': 'x-amz-apigw-id',\n",
    "                    'condition': 'EXISTS',\n",
    "                    'val': '',\n",
    "                    'test_result_on_true': None,\n",
    "                    'test_result_on_false': None,\n",
    "                    'variable_name': 'X_AMZ_APIGW_ID',\n",
    "                })\n",
    "                needs_save = True\n",
    "                print(' O', end='')\n",
    "            else:\n",
    "                print(' o', end='')\n",
    "\n",
    "            has_cond = find_condition('HEADERS', 'X-Amzn-Trace-Id', cond_array)\n",
    "            if not has_cond:\n",
    "                cond_array.append({\n",
    "                    'source': 'HEADERS',\n",
    "                    'variable_path': 'X-Amzn-Trace-Id',\n",
    "                    'condition': 'EXISTS',\n",
    "                    'val': '',\n",
    "                    'test_result_on_true': None,\n",
    "                    'test_result_on_false': None,\n",
    "                    'variable_name': 'X_AMZN_TRACE_ID',\n",
    "                })\n",
    "                needs_save = True\n",
    "                print(' O', end='')\n",
    "            else:\n",
    "                print(' o', end='')\n",
    "\n",
    "        if needs_save:\n",
    "            CLIENT.set_call_conditions(call['id'], dict(conditions=cond_array))\n",
    "            print(' !')\n",
    "        else:\n",
    "            print(' .')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.8.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}