Changeset 4468

Timestamp:

May 21, 2007, 11:04:04 AM (18 years ago)

Author:

nick

Message:

add store_parameter which writes meta data to a file

Location:

anuga_core/source/anuga/abstract_2d_finite_volumes

Files:

• test_util.py (modified) (2 diffs)
• util.py (modified) (2 diffs)

anuga_core/source/anuga/abstract_2d_finite_volumes/test_util.py

@@ -1287 +1287 @@
         
         assert allclose(x[:,0], [1.3, 0.0,4.5, 1.0])
+        
+    def test_store_parameters(self):
+        
+        from os import sep, getenv
+        
+        home = getenv('INUNDATIONHOME')
+        output_dir=home+sep+'data'
+        
+        kwargs = {'file_name':'new2.txt','output_dir':output_dir,'who':'me', 'what':'detail', 'how':2, 'why':241,'Completed':'yes'}
+#        {'data_origin': data_georef.get_origin(),
+#                  'mesh_origin': mesh_georef.get_origin(),
+#                  'alpha': alpha,
+#                  'verbose': verbose}
+        
+#        fileName = tempfile.mktemp(".csv")
+#        store_parameters(kwargs,fileName)
+#        file_name='temp.csv'
+        store_parameters(**kwargs)
+#        store_parameters(file_name=file_name, scenario_name='dampier',\
+#                         who="me", time_thining=12,tide=2.6)
+
+        
+        fid = open(str(kwargs.pop('file_name')))
+        header = fid.readline()
+        line = fid.readline()
+        fid.close()
+        print'header',header,'lines',line
+        
+        #file exists
+#        assert access(file_name,F_OK)
+        a=['who,time']
+#        assert allclose(header, a)
+#        assert allclose(header, [who,time])
+#        assert header=='who,time1
+#        assert allclose(line,['me',12])
+        
 
      
@@ -1294 +1330 @@
 if __name__ == "__main__":
     suite = unittest.makeSuite(Test_Util,'test')
-    #suite = unittest.makeSuite(Test_Util,'test_get_data_from_file')
+#    suite = unittest.makeSuite(Test_Util,'test_store_parameters')
     runner = unittest.TextTestRunner()
     runner.run(suite)

anuga_core/source/anuga/abstract_2d_finite_volumes/util.py

@@ -1383 +1383 @@
 def copy_code_files(dir_name, filename1, filename2):
     """Copies "filename1" and "filename2" to "dir_name". Very useful for 
-    information management """
+    information management 
+    filename1 and filename2 are both absolute pathnames    
+    """
 
     if access(dir_name,F_OK) == 0:
@@ -1451 +1453 @@
     return header_fields, data
 
-
-
+def store_parameters(verbose=False,**kwargs):
+    """
+    Must have a file_name keyword arg, this is what is writing to.
+    might be a better way to do this using CSV module Writer and writeDict
+    
+    writes file to "output_dir" unless "completed" is in kwargs, then it writes to 
+    "file_name" kwargs 
+     Returns a object which is a subset of the original
+        and the data points and attributes in this new object refer to
+        the indices provided
+        
+        Input
+            indices- a list of integers that represent the new object
+        Output
+            New geospatial data object representing points specified by 
+            the indices 
+    """
+    import types
+    import os
+
+    # Check that kwargs is a dictionary
+    if type(kwargs) != types.DictType:
+        raise TypeError
+    
+    try:
+        kwargs['completed']
+        completed=True
+    except:
+        completed=False
+
+    # assert that a file_name exists
+    
+    #get file name and removes from dict
+    if completed:
+        try:
+            file = str(kwargs.pop('file_name'))
+        except:
+            raise 'kwargs must have file_name'
+    else:
+        try:
+            file = str(kwargs.pop('output_dir'))+'detail_temp.csv'
+        except:
+            raise 'kwargs must have output_dir'
+        
+    
+#    print kwargs
+    #extracts the header info and the new line info
+    line=''
+    header=''
+    count=0
+    keys = kwargs.keys()
+#    print 'keys',keys
+    keys.sort()
+#    print 'keys',keys
+    
+#    for k in kwargs.keys():
+    #used the sorted keys to create the header and line data
+    for k in keys:
+        print "%s = %s" %(k, kwargs[k]) 
+        header = header+str(k)
+        line = line+str(kwargs[k])
+        count+=1
+        if count <len(kwargs):
+            header = header+','
+            line = line+','
+
+
+    # checks the header info, if the same, then write, if not create a new file
+    #try to open!
+#    print'file name',file
+    try:
+        fid = open(file,"r")
+        file_header=fid.readline()
+        fid.close()
+        if verbose: print 'read file header %s' %file_header
+        
+    except:
+        msg = 'try to create new file',file
+        if verbose: print msg
+        #tries to open file, maybe directory is bad
+        try:
+            fid = open(file,"w")
+            fid.writelines(header+'\n')
+            fid.close()
+            file_header=header
+        except:
+            msg = 'cannot create new file',file
+            raise msg
+            
+    #if header is same or this is a new file
+    if file_header.strip('\n')==header:
+        fid=open(file,"a")
+        #write new line
+        fid.writelines(line+'\n')
+        fid.close()
+    else:
+        #backup plan, if header is different and has completed will append info to 
+        #end of details_temp.cvs file in output directory
+        file = str(kwargs['output_dir'])+'detail_temp.csv'
+        fid=open(file,"a")
+        fid.writelines(header+'\n')
+        fid.writelines(line+'\n')
+        fid.close()
+        msg = 'file header does not match input info, the input variables have changed, change file name'
+        raise msg
+
+      
+    
+    
+        
+    
+    
+     
+
+