有时我们需要在Python中使用zip文件,而在1.6版中,Python就已经提供了zipfile模块可以进行这样的操作。不过Python中的zipfile模块不能处理多卷的情况,不过这种情况并不多见,因此在通常情况下已经足够使用了。下面我只是对一些基本的zipfile操作进行了记录,足以应付大部分的情况了。
zipfile模块可以让你打开或写入一个zip文件。比如:
import zipfile
z = zipfile.ZipFile('zipfilename', mode='r')
这样就打开了一个zip文件,如果mode为'w'或'a'则表示要写入一个zip文件。如果是写入,则还可以跟上第三个参数:
compression=zipfile.ZIP_DEFLATED或
compression=zipfile.ZIP_STORED ZIP_DEFLATED是压缩标志,如果使用它需要编译了zlib模块。而后一个只是用zip进行打包,并不压缩。
在打开了zip文件之后就可以根据需要是读出zip文件的内容还是将内容保存到zip文件中。
读出zip中的内容
很简单,zipfile对象提供了一个read(name)的方法。name为zip文件中的一个文件入口,执行完成之后,将返回读出的内容,你把它保存到想到的文件中即可。
写入zip文件
有两种方式,一种是直接写入一个已经存在的文件,另一种是写入一个字符串。
对于第一种使用zipfile对象的write(filename, arcname, compress_type),后两个参数是可以忽略的。第一个参数是文件名,第二个参数是表示在zip文件中的名字,如果没有给出,表示使用与filename一样的名字。compress_type是压缩标志,它可以覆盖创建zipfile时的参数。第二种是使用zipfile对象的writestr(zinfo_or_arcname, bytes),第一个参数是zipinfo对象或写到压缩文件中的压缩名,第二个参数是字符串。使用这个方法可以动态的组织文件的内容。
需要注意的是在读出时,因为只能读出内容,因此如果想实现按目录结构展开zip文件的话,这些操作需要自已来完成,比如创建目录,创建文件并写入。而写入时,则可以根据需要动态组织在zip文件中的目录结构,这样可以不按照原来的目录结构来生成zip文件。
于是我为了方便使用,创建了自已的一个ZFile类,主要是实现象winrar的右键菜单中的压缩到的功能--即将一个zip文件压缩到指定目录,自动创建相应的子目录。再有就是方便生成zip文件。类源码为:
#coding:cp936
#Zfile.py
#
importzipfile
importos.path
importos
classZFile(object):
def__init__(self,filename,mode='r',basedir=''):
self.filename=filename
self.mode=mode
ifself.modein('w','a'):
self.zfile=zipfile.ZipFile(filename,self.mode,compression=zipfile.ZIP_DEFLATED)
else:
self.zfile=zipfile.ZipFile(filename,self.mode)
self.basedir=basedir
ifnotself.basedir:
self.basedir=os.path.dirname(filename)
defaddfile(self,path,arcname=None):
path=path.replace('//','/')
ifnotarcname:
ifpath.startswith(self.basedir):
arcname=path[len(self.basedir):]
else:
arcname=''
self.zfile.write(path,arcname)
defaddfiles(self,paths):
forpathinpaths:
ifisinstance(path,tuple):
self.addfile(*path)
else:
self.addfile(path)
defclose(self):
self.zfile.close()
defextract_to(self,path):
forpinself.zfile.namelist():
self.extract(p,path)
defextract(self,filename,path):
ifnotfilename.endswith('/'):
f=os.path.join(path,filename)
dir=os.path.dirname(f)
ifnotos.path.exists(dir):
os.makedirs(dir)
file(f,'wb').write(self.zfile.read(filename))
defcreate(zfile,files):
z=ZFile(zfile,'w')
z.addfiles(files)
z.close()
defextract(zfile,path):
z=ZFile(zfile)
z.extract_to(path)
z.close()