]> granicus.if.org Git - python/commitdiff
Another demo that at least runs again.
authorGeorg Brandl <georg@python.org>
Mon, 2 Aug 2010 23:15:58 +0000 (23:15 +0000)
committerGeorg Brandl <georg@python.org>
Mon, 2 Aug 2010 23:15:58 +0000 (23:15 +0000)
Demo/zlib/zlibdemo.py

index 52f71a097c8cc35c77e976bb8ac8a0562be485f0..40a36d4023b2064ca6ca87ad3a84e333aef9f59e 100755 (executable)
@@ -13,9 +13,8 @@ def main():
         filename = sys.argv[0]
     print('Reading', filename)
 
-    f = open(filename, 'rb')           # Get the data to compress
-    s = f.read()
-    f.close()
+    with open(filename, 'rb') as f:           # Get the data to compress
+        s = f.read()
 
     # First, we'll compress the string in one step
     comptext = zlib.compress(s, 1)
@@ -30,15 +29,15 @@ def main():
     chunk = 256
     compressor = zlib.compressobj(9)
     decompressor = zlib.decompressobj()
-    comptext = decomp = ''
+    comptext = decomp = b''
     for i in range(0, len(s), chunk):
-        comptext = comptext+compressor.compress(s[i:i+chunk])
+        comptext += compressor.compress(s[i:i+chunk])
     # Don't forget to call flush()!!
-    comptext = comptext + compressor.flush()
+    comptext += compressor.flush()
 
     for i in range(0, len(comptext), chunk):
-        decomp = decomp + decompressor.decompress(comptext[i:i+chunk])
-    decomp=decomp+decompressor.flush()
+        decomp += decompressor.decompress(comptext[i:i+chunk])
+    decomp += decompressor.flush()
 
     print('Progressive compression (level 9):')
     print('    Original:', len(s), 'Compressed:', len(comptext), end=' ')