zipfile的重新封装

# -*- coding: utf-8 -*-

import zipfile, os

class ZFile(zipfile.ZipFile):
    def __init__(self, file, mode='r', compression=0):
        zipfile.ZipFile.__init__(self, file, mode, compression)

    def extract(self, dir=''):
        for f in self.namelist():
            path = os.path.join(os.path.abspath(dir), f)
            if not os.path.exists(os.path.dirname(path)):
                os.mkdir(os.path.dirname(path))
            if not path.endswith('/'):
                file(path, 'wb').write(self.read(f))

    def writedir(self, dir):
        dir = os.path.abspath(dir)
        empty = []      #the list of the path without file in
        for root, dirs, files in os.walk(dir):
            if not (files and dirs):
                zipinfo = zipfile.ZipInfo(root.replace(dir+os.sep, '')+os.sep)
                empty.append(zipinfo)
            for f in files:
                path = os.path.join(root, f)
                print "compressing",path
                self.write(path, path.replace(dir+os.sep, ''))
        for f in empty:
            print "compressing",f.filename
            self.writestr(f, '')

测试代码：

z = ZFile('other.zip', 'w', zipfile.ZIP_DEFLATED)
z.writedir('other')
z.close()

other是个目录，里面还有文件、子目录和空的子目录。执行后，压缩包维持了原本的目录结构。