浏览代码

Add helper function read_zlib_chunks.

Jelmer Vernooij 16 年之前
父节点
当前提交
b13ec06edb
共有 1 个文件被更改,包括 7 次插入2 次删除
  1. 7 2
      dulwich/pack.py

+ 7 - 2
dulwich/pack.py

@@ -73,7 +73,7 @@ def take_msb_bytes(map, offset):
     return ret
     return ret
 
 
 
 
-def read_zlib(data, offset, dec_size):
+def read_zlib_chunks(data, offset, dec_size):
     obj = zlib.decompressobj()
     obj = zlib.decompressobj()
     ret = []
     ret = []
     fed = 0
     fed = 0
@@ -84,9 +84,14 @@ def read_zlib(data, offset, dec_size):
             add += "Z"
             add += "Z"
         fed += len(add)
         fed += len(add)
         ret.append(obj.decompress(add))
         ret.append(obj.decompress(add))
+    comp_len = fed-len(obj.unused_data)
+    return ret, comp_len
+
+
+def read_zlib(data, offset, dec_size):
+    ret, comp_len = read_zlib_chunks(data, offset, dec_size)
     x = "".join(ret)
     x = "".join(ret)
     assert len(x) == dec_size
     assert len(x) == dec_size
-    comp_len = fed-len(obj.unused_data)
     return x, comp_len
     return x, comp_len