mirror of
https://github.com/python/cpython.git
synced 2025-12-04 16:43:27 +00:00
Put code in a main() function; loosen up the spacing to match current code style
This commit is contained in:
parent
3725dea9c3
commit
3550613502
1 changed files with 39 additions and 26 deletions
|
|
@ -1,35 +1,48 @@
|
||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
# Takes an optional filename, defaulting to this file itself.
|
||||||
|
# Reads the file and compresses the content using level 1 and level 9
|
||||||
|
# compression, printing a summary of the results.
|
||||||
|
|
||||||
import zlib, sys
|
import zlib, sys
|
||||||
if len(sys.argv)>1: filename=sys.argv[1]
|
|
||||||
else: filename='zlibdemo.py'
|
|
||||||
print 'Reading', filename
|
|
||||||
f=open(filename, 'r') # Get the data to compress
|
|
||||||
s=f.read()
|
|
||||||
f.close()
|
|
||||||
|
|
||||||
# First, we'll compress the string in one step
|
def main():
|
||||||
comptext=zlib.compress(s, 1)
|
if len(sys.argv) > 1:
|
||||||
decomp=zlib.decompress(comptext)
|
filename = sys.argv[1]
|
||||||
|
else:
|
||||||
|
filename = sys.argv[0]
|
||||||
|
print 'Reading', filename
|
||||||
|
|
||||||
print '1-step compression: (level 1)'
|
f = open(filename, 'rb') # Get the data to compress
|
||||||
print ' Original:', len(s), 'Compressed:', len(comptext),
|
s = f.read()
|
||||||
print 'Uncompressed:', len(decomp)
|
f.close()
|
||||||
|
|
||||||
# Now, let's compress the string in stages; set chunk to work in smaller steps
|
# First, we'll compress the string in one step
|
||||||
|
comptext = zlib.compress(s, 1)
|
||||||
|
decomp = zlib.decompress(comptext)
|
||||||
|
|
||||||
chunk=256
|
print '1-step compression: (level 1)'
|
||||||
compressor=zlib.compressobj(9)
|
print ' Original:', len(s), 'Compressed:', len(comptext),
|
||||||
decompressor=zlib.decompressobj()
|
print 'Uncompressed:', len(decomp)
|
||||||
comptext=decomp=''
|
|
||||||
for i in range(0, len(s), chunk):
|
|
||||||
comptext=comptext+compressor.compress(s[i:i+chunk])
|
|
||||||
comptext=comptext+compressor.flush() # Don't forget to call flush()!!
|
|
||||||
|
|
||||||
for i in range(0, len(comptext), chunk):
|
# Now, let's compress the string in stages; set chunk to work in smaller steps
|
||||||
decomp=decomp+decompressor.decompress(comptext[i:i+chunk])
|
|
||||||
decomp=decomp+decompressor.flush()
|
|
||||||
|
|
||||||
print 'Progressive compression (level 9):'
|
chunk = 256
|
||||||
print ' Original:', len(s), 'Compressed:', len(comptext),
|
compressor = zlib.compressobj(9)
|
||||||
print 'Uncompressed:', len(decomp)
|
decompressor = zlib.decompressobj()
|
||||||
|
comptext = decomp = ''
|
||||||
|
for i in range(0, len(s), chunk):
|
||||||
|
comptext = comptext+compressor.compress(s[i:i+chunk])
|
||||||
|
# Don't forget to call flush()!!
|
||||||
|
comptext = comptext + compressor.flush()
|
||||||
|
|
||||||
|
for i in range(0, len(comptext), chunk):
|
||||||
|
decomp = decomp + decompressor.decompress(comptext[i:i+chunk])
|
||||||
|
decomp=decomp+decompressor.flush()
|
||||||
|
|
||||||
|
print 'Progressive compression (level 9):'
|
||||||
|
print ' Original:', len(s), 'Compressed:', len(comptext),
|
||||||
|
print 'Uncompressed:', len(decomp)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue