python 文件压缩和解压缩

python 文件压缩和解压缩

导入zipfile包
zipfile模块
这模块的操作方式基本上和文件的操作方式差不多。
例子：

import zipfile
z =zipfile.ZipFile(filename,'r')   #获取zip文件并且读取
for i inz.namelist():                 #namelist函数获取zip包

里面的文件名，z.namelist() 会返回压缩包内所有文 #件名的列表
print i
z.close() #关闭zip文件

zipfile.ZipFile(filename,mode,compression,allowZip64)
filename:zip文件名
mode:和文件的操作一样，r表示读，a表示打开zip文件王里面添加内容，w表示写（必要时就创建）。
compression：表示压缩格式，默认为ZIP_STORED表示只打包不开启压缩功能类似于tar，ZIP_DEFLATED 表示开启压缩（开启压缩必须指定）
allowZip64：表示是否支持64位压缩，默认值为False表示不支持64位压缩，压缩文件不能大于2G，如果压缩文件大于2G的话必须指定allowZip64=True
zipfile.close() ：关闭压缩文件

zipfile.write(filename[, arcname[,compress_type]])
acrname是压缩文件中该文件的名字，默认情况下和filename一样
compress_type的存在是因为zip文件允许被压缩的文件可以有不同的压缩类型。
解压缩
zipfile.extractall([path[, member[,password]]])
path解压缩目录，没什么可说的
member需要解压缩的文件名列表
password当zip文件有密码时需要该选项
对于简单的应用，这么多就够了。

zipfile.is_zipfile(filename)：判断一个文件是不是压缩文件

ZipFile.namelist()：返回文件列表

os.walk(file_dir)用来生成目录树，产生一个三维的Tuple（dirpath, dirnames, filenames）
文件的完整路径用do os.path.join(dirpath, name)

import os
import os.path
import zipfile


def gzip(zip_name, file_dir):
    filelist = []
    if os.path.isfile(file_dir):  #判断是否为文件
        filelist.append(file_dir)
    else: 
        for root, dirs, files in os.walk(file_dir):
            for file in files:
                filelist.append(os.path.join(root, file))

    zf = zipfile.ZipFile(zip_name, 'w', zipfile.zlib.DEFLATED)
    for tar in filelist:
        arcname = tar[len(file_dir):]
        zf.write(tar, arcname)
    zf.close()


def unzip(zip_name, unzip_dir):
    if not os.path.exists(unzip_dir):
        os.mkdir(unzip_dir)
    zfobj = zipfile.ZipFile(zip_name)
    for file_name in zfobj.namelist():
        file_name = file_name.replace('\\', '/')
        if file_name.endswith('/'):
            os.mkdir(os.path.join(unzip_dir, file_name))
        else:
            ext_filename = os.path.join(unzip_dir, file_name)
            ext_filedir = os.path.dirname(ext_filename)
            if not os.path.exists(ext_filedir):
                os.mkdir(ext_filedir)
            data = zfobj.read(file_name)
            with open(ext_filename, 'w') as f:
                f.write(data)
    zfobj.close()

if __name__ == '__main__':
    #gzip('E:\python\compress\zip.zip','E:\python\compress')
    unzip('E:\python\decompress\zip.zip','E:\python\decompress')