Use opengribs.org API

Author: svenseeberg

opendrift_leeway_webgui/leeway/tasks.py

@@ -8,7 +8,7 @@
 from django.utils import timezone
 
 from .celery import app
-from .utils import send_result_mail
+from .utils import request_opengribs_file, send_result_mail
 
 logger = logging.getLogger(__name__)
 
@@ -24,6 +24,7 @@ def run_leeway_simulation(request_id):
     simulation = LeewaySimulation.objects.get(uuid=request_id)
     simulation.simulation_started = timezone.now()
     simulation.save()
+    get_opengribs_data(simulation.longitude, simulation.latitude)
     params = [
         "docker",
         "run",
@@ -50,6 +51,7 @@ def run_leeway_simulation(request_id):
         str(simulation.duration),
         "--id",
         str(simulation.uuid),
+        "--no-web",
     ]
     with subprocess.Popen(
         params, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True

opendrift_leeway_webgui/leeway/utils.py

@@ -2,7 +2,11 @@
 Utilities
 """
 
+import os
+import tempfile
+
 from django.apps import apps
+from django.conf import settings
 from django.contrib.auth import get_user_model
 from django.core.mail import EmailMessage, send_mail
 from dms2dec.dms_convert import dms2dec
@@ -235,3 +239,55 @@ def normalize_dms2dec(data):
     if "°" in data:
         return dms2dec(data)
     return data
+
+
+def create_opengribs_bounding_box(longitude, latitude):
+    """
+    Calculate Bounding Box for GRIBS file download.
+    """
+    return {
+        "long_min": longitude - 1,
+        "long_max": longitude + 1,
+        "lat_min": latitude - 1,
+        "lat_max": latitude + 1,
+    }
+
+
+def request_opengribs_file(bounding_box):
+    """
+    Request GRIBS file on opengribs.org
+    """
+    request = requests.get(
+        url=(
+            f"http://grbsrv.opengribs.org/getmygribs2.php?osys=Unknown&ver=1.2.6&model=icon_p25_&"
+            f"la1={bounding_box['lat_min']}&la2={bounding_box['lat_max']}&"
+            f"lo1={bounding_box['long_min']}&lo2={bounding_box['long_max']}&"
+            f"intv=3&days=8&cyc=last&par=W;&wmdl=none&wpar=h;d;p;"
+        )
+    )
+    return request.json()["message"]["url"]
+
+
+def get_opengribs_file(url):
+    """
+    Get OpenGRIBS file and save to disk.
+    """
+    response = requests.get(url, stream=True)
+    if response.status_code != 200:
+        return
+    total_size = int(response.headers.get("content-length", 0))
+    os.makedir(os.path.join(settings.SIMULATION_ROOT, "input"))
+    filename = os.path.join(settings.SIMULATION_ROOT, "input", url.split("/")[-1])
+    with open(filename, "wb") as f:
+        for chunk in response.iter_content(chunk_size=1024):
+            f.write(chunk)
+    return filename
+
+
+def get_opengribs_data(longitude, latitde):
+    """
+    Get data from OpenGRIBS
+    """
+    bbox = create_opengribs_bounding_box(longitude, latitude)
+    download_url = request_opengribs_file(bbox)
+    return get_opengribs_file(download_url)

opendrift_leeway_webgui/simulation.py

@@ -88,7 +88,7 @@ def main():
     sources = [
         os.path.join(INPUTDIR, data_file)
         for data_file in os.listdir(INPUTDIR)
-        if data_file.endswith(".nc")
+        if data_file.endswith(".nc") or data_file.endswith(".grb2")
     ]
 
     if not args.no_web:

setup.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-""" Setup.py """
+"""Setup.py"""
 from setuptools import setup
 
 setup()