From 637edcf7b9da61edebfe5c8d0c7bafea3543ebba Mon Sep 17 00:00:00 2001
From: Jonas Gunz <himself@jonasgunz.de>
Date: Tue, 7 Nov 2023 00:22:43 +0100
Subject: description in aggregator

---
 aggregator/dwd_icon.py        | 18 +++++++++++-------
 plotter/vertical_from_grib.py |  4 ++--
 2 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/aggregator/dwd_icon.py b/aggregator/dwd_icon.py
index b8a4470..2bd3a61 100755
--- a/aggregator/dwd_icon.py
+++ b/aggregator/dwd_icon.py
@@ -45,7 +45,7 @@ def unpack_bz2(dest):
 def download_dwd_gribs(
         date, run, target, output, model, steps, model_long,
         pressure_level_parameters, parameter_caps_in_filename,
-        single_level_parameters
+        single_level_parameters, pressure_levels
 ):
     misc.create_output_dir(output)
 
@@ -54,14 +54,14 @@ def download_dwd_gribs(
     for step in steps:
         step_str = f'{step:03d}'
 
-        for parameter in config['pressure_level_parameters']:
-            parameter2 = parameter.upper() if config['parameter_caps_in_filename'] else parameter
+        for parameter in pressure_level_parameters:
+            parameter2 = parameter.upper() if parameter_caps_in_filename else parameter
 
-            for level in config['pressure_levels']:
+            for level in pressure_levels:
                 filename = f'{model_long}_regular-lat-lon_pressure-level_{date}{run}_{step_str}_{level}_{parameter2}.grib2.bz2'
                 URL = f'{BASE}/{model}/grib/{run}/{parameter}/{filename}'
 
-                to_download.append((URL, os.path.join(config['output'], filename)))
+                to_download.append((URL, os.path.join(output, filename)))
 
         for parameter in single_level_parameters:
             parameter2 = parameter.upper() if parameter_caps_in_filename else parameter
@@ -89,7 +89,7 @@ def download_dwd_gribs(
     if res.returncode != 0:
         print('rm failed with: ', res.stderr)
 
-def load_data(name, output, **kwargs):
+def load_data(name, output, description = None, **kwargs):
     run, date = get_current_run()
     target = os.path.join(output, f'{name}_{date}_{run}.grib2')
 
@@ -98,7 +98,11 @@ def load_data(name, output, **kwargs):
     else:
         print(f'{target} alreasy exists. Using the cached version.')
 
-    return xr.load_dataset(target, engine='cfgrib')
+    ds = xr.load_dataset(target, engine='cfgrib')
+    if description is not None:
+        ds.attrs['_description'] = description
+    return ds
+
 
 debug_config = {
     'output':'dwd_icon-eu',
diff --git a/plotter/vertical_from_grib.py b/plotter/vertical_from_grib.py
index 9f927a7..726c8be 100755
--- a/plotter/vertical_from_grib.py
+++ b/plotter/vertical_from_grib.py
@@ -66,8 +66,8 @@ def _plot(data, output, lat, lon, name, analysis=None):
         if analysis is not None:
             skt.addAnalysis(shade=True, analysis=analysis)
 
-        # TODO get from source!
-        skt.addInfo("FORECAST DWD ICON-D2")
+        if '_description' in data.attrs:
+            skt.addInfo(data.attrs['_description'])
 
         init_for_filename = init.strftime('%Y-%m-%d-%HUTC')
 
-- 
cgit v1.2.3