aboutsummaryrefslogtreecommitdiff
path: root/aggregator/dwd_icon.py
diff options
context:
space:
mode:
Diffstat (limited to 'aggregator/dwd_icon.py')
-rwxr-xr-xaggregator/dwd_icon.py27
1 files changed, 15 insertions, 12 deletions
diff --git a/aggregator/dwd_icon.py b/aggregator/dwd_icon.py
index 88288f3..0827469 100755
--- a/aggregator/dwd_icon.py
+++ b/aggregator/dwd_icon.py
@@ -97,18 +97,21 @@ def clean_output_dir(directory, target):
for f in to_delete:
os.unlink(os.path.join(directory, f))
-def load_data(name, output, description = None, clean = False, **kwargs):
- run, date = get_current_run()
- filename = f'{name}_{date}_{run}.grib2'
- target = os.path.join(output, filename)
-
- if not os.path.exists(target):
- download_dwd_gribs(date, run, target, output, **kwargs)
- else:
- print(f'{target} alreasy exists. Using the cached version.')
-
- if clean:
- clean_output_dir(output, filename)
+def load_data(name, output, description = None, clean = False, force_filename = None, **kwargs):
+ target = force_filename
+
+ if target is None:
+ run, date = get_current_run()
+ filename = f'{name}_{date}_{run}.grib2'
+ target = os.path.join(output, filename)
+
+ if not os.path.exists(target):
+ download_dwd_gribs(date, run, target, output, **kwargs)
+ else:
+ print(f'{target} already exists. Using the cached version.')
+
+ if clean:
+ clean_output_dir(output, filename)
ds = xr.load_dataset(target, engine='cfgrib')
if description is not None: