Changeset efdb01a in flex_extract.git for python/prepareFLEXPART.py

Timestamp:

May 9, 2018, 12:15:00 PM (6 years ago)

Author:

Anne Philipp <anne.philipp@…>

Branches:

master, ctbto, dev

Children:

991df6a

Parents:

02c8c50

Message:

whole bunch of modifications due to new structure of ECMWFDATA, added basics of documentation, minor programming corrections

File:

• python/prepareFLEXPART.py (modified) (9 diffs, 1 prop)

python/prepareFLEXPART.py

@@ -30 +30 @@
         - added documentation
         - minor changes in programming style for consistence
+        - BUG: removed call of cleanup-Function after call of prepareFlexpart
+                since it is already called in prepareFlexpart at the end!
 
 @License:
-    (C) Copyright 2014.
+    (C) Copyright 2014-2018.
 
     This software is licensed under the terms of the Apache Licence Version 2.0
@@ -40 +42 @@
     - A standard python 2.6 or 2.7 installation
     - dateutils
-    - matplotlib (optional, for debugging)
     - ECMWF specific packages, all available from https://software.ecmwf.int/
         ECMWF WebMARS, gribAPI with python enabled, emoslib and
@@ -52 +53 @@
         surface fields for FLEXPART runs
 """
-
+# ------------------------------------------------------------------------------
+# MODULES
+# ------------------------------------------------------------------------------
 import calendar
 import shutil
 import datetime
 import time
-import os,inspect,sys
+import os
+import inspect
+import sys
 import socket
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
-# add path to submit.py to pythonpath so that python finds its buddies
-localpythonpath=os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
-if localpythonpath not in sys.path:
-    sys.path.append(localpythonpath)
-from UIOTools import UIOFiles
-#from string import strip
-from GribTools import GribTools
-from FlexpartTools import ECFlexpart, Control, interpret_args_and_control, cleanup
+import UIOFiles
+import Control
+import Tools
+import ECFlexpart
 
-hostname=socket.gethostname()
-ecapi= 'ecmwf' not in hostname
-
+hostname = socket.gethostname()
+ecapi = 'ecmwf' not in hostname
 try:
     if ecapi:
@@ -78 +78 @@
     ecapi = False
 
-
+# add path to submit.py to pythonpath so that python finds its buddies
+localpythonpath=os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
+if localpythonpath not in sys.path:
+    sys.path.append(localpythonpath)
+# ------------------------------------------------------------------------------
+# FUNCTION
+# ------------------------------------------------------------------------------
 def prepareFLEXPART(args, c):
     '''
@@ -121 +127 @@
                         day=int(c.end_date[6:]))
 
-    # to deaccumulated the fluxes correctly
+    # to deaccumulate the fluxes correctly
     # one day ahead of the start date and
     # one day after the end date is needed
@@ -128 +134 @@
 
     # get all files with flux data to be deaccumulated
-    inputfiles = UIOFiles(['.grib', '.grb', '.grib1',
+    inputfiles = UIOFiles.UIOFiles(['.grib', '.grb', '.grib1',
                            '.grib2', '.grb1', '.grb2'])
 
@@ -138 +144 @@
 
     # deaccumulate the flux data
-    flexpart = ECFlexpart(c, fluxes=True)
+    flexpart = ECFlexpart.ECFlexpart(c, fluxes=True)
     flexpart.write_namelist(c, 'fort.4')
     flexpart.deacc_fluxes(inputfiles, c)
@@ -146 +152 @@
 
     # get a list of all files from the root inputdir
-    inputfiles = UIOFiles(['.grib', '.grb', '.grib1',
+    inputfiles = UIOFiles.UIOFiles(['.grib', '.grb', '.grib1',
                            '.grib2', '.grb1', '.grb2'])
 
@@ -157 +163 @@
         start = startm1
 
-    flexpart = ECFlexpart(c, fluxes=False)
+    flexpart = ECFlexpart.ECFlexpart(c, fluxes=False)
     flexpart.create(inputfiles, c)
     flexpart.process_output(c)
 
     # check if in debugging mode, then store all files
+    # otherwise delete temporary files
     if int(c.debug) != 0:
         print('Temporary files left intact')
     else:
-        cleanup(c)
+        Tools.cleanup(c)
 
     return
 
 if __name__ == "__main__":
-    args, c = interpret_args_and_control()
+    args, c = Tools.interpret_args_and_control()
     prepareFLEXPART(args, c)
-    cleanup(c)
-