#17: speeded up loading of compressed files

pymule/loader.py

@@ -4,6 +4,7 @@ from errortools import *
 import os
 import inspect
 import tarfile
+import StringIO
 
 loadargs = {}
 
@@ -14,7 +15,18 @@ def importreg(r, folder='.', filenames=None,
 
     tar = None
     if not os.path.isdir(folder):
-        tar = tarfile.open(folder)
+        if os.path.splitext(folder)[1] == '.bz2':
+            import bz2
+            with bz2.BZ2File(folder, 'r') as b:
+                fp = StringIO.StringIO(b.read())
+            tar = tarfile.open(fileobj=fp)
+        elif os.path.splitext(folder)[1] == '.gz':
+            import gzip
+            with gzip.open(folder, 'r') as b:
+                fp = StringIO.StringIO(b.read())
+            tar = tarfile.open(fileobj=fp)
+        else:
+            tar = tarfile.open(folder)
 
     if filenames is None:
         if tar: