您的位置:

Python Zipfile模块:如何压缩和解压文件

在日常的软件开发过程中,文件的压缩和解压是我们经常遇到的任务。在Python中,使用Zipfile模块来完成文件的压缩和解压非常方便。在本文中,我们将从以下几个方面详细介绍Zipfile模块的使用方法:

一、压缩单个文件

我们首先看一下如何使用Zipfile模块来压缩单个文件。Zipfile模块提供了一个ZipFile类,我们可以通过实例化一个ZipFile类来进行文件的压缩。下面是一个示例代码:

import zipfile
 
def zip_file(file_path):
    zf = zipfile.ZipFile(file_path + '.zip', mode='w', allowZip64=True)
    try:
        zf.write(file_path, compress_type=zipfile.ZIP_DEFLATED)
    finally:
        zf.close()

在上面的代码中,我们使用Zipfile模块的ZipFile类创建了一个zip文件对象,并指定了文件的压缩模式(mode)。使用write()方法来将文件添加到压缩文件中,并指定压缩的方式(compress_type)。

二、压缩多个文件

除了压缩单个文件之外,Zipfile模块还提供了压缩多个文件的方法。我们可以使用ZipFile类的write()方法,每次传入一个文件路径即可实现多个文件的压缩。下面是一个示例代码:

import os
import zipfile
  
def zip_dir(dirname, zipfilename):
    filelists = []
  
    if os.path.isfile(dirname):
        filelists.append(dirname)
    else:
        for root, dirs, files in os.walk(dirname):
            for name in files:
                filelists.append(os.path.join(root, name))
  
    zf = zipfile.ZipFile(zipfilename, "w", zipfile.ZIP_DEFLATED)
    for tar in filelists:
        arcname = tar[len(dirname):]
        zf.write(tar, arcname)
    zf.close()

在上面的代码中,我们使用os.walk()方法遍历目录下所有的文件,然后使用ZipFile类的write()方法将文件添加到压缩文件中。由于ZipFile类支持压缩目录,我们在压缩时需要指定文件的根目录。

三、解压缩文件

如果我们需要对已经压缩的文件进行解压,同样可以使用Zipfile模块。Zipfile模块提供了ZipFile类的extract()和extractall()方法,可以用来分别解压缩单个文件和压缩目录下的所有文件。下面是一个示例代码:

import zipfile
 
def unzip_file(zip_file_path, unzip_path):
    zf = zipfile.ZipFile(zip_file_path)
    try:
        zf.extractall(path=unzip_path)
    except Exception as e:
        print(e)
    finally:
        zf.close()

在上面的代码中,我们使用ZipFile类的extractall()方法解压缩文件到指定的路径。在解压缩时,我们需要注意文件的编码以及路径的长度等问题,以避免解压缩失败。

四、压缩文件时保留文件权限

在压缩文件时,有时我们需要保留原有的文件权限。Zipfile模块提供了ZipFile类的writepy()方法,可以在压缩文件时保留文件权限。下面是一个示例代码:

import os
import zipfile
  
def zip_dir_with_permission(dirname, zipfilename):
    filelists = []
  
    if os.path.isfile(dirname):
        filelists.append(dirname)
    else:
        for root, dirs, files in os.walk(dirname):
            for name in files:
                filelists.append(os.path.join(root, name))
  
    zf = zipfile.ZipFile(zipfilename, "w", zipfile.ZIP_DEFLATED, True)
    for tar in filelists:
        arcname = tar[len(dirname):]
        zf.write(tar, arcname)
    zf.close()

在上面的代码中,我们在创建ZipFile对象时,将保留权限参数(True)传给ZipFile类的构造函数。在每次向压缩文件中添加文件时,ZipFile类都会保留原有的文件权限。

五、结语

通过本文的介绍,相信大家已经掌握了使用Python ZipFile模块进行文件压缩和解压缩的基本技巧。如果大家还有什么问题或者疑惑,欢迎在评论区留言,我们一起学习探讨。