python压缩与解压缩ZIP文件(zipfile模块)

发布时间:2020-12-12编辑:脚本学堂
本文介绍下,python zipfile模块实现文件的压缩与解压缩的例子,学习下python zipfile模块的用法,有需要的朋友参考下。

介绍下pythonmokuai/ target=_blank class=infotextkey>python模块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 文件。

ZFile类源码:
 

复制代码 代码示例:
# coding:cp936 
# Zfile.py 
# jb200.com 
import zipfile  
import os.path  
import os  
  
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)  
            file(f, 'wb').write(self.zfile.read(filename))  
             
         
def create(zfile, files):  
    z = ZFile(zfile, 'w')  
    z.addfiles(files)  
    z.close()  
     
def extract(zfile, path):  
    z = ZFile(zfile)  
    z.extract_to(path)  
    z.close()