Changeset 403cbf1 in flex_extract.git for source/python/mods/get_mars_data.py

Timestamp:

Nov 16, 2018, 10:27:15 PM (5 years ago)

Author:

Anne Philipp <anne.philipp@…>

Branches:

master, ctbto, dev

Children:

ac48dae

Parents:

2e62398

Message:

added header writing to marsrequest file and corrected bug in normalexit call in submit

File:

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

source/python/mods/get_mars_data.py

@@ -74 +74 @@
 from classes.EcFlexpart import EcFlexpart
 from classes.UioFiles import UioFiles
+from classes.MarsRetrieval import MarsRetrieval
 
 try:
@@ -131 +132 @@
         make_dir(c.inputdir)
 
-    if c.request == 0 or c.request == 2:
+    if c.request == 0:
         print("Retrieving EC data!")
-    elif c.request == 1:
-        print("Printing mars requests!")
+    else:
+        if c.request == 1:
+            print("Printing mars requests!")
+        elif c.request == 2:
+            print("Retrieving EC data and printing mars request!")
+        # first, write header with the mars parameter to file
+        # create a dummy MarsRetrieval to get parameter
+        MR = MarsRetrieval(None, None)
+        attrs = vars(MR).copy()
+        del attrs['server']
+        del attrs['public']
+        marsfile = os.path.join(c.inputdir, _config.FILE_MARS_REQUESTS)
+        with open(marsfile, 'w') as f:
+            f.write('request_number' + ', ')
+            f.write(', '.join(str(key) for key in sorted(attrs.iterkeys())))
+            f.write('\n')
 
     print("start date %s " % (c.start_date))