Changeset 45b99e6 in flex_extract.git for source/python/mods/get_mars_data.py

Timestamp:

Feb 15, 2019, 3:48:31 PM (5 years ago)

Author:

Anne Philipp <anne.philipp@…>

Branches:

master, ctbto, dev

Children:

b9f1214

Parents:

ae2756e

Message:

added possibility to extract ensemble members with ELDA stream (and ENFO)

File:

• source/python/mods/get_mars_data.py (modified) (2 diffs)

source/python/mods/get_mars_data.py

@@ -209 +209 @@
 
 
+def check_dates_for_nonflux_fc_times(types, times):
+    '''
+    '''
+    for ty, ti in zip(types,times):
+        if ty.upper() == 'FC' and int(ti) == 18:
+            return True
+    return False
+
+
 def mk_dates(c, fluxes):
     '''Prepares start and end date depending on flux or non flux data.
@@ -258 +267 @@
         start = start - timedelta(days=1)
         end = end + timedelta(days=1)
+
+    # if we have non-flux forecast data starting at 18 UTC
+    # we need to start retrieving data one day in advance
+    if not fluxes and check_dates_for_nonflux_fc_times(c.type, c.time):
+        start = start - timedelta(days=1)
 
     return start, end, chunk