Changeset 25125 for issm/trunk-jpl/src/m/io/loadvars.py

Timestamp:

06/23/20 15:42:10 (5 years ago)

Author:

jdquinn

Message:

CHG: Saving progress on MATLAB -> Python translations; cleanup

File:

• issm/trunk-jpl/src/m/io/loadvars.py (modified) (6 diffs)

issm/trunk-jpl/src/m/io/loadvars.py

@@ +1 @@
+from collections import OrderedDict
+from netCDF4 import Dataset
+import numpy as np
+from re import findall
 import shelve
-import numpy as np
-from netCDF4 import Dataset
-from re import findall
-from collections import OrderedDict
+
 from model import *
 #hack to keep python 2 compatibility
@@ -15 +16 @@
 
 def loadvars(*args, OL):
-    """
-    LOADVARS - function to load variables to a file.
-
-    This function loads one or more variables from a file.  The names of the variables
-    must be supplied.  If more than one variable is specified, it may be done with
-    a list of names or a dictionary of name as keys.  The output type will correspond
-    to the input type.  All the variables in the file may be loaded by specifying only
-    the file name.
+    '''
+    LOADVARS - function to load variables from a file.
+
+    This function loads one or more variables from a file. The names of the 
+    variables must be supplied. If more than one variable is specified, it may 
+    be done with a list of names or a dictionary of name as keys. The output 
+    type will correspond to the input type. All the variables in the file may 
+    be loaded by specifying only the file name.
 
     Usage:
@@ -29 +30 @@
         nvdict = loadvars('shelve.dat', {'a':None, 'b':None})
         nvdict = loadvars('shelve.dat')
-
-    """
+    '''
 
     filename = ''
@@ -40 +40 @@
         if not filename:
             filename = '/tmp/shelve.dat'
-
     else:
         raise TypeError("Missing file name.")
@@ -47 +46 @@
         for name in args[1:]:
             nvdict[name] = None
-
     elif len(args) == 2 and isinstance(args[1], list):  # (filename, [names])
         for name in args[1]:
             nvdict[name] = None
-
     elif len(args) == 2 and isinstance(args[1], dict):  # (filename, {names:values})
         nvdict = args[1]
-
     elif len(args) == 1:  #  (filename)
         pass
-
     else:
         raise TypeError("Unrecognized input arguments.")
@@ -78 +73 @@
                 print(("Variable '%s' loaded." % name))
         my_shelf.close()
-
     else:  #We used netcdf for the save
         try: