Changeset ae2756e in flex_extract.git

Timestamp:

Feb 15, 2019, 3:46:45 PM (5 years ago)

Author:

Anne Philipp <anne.philipp@…>

Branches:

master, ctbto, dev

Children:

45b99e6

Parents:

5364d08

Message:

added a check for the number parameter

Location:

source/python

Files:

• classes/ControlFile.py (modified) (2 diffs)
• mods/checks.py (modified) (2 diffs)

source/python/classes/ControlFile.py

@@ -59 +59 @@
                          check_acctime, check_accmaxstep, check_time,
                          check_logicals_type, check_len_type_time_step,
-                         check_addpar, check_job_chunk)
+                         check_addpar, check_job_chunk, check_number)
 
 # ------------------------------------------------------------------------------
@@ -646 +646 @@
         self.job_chunk = check_job_chunk(self.job_chunk)
 
+        self.number = check_number(self.number, self.mailfail)
+
         return
 

source/python/mods/checks.py

@@ -30 +30 @@
 from tools import my_error, silent_remove
 from datetime import datetime
+import numpy as np
 # ------------------------------------------------------------------------------
 # FUNCTIONS
@@ -825 +826 @@
 
     return job_chunk
+
+
+def check_number(number, mailfail):
+    '''Check for correct string format of ensemble member numbers.
+
+    Parameters
+    ----------
+    number : str
+        List of ensemble member forecast runs.
+
+    mailfail : list of str
+        Contains all email addresses which should be notified.
+        It might also contain just the ecmwf user name which will trigger
+        mailing to the associated email address for this user.
+
+    Return
+    ------
+    number : str
+        String with list of ensemble member forecast runs. E.g. '01/02/03/04'
+    '''
+
+    if '/' in number:
+        numbers = number.split('/')
+        if 'to' in number.lower() and 'by' in number.lower():
+            number = '{:0>3}'.format(int(numbers[0])) + '/TO/' + \
+                     '{:0>3}'.format(int(numbers[2])) + '/BY/' + \
+                     '{:0>3}'.format(int(numbers[4]))
+        elif 'to' in number.lower() and 'by' not in number.lower():
+            number = '{:0>3}'.format(int(numbers[0])) + '/TO/' + \
+                     '{:0>3}'.format(int(numbers[2]))
+        else:
+            numbers = ['{:0>3}'.format(i) for i in numbers]
+            number = '{:0>3}/'.join(numbers)
+    elif number.isdigit():
+        number = '{:0>3}'.format(int(number))
+    else:
+        pass
+
+    return number