Make zlib compression level a parameter for Client

Signed-off-by: Zoran Zaric <zz@zoranzaric.de>

diff --git a/lib/bup/client.py b/lib/bup/client.py
index 5563905569e5d489b2ef46ae166bf03a5de0824e..c2c83e2f0c42dd7ea28d5b41b99d5f4835030872 100644 (file)
--- a/lib/bup/client.py
+++ b/lib/bup/client.py
@@ -51,9 +51,10 @@ def parse_remote(remote):
 
 
 class Client:
-    def __init__(self, remote, create=False):
+    def __init__(self, remote, create=False, compression_level=1):
         self._busy = self.conn = None
         self.sock = self.p = self.pout = self.pin = None
+        self.compression_level = compression_level
         is_reverse = os.environ.get('BUP_SERVER_REVERSE')
         if is_reverse:
             assert(not remote)
@@ -245,7 +246,8 @@ class Client:
                                  suggest_packs = self._suggest_packs,
                                  onopen = _set_busy,
                                  onclose = self._not_busy,
-                                 ensure_busy = self.ensure_busy)
+                                 ensure_busy = self.ensure_busy,
+                                 compression_level = self.compression_level)
 
     def read_ref(self, refname):
         self.check_busy()
@@ -282,7 +284,8 @@ class Client:
 class PackWriter_Remote(git.PackWriter):
     def __init__(self, conn, objcache_maker, suggest_packs,
                  onopen, onclose,
-                 ensure_busy):
+                 ensure_busy,
+                 compression_level=1):
         git.PackWriter.__init__(self, objcache_maker)
         self.file = conn
         self.filename = 'remote socket'
@@ -293,6 +296,7 @@ class PackWriter_Remote(git.PackWriter):
         self._packopen = False
         self._bwcount = 0
         self._bwtime = time.time()
+        self.compression_level = compression_level
 
     def _open(self):
         if not self._packopen: