Changeset 1895 for inundation/caching/caching.py

Timestamp:

Oct 11, 2005, 5:00:44 PM (19 years ago)

Author:

ole

Message:

Caught potential memory problem

File:

• inundation/caching/caching.py (modified) (1 diff)

inundation/caching/caching.py

@@ -1254 +1254 @@
       
 
-    Ts  = pickler.dumps(T,bin)
-    TsC = zlib.compress(Ts,comp_level)
-    file.write(TsC)
+    try:
+      Ts  = pickler.dumps(T,bin)
+    except MemoryError:
+      msg = '****WARNING (caching.py): Could not pickle data for compression.'
+      msg += ' Try using compression = False'
+      raise MemoryError, msg
+    else:  
+      #Compressed pickling      
+      TsC = zlib.compress(Ts,comp_level)
+      file.write(TsC)
   else:
-    pickler.dump(T,file,bin)
+      #Uncompressed pickling
+      pickler.dump(T,file,bin)
+
+      
 
 # -----------------------------------------------------------------------------