{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "92475c69",
   "metadata": {},
   "source": [
    "# Python API Example - Gas Transit Costs\n",
    "\n",
    "This guide is designed to provide an example of how to access the Spark API:\n",
    "- The path to your client credentials is the only input needed to run this script (just before Section 2)\n",
    "- This script has been designed to display the raw outputs of requests from the API, and then shows you how to format those outputs to enable easy reading and analysis\n",
    "- This script can be copied and pasted by customers for quick use of the API\n",
    "\n",
    "__N.B. This guide is just for Gas Transit data. If you're looking for other API data products (such as contract prices, Freight routes or Netbacks), please refer to their according code example files.__ "
   ]
  },
  {
   "cell_type": "markdown",
   "id": "fc7443de",
   "metadata": {},
   "source": [
    "### Have any questions?\n",
    "\n",
    "If you have any questions regarding our API, or need help accessing specific datasets, please contact us at:\n",
    "\n",
    "__data@sparkcommodities.com__\n",
    "\n",
    "or refer to our API website for more information about this endpoint: https://www.sparkcommodities.com/api/request/access.html"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "c5716130",
   "metadata": {},
   "source": [
    "## 1. Importing Data\n",
    "\n",
    "Here we define the functions that allow us to retrieve the valid credentials to access the Spark API.\n",
    "\n",
    "This section can remain unchanged for most Spark API users."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "552e6353",
   "metadata": {},
   "outputs": [],
   "source": [
    "import json\n",
    "import os\n",
    "import sys\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "from base64 import b64encode\n",
    "from urllib.parse import urljoin\n",
    "from io import StringIO\n",
    "import datetime\n",
    "\n",
    "try:\n",
    "    from urllib import request, parse\n",
    "    from urllib.error import HTTPError\n",
    "except ImportError:\n",
    "    raise RuntimeError(\"Python 3 required\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fe759439",
   "metadata": {},
   "outputs": [],
   "source": [
    "API_BASE_URL = \"https://api.sparkcommodities.com\"\n",
    "\n",
    "\n",
    "def retrieve_credentials(file_path=None):\n",
    "    \"\"\"\n",
    "    Find credentials either by reading the client_credentials file or reading\n",
    "    environment variables\n",
    "    \"\"\"\n",
    "    if file_path is None:\n",
    "        client_id = os.getenv(\"SPARK_CLIENT_ID\")\n",
    "        client_secret = os.getenv(\"SPARK_CLIENT_SECRET\")\n",
    "        if not client_id or not client_secret:\n",
    "            raise RuntimeError(\n",
    "                \"SPARK_CLIENT_ID and SPARK_CLIENT_SECRET environment vars required\"\n",
    "            )\n",
    "    else:\n",
    "        # Parse the file\n",
    "        if not os.path.isfile(file_path):\n",
    "            raise RuntimeError(\"The file {} doesn't exist\".format(file_path))\n",
    "\n",
    "        with open(file_path) as fp:\n",
    "            lines = [l.replace(\"\\n\", \"\") for l in fp.readlines()]\n",
    "\n",
    "        if lines[0] in (\"clientId,clientSecret\", \"client_id,client_secret\"):\n",
    "            client_id, client_secret = lines[1].split(\",\")\n",
    "        else:\n",
    "            print(\"First line read: '{}'\".format(lines[0]))\n",
    "            raise RuntimeError(\n",
    "                \"The specified file {} doesn't look like to be a Spark API client \"\n",
    "                \"credentials file\".format(file_path)\n",
    "            )\n",
    "\n",
    "    print(\">>>> Found credentials!\")\n",
    "    \n",
    "    return client_id, client_secret\n",
    "\n",
    "\n",
    "def do_api_post_query(uri, body, headers):\n",
    "    \"\"\"\n",
    "    OAuth2 authentication requires a POST request with client credentials before accessing the API.\n",
    "    This POST request will return an Access Token which will be used for the API GET request.\n",
    "    \"\"\"\n",
    "    url = urljoin(API_BASE_URL, uri)\n",
    "\n",
    "    data = json.dumps(body).encode(\"utf-8\")\n",
    "\n",
    "    # HTTP POST request\n",
    "    req = request.Request(url, data=data, headers=headers)\n",
    "    try:\n",
    "        response = request.urlopen(req)\n",
    "    except HTTPError as e:\n",
    "        print(\"HTTP Error: \", e.code)\n",
    "        print(e.read())\n",
    "        sys.exit(1)\n",
    "\n",
    "    resp_content = response.read()\n",
    "\n",
    "    # The server must return HTTP 201. Raise an error if this is not the case\n",
    "    assert response.status == 201, resp_content\n",
    "\n",
    "    # The server returned a JSON response\n",
    "    content = json.loads(resp_content)\n",
    "\n",
    "    return content\n",
    "\n",
    "\n",
    "def do_api_get_query(uri, access_token, format='json'):\n",
    "    \"\"\"\n",
    "    After receiving an Access Token, we can request information from the API.\n",
    "    Supports both JSON (default) and CSV responses.\n",
    "    \"\"\"\n",
    "    url = urljoin(API_BASE_URL, uri)\n",
    "\n",
    "    if format == 'json':\n",
    "        headers = {\n",
    "            \"Authorization\": \"Bearer {}\".format(access_token),\n",
    "            \"Accept\": \"application/json\",\n",
    "        }\n",
    "    elif format == 'csv':\n",
    "        headers = {\n",
    "            \"Authorization\": \"Bearer {}\".format(access_token),\n",
    "            \"Accept\": \"text/csv\",\n",
    "        }\n",
    "    else:\n",
    "        raise ValueError(\"The format parameter only takes 'csv' or 'json' as inputs\")\n",
    "\n",
    "    print(\"Fetching {}\".format(url))\n",
    "\n",
    "    # HTTP GET request\n",
    "    req = request.Request(url, headers=headers)\n",
    "    try:\n",
    "        response = request.urlopen(req)\n",
    "    except HTTPError as e:\n",
    "        print(\"HTTP Error: \", e.code)\n",
    "        print(e.read())\n",
    "        sys.exit(1)\n",
    "\n",
    "    resp_content = response.read()\n",
    "\n",
    "    # The server must return HTTP 200. Raise an error if this is not the case\n",
    "    assert response.status == 200, resp_content\n",
    "\n",
    "    # Storing response based on requested format\n",
    "    if format == 'json':\n",
    "        content = json.loads(resp_content)\n",
    "    elif format == 'csv':\n",
    "        content = resp_content\n",
    "\n",
    "    return content\n",
    "\n",
    "\n",
    "def get_access_token(client_id, client_secret):\n",
    "\n",
    "    payload = \"{}:{}\".format(client_id, client_secret).encode()\n",
    "    headers = {\n",
    "        \"Authorization\": b64encode(payload).decode(),\n",
    "        \"Accept\": \"application/json\",\n",
    "        \"Content-Type\": \"application/json\",\n",
    "    }\n",
    "    body = {\n",
    "        \"grantType\": \"clientCredentials\",\n",
    "    }\n",
    "\n",
    "    content = do_api_post_query(uri=\"/oauth/token/\", body=body, headers=headers)\n",
    "\n",
    "    print(\n",
    "        \">>>> Successfully fetched an access token {}****, valid {} seconds.\".format(\n",
    "            content[\"accessToken\"][:5], content[\"expiresIn\"]\n",
    "        )\n",
    "    )\n",
    "\n",
    "    return content[\"accessToken\"]"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "a51e1ef0",
   "metadata": {},
   "source": [
    "## N.B. Credentials\n",
    "\n",
    "Here we call the above functions, and input the file path to our credentials.\n",
    "\n",
    "N.B. You must have downloaded your client credentials CSV file before proceeding. Please refer to the API documentation if you have not dowloaded them already.  Instructions for downloading your credentials can be found here:\n",
    "\n",
    "https://www.sparkcommodities.com/api/introduction.html\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0817250f",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Insert file path to your client credentials here\n",
    "client_id, client_secret = retrieve_credentials(file_path=\"/tmp/client_credentials.csv\")\n",
    "\n",
    "# Authenticate:\n",
    "access_token = get_access_token(client_id, client_secret)\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "bf7d9183",
   "metadata": {},
   "source": [
    "# 2. Reference Data\n",
    "\n",
    "Here we fetch the relevant reference data for the endpoint - this is a list of all available transit route options available on the endpoint. Countries for each hub are also included for easy reference."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "71c883c6",
   "metadata": {},
   "outputs": [],
   "source": [
    "from io import StringIO\n",
    "\n",
    "## Defining the function\n",
    "\n",
    "def fetch_ref_data(access_token, format='json'):\n",
    "\n",
    "    uri=\"v1.0/gas/transit/reference-data/\"\n",
    "    print(uri)\n",
    "    \n",
    "    content = do_api_get_query(\n",
    "        uri=uri, access_token=access_token, format=format\n",
    "    )\n",
    "    \n",
    "    if format == 'json':\n",
    "        my_dict = content\n",
    "    elif format == 'csv':\n",
    "        # if there's no data to show, returns raw response (empty string) and \"No Data to Show\" message\n",
    "        if len(content) == 0:\n",
    "            my_dict = content\n",
    "            print('No Data to Show')\n",
    "        else:\n",
    "            my_dict = content.decode('utf-8')\n",
    "            my_dict = pd.read_csv(StringIO(my_dict))  # automatically converting into a Pandas DataFrame when choosing CSV format\n",
    "    \n",
    "    return my_dict\n",
    "\n",
    "ref_df = fetch_ref_data(access_token, format='csv')\n",
    "ref_df"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "e1feef7f",
   "metadata": {},
   "source": [
    "# 3. Calling Transit Costs data\n",
    "\n",
    "Here we define a function to fetch the Gas Transit costs dataset. The available parameters can be found on our API docs:\n",
    "\n",
    "https://www.sparkcommodities.com/api/gas/transit.html\n",
    "\n",
    "__N.B.:__ This endpoint provides the option to return a JSON or CSV formatted response. Metadata is only available in the JSON format."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "438529d1",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "def fetch_transit_prices(access_token, unit, origin=None, destination=None, start=None, end=None, limit=None, offset=None, \n",
    "                        booking_type=None, product_type=None, format='json'):\n",
    "    \n",
    "    query_params = \"?unit={}\".format(unit)\n",
    "\n",
    "    if origin is not None:\n",
    "        query_params += \"&origin={}\".format(origin)\n",
    "\n",
    "    if destination is not None:\n",
    "        query_params += \"&destination={}\".format(destination)\n",
    "\n",
    "    if start is not None:\n",
    "        query_params += \"&start={}\".format(start)\n",
    "    if end is not None:\n",
    "        query_params += \"&end={}\".format(end)\n",
    "    \n",
    "    if limit is not None:\n",
    "        query_params += \"&limit={}\".format(limit)\n",
    "    if offset is not None:\n",
    "        query_params += \"&offset={}\".format(offset)\n",
    "\n",
    "    if booking_type is not None:    \n",
    "        query_params += \"&type={}\".format(booking_type)\n",
    "    \n",
    "    if product_type is not None:\n",
    "        query_params += \"&product={}\".format(product_type)\n",
    "\n",
    "    content = do_api_get_query(\n",
    "        uri=\"/v1.0/gas/transit/costs/{}\".format(query_params),\n",
    "        access_token=access_token, format=format\n",
    "    )\n",
    "\n",
    "    if format == 'json':\n",
    "        data = content\n",
    "    elif format == 'csv':\n",
    "        # if there's no data to show, returns raw response (empty string) and \"No Data to Show\" message\n",
    "        if len(content) == 0:\n",
    "            data = content\n",
    "            print('No Data to Show')\n",
    "        else:\n",
    "            data = content.decode('utf-8')\n",
    "            data = pd.read_csv(StringIO(data))  # automatically converting into a Pandas DataFrame when choosing CSV format\n",
    "\n",
    "    return data"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "5b169c31",
   "metadata": {},
   "source": [
    "#### JSON Response\n",
    "\n",
    "__N.B.__ The JSON response has a limited historical API call range of approximately 10 days (with no filter params applied) due to API dataset size limitations. Please use the CSV format to call up to 1 year's worth of historical data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "17b9e8b5",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Calling historical data - JSON response\n",
    "json_raw = fetch_transit_prices(access_token, unit='eur-per-mwh', start='2026-05-17', end='2026-05-27', format='json')\n",
    "json_raw"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c99df66f",
   "metadata": {},
   "outputs": [],
   "source": [
    "# checking metaData\n",
    "json_raw['metaData']"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "cc5e573c",
   "metadata": {},
   "source": [
    "#### CSV Response"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "acf84906",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Calling historical data - CSV response\n",
    "csv_df = fetch_transit_prices(access_token, unit='eur-per-mwh', start='2026-01-01', end='2026-05-27', format='csv')\n",
    "csv_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "aeb8f589",
   "metadata": {},
   "outputs": [],
   "source": [
    "# listing the column names of the DataFrame\n",
    "csv_df.columns"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "1ef02f48",
   "metadata": {},
   "source": [
    "## N.B. Historical Data Limits\n",
    "\n",
    "Currently, a maximum of 1 year's worth of historical data can be called at one time due to the size of the data file. \n",
    "\n",
    "If more data points are required, the below code can be used: the function calls the historical data 1 year at a time and combines the data into one Pandas DataFrame"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "9f0930ab",
   "metadata": {},
   "outputs": [],
   "source": [
    "import datetime\n",
    "\n",
    "def loop_historical_data(access_token, unit, start, end, origin=None, destination=None, booking_type=None, product_type=None):\n",
    "    \n",
    "    hist_diff = (datetime.datetime.strptime(end, '%Y-%m-%d') - datetime.datetime.strptime(start, '%Y-%m-%d')).days\n",
    "    t = 0\n",
    "\n",
    "    w = 365\n",
    "    print(hist_diff)\n",
    "    while t < hist_diff:\n",
    "        print(t)\n",
    "        # initialising dataframe\n",
    "        if t == 0 and hist_diff>w:\n",
    "            diff_end = datetime.datetime.strftime(datetime.datetime.strptime(start, '%Y-%m-%d') + pd.Timedelta(days=w), '%Y-%m-%d')\n",
    "            hist_df = fetch_transit_prices(access_token, unit=unit, start=start, end=diff_end, origin=origin, destination=destination,\n",
    "                        booking_type=booking_type, product_type=product_type, format='csv')\n",
    "        \n",
    "        elif t==0 and hist_diff<=w:\n",
    "            hist_df = fetch_transit_prices(access_token, unit=unit, start=start, end=end, origin=origin, destination=destination,\n",
    "                        booking_type=booking_type, product_type=product_type, format='csv')\n",
    "\n",
    "        # appending additional historical data\n",
    "        else:\n",
    "            if t < hist_diff-w:\n",
    "                diff_start = datetime.datetime.strftime(datetime.datetime.strptime(start, '%Y-%m-%d') + pd.Timedelta(days=t+1), '%Y-%m-%d')\n",
    "                diff_end = datetime.datetime.strftime(datetime.datetime.strptime(diff_start, '%Y-%m-%d') + pd.Timedelta(days=w), '%Y-%m-%d')\n",
    "                historical_addition = fetch_transit_prices(access_token, unit=unit, start=diff_start, end=diff_end, origin=origin, destination=destination,\n",
    "                        booking_type=booking_type, product_type=product_type, format='csv')\n",
    "                try:\n",
    "                    hist_df = pd.concat([hist_df,historical_addition])\n",
    "                #exception if hist_df is empty\n",
    "                except:\n",
    "                    hist_df = historical_addition.copy()\n",
    "\n",
    "            else:\n",
    "                diff_start = datetime.datetime.strftime(datetime.datetime.strptime(start, '%Y-%m-%d') + pd.Timedelta(days=t+1), '%Y-%m-%d')\n",
    "                diff_end = datetime.datetime.strftime(datetime.datetime.strptime(diff_start, '%Y-%m-%d') + pd.Timedelta(days=hist_diff-t), '%Y-%m-%d')\n",
    "                historical_addition = fetch_transit_prices(access_token, unit=unit, start=diff_start, end=diff_end, origin=origin, destination=destination,\n",
    "                        booking_type=booking_type, product_type=product_type, format='csv')\n",
    "                hist_df = pd.concat([hist_df, historical_addition])\n",
    "        \n",
    "        #looping by year\n",
    "        t += w\n",
    "\n",
    "    for c in list(hist_df.columns)[13:]:\n",
    "        hist_df[c] = pd.to_numeric(hist_df[c])\n",
    "    hist_df['ReleaseDate'] = pd.to_datetime(hist_df['ReleaseDate'])\n",
    "\n",
    "    return hist_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fca50929",
   "metadata": {},
   "outputs": [],
   "source": [
    "\"\"\"\n",
    "hist_df = loop_historical_data(access_token, unit='eur-per-mwh', start='2026-01-01', end='2026-05-30', \n",
    "                                origin=None, destination=None, booking_type=None, product_type=None)\n",
    "\n",
    "hist_df\n",
    "\"\"\""
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "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.13.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}