wms_getmap.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Imports\n",
    "import csv\n",
    "import json\n",
    "import pathlib\n",
    "import requests\n",
    "import time\n",
    "import pandas as pd\n",
    "import matplotlib.pyplot as plt\n",
    "from arcgis.gis import GIS"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Tests\n",
    "\n",
    "test = 'WMS Get Maps'\n",
    "timestamp = time.strftime(\"%Y%m%d-%H%M%S\")\n",
    "\n",
    "base_url = 'https://gis.earthdata.nasa.gov/image/rest/services/POWER/POWER_901_ANNUAL_METEOROLOGY_UTC/ImageServer'\n",
    "wms_url = base_url + '/WMSServer?'\n",
    "multi_url = base_url + '/multidimensionalInfo?f=pjson'\n",
    "\n",
    "multi_get = requests.get(multi_url)\n",
    "multi_content = multi_get.json()\n",
    "variables = multi_content['multidimensionalInfo']['variables']\n",
    "names = [var.get('name') for var in variables]\n",
    "\n",
    "bbox = [-90, -180, 90, 180]\n",
    "bbox_str = ','.join([str(x) for x in bbox])\n",
    "\n",
    "start_time = time.time()\n",
    "    \n",
    "'''\n",
    "    TODO:\n",
    "        - Figure out which of these can be:\n",
    "            - hardcoded\n",
    "            - derived\n",
    "            - function parameters\n",
    "        - Other parameters are optional\n",
    "'''\n",
    "for var in names:\n",
    "    # TODO: GEORGE can you check:\n",
    "        # my adaptation of get_multiInfo code to get the time slices from the variables\n",
    "        # my inclusion of first time extent into the optional parameters   \n",
    "    if var.get(\"dimensions\"):\n",
    "        # Search for the time dimension and extract the first time extent value\n",
    "        for dimension in var.get(\"dimensions\"):\n",
    "            if dimension.get(\"name\") == \"StdTime\" or dimension.get(\"field\") == \"StdTime\":\n",
    "                time_extent = dimension.get(\"extent\", [])\n",
    "                if time_extent:\n",
    "                    first_time_extent_value = time_extent[0]\n",
    "                    last_time_extent_value = time_extent[-1]\n",
    "                    break\n",
    "    else:\n",
    "        print(\"No dimensions found for the first variable.\")\n",
    "    \n",
    "    # Calculate the time range\n",
    "    time_range = None\n",
    "    if first_time_extent_value:\n",
    "        time_range = (first_time_extent_value, last_time_extent_value)\n",
    "    \n",
    "    params = {\n",
    "        'request': 'GetMap',\n",
    "        'service': 'WMS',\n",
    "        'version': '1.1.1', # Service version. Value is one of 1.0.0, 1.1.0, 1.1.1, 1.3.0\n",
    "        'layers': var, # Layers to display on map. Value is a comma-separated list of layer names.\n",
    "        'styles': '', # Styles in which layers are to be rendered.\n",
    "                      # Value is a comma-separated list of style names, or empty if default styling is required. \n",
    "                      # Style names may be empty in the list, to use default layer styling.\n",
    "        'srs': 'EPSG:4326', # Spatial Reference System for map output.\n",
    "                            # Value is in form EPSG:nnnn. crs is the parameter key used in WMS 1.3.0.\n",
    "        'bbox': bbox_str, # Bounding box for map extent. Value is minx,miny,maxx,maxy in units of the SRS.\n",
    "        'width': '780', # Width of map output, in pixels.\n",
    "        'height': '330', # Height of map output, in pixels.\n",
    "        'format': 'image/png' # Output format of map. Valid values are image/jpeg, image/png, and image/gif.\n",
    "    }\n",
    "       \n",
    "    opt_params = {\n",
    "        'transparent': 'False', # Whether the map background should be transparent. Values are true or false. Default is false\n",
    "        'bgcolor': 'FFFFFF', # Background color for the map image. Value is in the form RRGGBB. Default is FFFFFF (white)\n",
    "        'exceptions': '', # Format in which to report exceptions. Default value is application/vnd.ogc.se_xml\n",
    "        'time': first_time_extent_value, # Time value or range for map data. See Time Support in GeoServer WMS for more information\n",
    "                  # https://docs.geoserver.org/2.22.x/en/user/services/wms/time.html#wms-time\n",
    "        'sld': '', # A URL referencing a StyledLayerDescriptor XML file which controls or enhances map layers and styling\n",
    "        'sld_body': '' # A URL-encoded StyledLayerDescriptor XML document which controls or enhances map layers and styling\n",
    "    }\n",
    "    response = requests.get(wms_url, params=params)\n",
    "    image_url = response.url\n",
    "\n",
    "    export_path = pathlib.Path(f\"exports/{test}_{var}_{timestamp}.png\")\n",
    "    with export_path.open(\"wb\") as f:\n",
    "        f.write(response.content)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "egis",
   "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.19"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}