Changeset 2783 in lmdz_wrf for trunk/tools

Timestamp:

Jan 14, 2020, 2:24:25 PM (5 years ago)

Author:

lfita

Message:

Adding 'kfind' option (how to look into string variables) in `stations_values'

Location:

trunk/tools

Files:

• generic.py (modified) (2 diffs)
• generic_tools.py (modified) (4 diffs)

trunk/tools/generic.py

@@ -474 +474 @@
 
 elif oper == 'stations_values':
-    Nvals = 2
+    Nvals = 3
     vals = opts.values.split(cS)
     if vals[0] == 'h':
@@ -486 +486 @@
             print gen.stations_values.__doc__
             quit(-1)
-        result = gen.stations_values(vals[0], vals[1])
+        result = gen.stations_values(vals[0], vals[1], vals[2])
         print gen.numVector_String(result,':')
 

trunk/tools/generic_tools.py

@@ -17370 +17370 @@
     return newlist
 
-def stations_values(stcrit,stval,Nexpctst='all',Notfound=True):
+def stations_values(stcrit,stval,kfind='strict',Nexpctst='all',Notfound=True):
     """ Function to provide information from a given station from one of its values 
       from ASCII file 'OBStations.csv'
-    stations_values([stcrit],[stval],[Nexpctst],[Notfound])
+    stations_values([stcrit],[stval],[kfind],[Nexpctst],[Notfound])
         [stcrit]: name of the criteria from OBStations.csv
           one of: 'station_name', 'WMOid', 'longitude', 'lon_deg', 'lon_min', 
@@ -17395 +17395 @@
           [add2]: additional value 2
         [stval]: value of [stcrit] to be matched
+        [kfind]: type of finding (only valid for string values)
+          'strict': the value to find must be exactly meet
+          'partial': the value to find can be partialy meet (present within)
         [Nexpctst]: expected quantity of stations to retrieve ('all', default)
           'all': all matching entries. It will return a dictionary
@@ -17405 +17408 @@
     >>> stations_values('WMOid','87582','all')
     {0: ['AEROPARQUE BUENOS AIRES', 87582, 34.566667, 34, 34, 0.0, 58.416667, 58, 
-       25, 0.0, 6.0, u'CAPITAL FEDERAL', u'Argentina',
-       u'Aeroparque Airport Buenos Aires', ' ', ' ']}
+      25, 0.0, 6.0, u'CAPITAL FEDERAL', u'Argentina',
+      u'Aeroparque Airport Buenos Aires', ' ', ' ']}
     >>> stations_values('WMOid','87582','one')
     ['AEROPARQUE BUENOS AIRES', 87582, 34.566667, 34, 34, 0.0, 58.416667, 58, 
-       25, 0.0, 6.0, u'CAPITAL FEDERAL', u'Argentina', 
-       u'Aeroparque Airport Buenos Aires', ' ', ' ']
+      25, 0.0, 6.0, u'CAPITAL FEDERAL', u'Argentina', 
+      u'Aeroparque Airport Buenos Aires', ' ', ' ']
+    >>> stations_values('station_name','Carlini','partial','all')
+    {0: ['BASE Carlini', -99999, -58.6677777778, -58, 40, 4.0, -62.2383333333, -62, 
+      14, 18.0, 10.0, u' Antartic', u' Argentina', u' Base ant\xe1rtica Carlini', 
+      u'-', u'-']}
     """
     import subprocess as sub
     fname='stations_values'
+
+    availkfind = ['strict', 'partial']
 
     criteriavail = ['station_name', 'WMOid', 'longitude', 'lon_deg', 'lon_min',      \
@@ -17470 +17479 @@
             if type(stvalues[icrit]) == type('S') or type(stvalues[icrit]) ==        \
               type(unicode('S')):
-                if begend_spaces(stvalues[icrit]) == begend_spaces(stv):
-                    stations[Nst] = stvalues
-                    Nst = Nst + 1
+                cleanstval = begend_spaces(stvalues[icrit])
+                cleanval = begend_spaces(stv)
+                if kfind == 'strict':
+                    # Strict finding
+                    if  cleanstval == cleanval:
+                        stations[Nst] = stvalues
+                        Nst = Nst + 1
+                elif kfind == 'partial':
+                    # Relaxed finding
+                    if cleanstval.find(cleanval) != -1:
+                        stations[Nst] = stvalues
+                        Nst = Nst + 1
+                else:
+                    print errormsg
+                    print '  ' + fname + ": kind of finding '" + kfind +             \
+                      "' not avaialble !!"
+                    print '    available ones:', availkfind
+                    quit(-1)
+                
             else:
                 if stvalues[icrit] == stv: