@@ -13,9 +13,8 @@ def main():
1313 filename = sys .argv [0 ]
1414 print ('Reading' , filename )
1515
16- f = open (filename , 'rb' ) # Get the data to compress
17- s = f .read ()
18- f .close ()
16+ with open (filename , 'rb' ) as f : # Get the data to compress
17+ s = f .read ()
1918
2019 # First, we'll compress the string in one step
2120 comptext = zlib .compress (s , 1 )
@@ -30,15 +29,15 @@ def main():
3029 chunk = 256
3130 compressor = zlib .compressobj (9 )
3231 decompressor = zlib .decompressobj ()
33- comptext = decomp = ''
32+ comptext = decomp = b ''
3433 for i in range (0 , len (s ), chunk ):
35- comptext = comptext + compressor .compress (s [i :i + chunk ])
34+ comptext += compressor .compress (s [i :i + chunk ])
3635 # Don't forget to call flush()!!
37- comptext = comptext + compressor .flush ()
36+ comptext += compressor .flush ()
3837
3938 for i in range (0 , len (comptext ), chunk ):
40- decomp = decomp + decompressor .decompress (comptext [i :i + chunk ])
41- decomp = decomp + decompressor .flush ()
39+ decomp += decompressor .decompress (comptext [i :i + chunk ])
40+ decomp += decompressor .flush ()
4241
4342 print ('Progressive compression (level 9):' )
4443 print (' Original:' , len (s ), 'Compressed:' , len (comptext ), end = ' ' )
0 commit comments