From cc335316b6c468690ffecae2616efcf0fb1247f7 Mon Sep 17 00:00:00 2001
From: lxnay <lxnay@cd1c1023-2f26-0410-ae45-c471fc1f0318>
Date: Fri, 6 Mar 2009 20:52:52 +0000
Subject: [PATCH] Entropy/entropyTools: - compress_files, new handy function
 that compresses several files into one (no subdirs handling)

git-svn-id: http://svn.sabayonlinux.org/projects/entropy/trunk@3113 cd1c1023-2f26-0410-ae45-c471fc1f0318
---
 libraries/entropyTools.py | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/libraries/entropyTools.py b/libraries/entropyTools.py
index 0532fcefc..f7f14cef0 100644
--- a/libraries/entropyTools.py
+++ b/libraries/entropyTools.py
@@ -645,6 +645,28 @@ def compress_file(file_path, destination_path, opener, compress_level = None):
     f_out.close()
     f_in.close()
 
+# files_to_compress must be a list of valid file paths
+def compress_files(dest_file, files_to_compress, compressor = "bz2"):
+
+    if compressor not in ("bz2","gz",):
+        raise AttributeError("invalid compressor specified")
+
+    import tarfile
+    id_strings = {}
+    tar = tarfile.open(dest_file,"w:%s" % (compressor,))
+    try:
+        for path in files_to_compress:
+            exist = os.lstat(path)
+            tarinfo = tar.gettarinfo(path, os.path.basename(path))
+            tarinfo.uname = id_strings.setdefault(tarinfo.uid, str(tarinfo.uid))
+            tarinfo.gname = id_strings.setdefault(tarinfo.gid, str(tarinfo.gid))
+            if not stat.S_ISREG(exist.st_mode): continue
+            tarinfo.type = tarfile.REGTYPE
+            with open(path) as f:
+                tar.addfile(tarinfo, f)
+    finally:
+        tar.close()
+
 def unpackGzip(gzipfilepath):
     import gzip
     filepath = gzipfilepath[:-3] # remove .gz
@@ -1960,9 +1982,6 @@ def uncompressTarBz2(filepath, extractPath = None, catchEmpty = False):
     finally:
         tar.close()
 
-    #'''
-
-
     if os.listdir(extractPath):
         return 0
     return -1