python gzip模块实现文件压缩的方法

pythongzip模块实现文件压缩的方法

使用gzip格式压缩文件，注意引入gzip包。

代码：

复制代码代码示例:

#!/bin/python

#

#site: www.jbxue.com

import string

import gzip

from optparse import OptionParser

def readCommandLine():

parser = OptionParser()

#read the options in

parser.add_option("-f","--Full_file_location",

dest="File_to_be_run",

default=r"c:\tn.txt",

help="This is the fully qualified path name to the file location")

parser.add_option("-m","--Mode",

dest="modeTn",

default="r",

help="The mode of zip unzip")

parser.add_option("-c","--Compression",

dest="compress",

default=9,

help="The level of compression")

options, args = parser.parse_args()

#print options

return options

def zipit(filename, mode,compress):

#Saves/Zipps a compressed file to disk

#

r_file = open(filename, 'r')

# this is the zipping bit

w_file = gzip.GzipFile(filename + '.gz', mode, compress)

w_file.write(r_file.read())

w_file.flush()

w_file.close()

r_file.close()

def un_zipit(filename,mode):

#Unzips a compressed file from disk

#

#this is the unzipping bit

r_file = gzip.GzipFile(filename, mode)

write_file = string.rstrip(filename, '.gz')

w_file = open(write_file, 'w')

w_file.write(r_file.read())

w_file.close()

r_file.close()

if __name__ == "__main__":

#first thing to do is read the options in

options = readCommandLine()

if options.modeTn == "r":

#unzippit mode

if options.File_to_be_run[-3:] != '.gz':

# check to see if it has the extension .gz

print "This " + options.File_to_be_run + " is not a .gz file"

else:

#This should now unzipit

un_zipit(options.File_to_be_run,options.modeTn)

elif options.modeTn== "wb":

#this should zipit

zipit(options.File_to_be_run,options.modeTn,options.compress)

else:

# basically the wrong option was passed

print "ABORT something went wrong"

sys.exit()