如何使用 ZipFile 模块进行文件压缩和解压？

Python的zipfile模块是处理ZIP文件的强大工具，它提供了创建、读取和解压缩ZIP文件的功能，本文将详细介绍如何使用这个模块，包括其核心类ZipFile和ZipInfo的基本操作方法。

一、zipfile模块

Python中的zipfile模块是标准库的一部分，专门用于处理ZIP文件，ZIP文件是一种常见的数据压缩和文档存储格式，广泛用于各种操作系统和应用中，zipfile模块提供了两个主要类：ZipFile和ZipInfo，通过这两个类，用户可以方便地进行ZIP文件的创建、读取和解压缩操作。

二、ZipFile类

ZipFile类是zipfile模块中最核心的类，负责创建和读取ZIP文件，以下是ZipFile类的详细使用方法：

1. 创建ZipFile对象

要创建一个ZipFile对象，需要指定文件路径或类文件对象（file-like object），以及打开模式，常用的模式有：

'r'：读取已经存在的ZIP文件。

'w'：新建一个ZIP文件或覆盖已有的ZIP文件。

'a'：追加数据到一个现存的ZIP文件。

import zipfile
以写模式创建一个新的ZIP文件
with zipfile.ZipFile('example.zip', 'w') as myzip:
    myzip.write('file1.txt')
    myzip.write('file2.txt')

2. 常用方法

write(filename, arcname=None, compress_type=None)：将指定文件写入ZIP文件中。filename是要添加的文件路径，arcname是文件在ZIP文件中的名称，compress_type指定压缩方法。

extract(member, path=None, pwd=None)：从ZIP文件中提取指定文件到当前目录。member是要提取的文件名称或对应的ZipInfo对象，path指定解压路径，pwd为解压密码。

extractall([path[, members[, pwd]]])：解压ZIP文件中的所有文件到当前目录。path指定解压路径，members是要解压的文件列表，pwd为解压密码。

getinfo(name)：获取ZIP文件中指定文件的信息，返回一个ZipInfo对象。

infolist()：获取ZIP文件中所有文件的信息，返回一个ZipInfo对象的列表。

namelist()：获取ZIP文件中所有文件的名称列表。

setpassword(pwd)：设置ZIP文件的密码。

printdir()：将ZIP文件内的信息打印到控制台上。

三、ZipInfo类

ZipInfo类用于存储ZIP文件中每个文件的信息，通常与ZipFile类结合使用，可以通过ZipFile类的getinfo()方法获取某个文件的ZipInfo对象。

1. 属性

filename：文件名。

date_time：最后修改时间，以元组形式返回（年、月、日、时、分、秒）。

compress_type：压缩类型。

comment：文档说明。

ext_attr：扩展属性数据。

create_system：创建该ZIP文档的系统。

create_version：创建ZIP文档的PKZIP版本。

extract_version：解压ZIP文档所需的PKZIP版本。

reserved：预留字段，总是返回0。

flag_bits：标志位。

四、示例代码

以下是一个综合示例，展示了如何使用zipfile模块进行ZIP文件的创建、读取和解压缩操作：

import zipfile
import os
创建目录结构
zip_dir = "zip_dir"
os.makedirs(zip_dir, exist_ok=True)
创建ZIP文件并添加文件
with zipfile.ZipFile("zip_dir/abc.zip", 'w') as zipf:
    zipf.write("file1.txt")
    zipf.write("file2.txt")
读取ZIP文件并列出其中的所有文件
with zipfile.ZipFile("zip_dir/abc.zip", 'r') as zipf:
    print("ZIP文件中的所有文件：")
    for file_name in zipf.namelist():
        print(file_name)
解压缩ZIP文件中的所有文件到指定目录
with zipfile.ZipFile("zip_dir/abc.zip", 'r') as zipf:
    zipf.extractall("zip_dir/extracted")

五、常见问题解答（FAQs）

Q1：如何设置ZIP文件的密码？

A1：可以使用ZipFile类的setpassword()方法设置ZIP文件的密码，注意，该方法需要在写入模式下调用。

with zipfile.ZipFile('protected.zip', 'w') as myzip:
    myzip.setpassword(b'mypassword')
    myzip.write('file1.txt')

Q2：如何处理大文件超过2GB的情况？

A2：如果需要处理超过2GB的大文件，可以在创建ZipFile对象时将allowZip64参数设置为True。

with zipfile.ZipFile('large_file.zip', 'w', allowZip64=True) as myzip:
    myzip.write('large_file.dat')

小编有话说

通过上述介绍，相信大家对Python的zipfile模块有了更全面的了解，无论是创建、读取还是解压缩ZIP文件，zipfile模块都提供了简洁而强大的接口，希望大家在实际编程中能够灵活运用这些功能，提高工作效率，如果你有任何疑问或建议，欢迎留言讨论！