一个修改过的python zifile的包装模块，支持对目录压缩（包括空的文件夹），解压缩会解压缩空目录

'''

Created on 2010-5-26

@author: sean_long

'''

import os;

import zipfile

import os.path

class ZFile(object):

def __init__(self, filename, mode='r', basedir=''):

self.filename = filename

self.mode = mode

if self.mode in ('w', 'a'):

self.zfile = zipfile.ZipFile(filename, self.mode, compression=zipfile.ZIP_DEFLATED)

else:

self.zfile = zipfile.ZipFile(filename, self.mode)

self.basedir = basedir

if not self.basedir:

self.basedir = os.path.dirname(filename)

def addfile(self, path, arcname=None):

path = path.replace('//', '/')

if not arcname:

if path.startswith(self.basedir):

arcname = path[len(self.basedir):]

else:

arcname = '';

self.zfile.write(path, arcname)

def addfiles(self, paths):

for path in paths:

if isinstance(path, tuple):

self.addfile(*path)

else:

self.addfile(path)

def close(self):

self.zfile.close();

def extract_to(self, path):

for p in self.zfile.namelist():

self.extract(p, path)

def extract(self, filename, path):

if not filename.endswith('/'):

f = os.path.join(path, filename)

dir = os.path.dirname(f)

if not os.path.exists(dir):

os.makedirs(dir)

fd=file(f, 'wb');

fd.write(self.zfile.read(filename));

fd.close();

else:

d=os.path.join(path,filename);

if not os.path.exists(d):

os.makedirs(d);

def zipFolder(z,folderPath,basePath):

ar=os.listdir(folderPath);

for item in ar:

abtPath=os.path.join(folderPath,item);

bsPath=os.path.join(basePath,item);

if(os.path.isdir(abtPath)):

if(os.listdir(abtPath)==[]):

zif=zipfile.ZipInfo(bsPath+"/");

z.writestr(zif,"");

else:

zipFolder(z,abtPath,bsPath)

else:

z.write(abtPath,bsPath);

def createfolderZip(filePath,folderPath):

z=zipfile.ZipFile(filePath,"w",zipfile.ZIP_DEFLATED);

basePath=os.path.basename(folderPath);

zipFolder(z,folderPath,basePath);

z.close();

def extractZip(zfile, path):

z = ZFile(zfile)

z.extract_to(path)

z.close()