Python压缩文件夹/解压缩zip文件

#coding=utf-8

#甄码农python代码

#使用zipfile做目录压缩，解压缩功能

 

import os,os.path

import zipfile

 

def zip_dir(dirname,zipfilename):

    filelist = []

    if os.path.isfile(dirname):

        filelist.append(dirname)

    else :

        for root, dirs, files in os.walk(dirname):

            for name in files:

                filelist.append(os.path.join(root, name))

         

    zf = zipfile.ZipFile(zipfilename, "w", zipfile.zlib.DEFLATED)

    for tar in filelist:

        arcname = tar[len(dirname):]

        #print arcname

        zf.write(tar,arcname)

    zf.close()

 

 

def unzip_file(zipfilename, unziptodir):

    if not os.path.exists(unziptodir): os.mkdir(unziptodir, 0777)

    zfobj = zipfile.ZipFile(zipfilename)

    for name in zfobj.namelist():

        name = name.replace('\\','/')

        

        if name.endswith('/'):

            os.mkdir(os.path.join(unziptodir, name))

        else:           

            ext_filename = os.path.join(unziptodir, name)

            ext_dir= os.path.dirname(ext_filename)

            if not os.path.exists(ext_dir) : os.mkdir(ext_dir,0777)

            outfile = open(ext_filename, 'wb')

            outfile.write(zfobj.read(name))

            outfile.close()

 

if __name__ == '__main__':

    zip_dir(r'E:/python/learning',r'E:/python/learning/zip.zip')

    unzip_file(r'E:/python/learning/zip.zip',r'E:/python/learning2')