From 2a9909d4a8ac78853b7a50a203a312469e3c27fb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fatih=20A=C5=9F=C4=B1c=C4=B1?= <fatih@pardus.org.tr>
Date: Sun, 26 Sep 2010 09:06:26 +0000
Subject: [PATCH] file: Modify File class to accept multiple compression types

---
 pisi/api.py   |  7 +++----
 pisi/file.py  | 20 ++++++++++++--------
 pisi/index.py |  4 +++-
 3 files changed, 18 insertions(+), 13 deletions(-)

diff --git a/pisi/api.py b/pisi/api.py
index eeaf7311..f645854b 100644
--- a/pisi/api.py
+++ b/pisi/api.py
@@ -774,10 +774,9 @@ def index(dirs=None, output='pisi-index.xml', skip_sources=False, skip_signing=F
         ctx.ui.info(_('* Building index of PiSi files under %s') % repo_dir)
         index.index(repo_dir, skip_sources)
 
-    if skip_signing:
-        index.write(output, sha1sum=True, compress=pisi.file.File.xz, sign=None)
-    else:
-        index.write(output, sha1sum=True, compress=pisi.file.File.xz, sign=pisi.file.File.detached)
+    compression = pisi.file.File.COMPRESSION_TYPE_XZ
+    sign = None if skip_signing else pisi.file.File.detached
+    index.write(output, sha1sum=True, compress=compression, sign=sign)
     ctx.ui.info(_('* Index file written'))
 
 @locked
diff --git a/pisi/file.py b/pisi/file.py
index 8e1ecb3b..bf134bc4 100644
--- a/pisi/file.py
+++ b/pisi/file.py
@@ -51,12 +51,16 @@ class InvalidSignature(pisi.Error):
 
 class File:
 
+    # Compression types
+    COMPRESSION_TYPE_AUTO = 0
+    COMPRESSION_TYPE_BZ2 = 1
+    COMPRESSION_TYPE_XZ = 2
+
     (read, write) = range(2)            # modes
-    (xz, bz2, auto) = range(3)          # compress enums
     (detached, whatelse) = range(2)
 
-    __compressed_file_extensions = {".xz": xz,
-                                    ".bz2": bz2}
+    __compressed_file_extensions = {".xz": COMPRESSION_TYPE_XZ,
+                                    ".bz2": COMPRESSION_TYPE_BZ2}
 
     @staticmethod
     def make_uri(uri):
@@ -69,7 +73,7 @@ class File:
 
     @staticmethod
     def choose_method(filename, compress):
-        if compress == File.auto:
+        if compress == File.COMPRESSION_TYPE_AUTO:
             for ext, method in File.__compressed_file_extensions.items():
                 if filename.endswith(ext):
                     return method
@@ -85,11 +89,11 @@ class File:
     @staticmethod
     def decompress(localfile, compress):
         compress = File.choose_method(localfile, compress)
-        if compress == File.xz:
+        if compress == File.COMPRESSION_TYPE_XZ:
             import lzma
             open(localfile[:-3], "w").write(lzma.LZMAFile(localfile).read())
             localfile = localfile[:-3]
-        elif compress == File.bz2:
+        elif compress == File.COMPRESSION_TYPE_BZ2:
             import bz2
             open(localfile[:-4], "w").write(bz2.BZ2File(localfile).read())
             localfile = localfile[:-4]
@@ -185,7 +189,7 @@ class File:
         self.__file__.close()
         if self.mode == File.write:
             compressed_file = None
-            if self.compress == File.xz:
+            if self.compress & File.COMPRESSION_TYPE_XZ:
                 import lzma
                 compressed_file = self.localfile + ".xz"
                 options = {"level": 9}
@@ -194,7 +198,7 @@ class File:
                 lzma_file.write(open(self.localfile, "r").read())
                 lzma_file.close()
 
-            elif self.compress == File.bz2:
+            if self.compress & File.COMPRESSION_TYPE_BZ2:
                 import bz2
                 compressed_file = self.localfile + ".bz2"
                 bz2.BZ2File(compressed_file, "w").write(open(self.localfile, "r").read())
diff --git a/pisi/index.py b/pisi/index.py
index ee313440..c4f9167f 100644
--- a/pisi/index.py
+++ b/pisi/index.py
@@ -51,7 +51,9 @@ class Index(xmlfile.XmlFile):
 
     def read_uri(self, uri, tmpdir, force = False):
         return self.read(uri, tmpDir=tmpdir, sha1sum=not force,
-                  compress=pisi.file.File.auto, sign=pisi.file.File.detached, copylocal = True, nodecode = True)
+                         compress=pisi.file.File.COMPRESSION_TYPE_AUTO,
+                         sign=pisi.file.File.detached,
+                         copylocal=True, nodecode=True)
 
     # read index for a given repo, force means download even if remote not updated
     def read_uri_of_repo(self, uri, repo = None, force = False):