Python zipfile模块详解

Python zipfile模块详解

介绍

在Python中，我们可以使用zipfile模块来操作ZIP文件。ZIP文件是一种常见的压缩文件格式，可以将多个文件或文件夹打包成一个文件，便于传输和存储。

zipfile模块提供了一种简单而有效的方式来创建、读取和解压ZIP文件。在本文中，我们将详细介绍zipfile模块的使用方法，并给出一些示例代码来帮助你更好地理解。

基本用法

首先，我们需要导入zipfile模块：

import zipfile

创建ZIP文件

要创建一个ZIP文件，可以使用ZipFile类的构造函数，并指定文件名以及打开方式（写入或追加）：

# 创建一个名为"example.zip"的ZIP文件
with zipfile.ZipFile('example.zip', 'w') as myzip:
    # 将文件夹"folder1"添加到ZIP文件中
    myzip.write('folder1')
    # 将文件"file1.txt"添加到ZIP文件中，并指定保存时的名称为"newfile.txt"
    myzip.write('file1.txt', 'newfile.txt')

读取ZIP文件

要读取一个ZIP文件，可以使用ZipFile类的构造函数，并传入文件名以及打开方式（只读）：

# 打开名为"example.zip"的ZIP文件
with zipfile.ZipFile('example.zip', 'r') as myzip:
    # 获取ZIP文件中的所有文件名
    file_list = myzip.namelist()
    print(file_list)
    # 解压文件"newfile.txt"到当前目录
    myzip.extract('newfile.txt')

解压ZIP文件

要解压一个ZIP文件，可以使用extractall()方法：

# 解压名为"example.zip"的ZIP文件到当前目录
with zipfile.ZipFile('example.zip', 'r') as myzip:
    myzip.extractall()

高级用法

除了上述基本用法之外，zipfile模块还提供了一些高级的功能，例如设置密码、压缩级别等。

设置密码

要在创建ZIP文件时设置密码，可以在ZipFile类的构造函数中指定密码，并使用setpassword()方法设置文件密码：

# 创建一个名为"example_with_password.zip"的加密ZIP文件
with zipfile.ZipFile('example_with_password.zip', 'w') as myzip:
    myzip.setpassword(b'password')  # 设置密码为"password"
    myzip.write('file2.txt')

压缩级别

要指定压缩级别，可以在ZipFile类的构造函数中传入compression参数，并指定为zipfile.ZIP_DEFLATED：

# 创建一个名为"example_compressed.zip"的压缩ZIP文件
with zipfile.ZipFile('example_compressed.zip', 'w', compression=zipfile.ZIP_DEFLATED) as myzip:
    myzip.write('file3.txt')

压缩文件夹

要压缩整个文件夹，可以使用write()方法，并传入文件夹路径以及arcname参数指定压缩时的路径：

# 创建一个名为"folder1.zip"的ZIP文件，包含整个文件夹"folder1"
with zipfile.ZipFile('folder1.zip', 'w') as myzip:
    myzip.write('folder1', arcname='folder1')

总结

本文介绍了zipfile模块的基本用法和高级功能，包括创建ZIP文件、读取ZIP文件、解压ZIP文件、设置密码、压缩级别和压缩文件夹等。通过学习这些内容，我们可以更好地使用Python处理ZIP文件，在实际项目中更加高效地操作和管理文件。