Python压缩模块zipfile详解|极客教程

Python压缩模块zipfile详解

1. 简介

在进行文件处理和数据传输时，经常会涉及到文件的压缩和解压操作。Python提供了zipfile模块来进行ZIP格式文件的压缩和解压，可以方便地进行文件的打包和解包操作。本文将详细讲解zipfile模块的使用方法。

2. 压缩文件

下面我们将演示如何使用zipfile模块来压缩文件。

2.1 压缩单个文件

首先，我们需要导入zipfile模块，然后使用ZipFile来创建一个ZIP文件对象。通过write方法来向ZIP文件中添加文件，并最后关闭ZIP文件对象。

import zipfile

# 创建一个ZIP文件对象
with zipfile.ZipFile('sample.zip', 'w') as zip_file:
    # 向ZIP文件中添加文件
    zip_file.write('sample.txt')

运行以上代码后，会在当前目录下生成一个sample.zip的压缩文件，其中包含了sample.txt文件。

2.2 压缩文件夹

除了压缩单个文件外，我们还可以压缩整个文件夹。

import zipfile
import os

def compress_folder(folder_path, zip_path):
    with zipfile.ZipFile(zip_path, 'w') as zip_file:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                file_path = os.path.join(root, file)
                zip_file.write(file_path)

compress_folder('data', 'data.zip')

在上面的示例中，我们定义了一个compress_folder函数来压缩指定文件夹中的所有文件。通过os.walk来遍历文件夹中的所有文件，并将文件逐个添加到ZIP文件中。

3. 解压文件

接下来，我们将演示如何使用zipfile模块来解压ZIP文件。

3.1 解压文件

解压文件比较简单，只需要使用ZipFile对象的extractall方法即可。

import zipfile

# 创建一个ZIP文件对象
with zipfile.ZipFile('sample.zip', 'r') as zip_file:
    # 解压ZIP文件
    zip_file.extractall('extracted_files')

运行以上代码后，会将sample.zip文件解压到extracted_files文件夹中。

4. 查看ZIP文件内容

我们也可以通过namelist方法来查看ZIP文件中包含的文件列表。

import zipfile

# 创建一个ZIP文件对象
with zipfile.ZipFile('sample.zip', 'r') as zip_file:
    # 获取ZIP文件中的文件列表
    file_list = zip_file.namelist()
    print(file_list)

5. 压缩文件时设置密码

有时候我们需要对ZIP文件进行加密，可以通过setpassword方法来设置密码。

import zipfile

# 创建一个ZIP文件对象
with zipfile.ZipFile('secure.zip', 'w') as zip_file:
    # 设置密码
    zip_file.setpassword(b'password')
    zip_file.write('sample.txt')