Python ZIP 常见技术问题：如何使用 zipfile 模块压缩多个文件？

Question

**问题描述：** 在使用 Python 的 `zipfile` 模块进行文件压缩时，开发者常遇到的问题是如何将多个文件打包成一个 ZIP 压缩包。虽然 `zipfile` 提供了良好的接口支持，但对于初学者来说，如何正确创建 ZIP 文件、逐个添加多个文件并确保路径结构合理，仍是一个常见的技术难点。常见问题包括：添加文件时路径处理不当导致文件丢失、未正确关闭 ZIP 文件、以及未处理文件不存在的异常情况。本文将演示如何使用 `zipfile.ZipFile` 类，结合 `write()` 方法和上下文管理器，安全高效地压缩多个文件。

Answer 1

使用 Python 的 zipfile 模块高效打包多个文件

1. 基础概念：zipfile 模块简介

Python 标准库中的 zipfile 模块提供了创建、读取和提取 ZIP 文件的功能。其核心类为 ZipFile，通过该类可以实现对多个文件的打包操作。

常见使用场景包括：

• 将多个文件打包为一个 ZIP 文件进行分发
• 压缩日志文件以节省存储空间
• 构建自动化部署脚本时打包资源

2. 初级问题：如何正确创建 ZIP 文件

创建 ZIP 文件的基本方式是使用 ZipFile 构造函数，并指定压缩模式（如 'w' 表示写模式）。

示例代码如下：

import zipfile

with zipfile.ZipFile('example.zip', 'w') as zipf:
    zipf.write('file1.txt')
    zipf.write('file2.txt')

使用 with 上下文管理器能确保 ZIP 文件被正确关闭，避免资源泄漏。

3. 路径处理：保留文件结构还是仅保留文件名

在调用 write() 方法时，默认会将文件的完整路径也打包进 ZIP 中，这可能导致目标 ZIP 文件中包含不必要的目录结构。

方式	说明
zipf.write('docs/file.txt')	ZIP 中包含 docs/file.txt
zipf.write('docs/file.txt', 'file.txt')	ZIP 中仅包含 file.txt

4. 异常处理：如何应对文件不存在的情况

在打包过程中，若某个文件不存在，程序会抛出异常。建议使用 try-except 块进行捕获，确保程序健壮性。

import os

files_to_zip = ['file1.txt', 'file2.txt', 'missing.txt']

with zipfile.ZipFile('safe_example.zip', 'w') as zipf:
    for f in files_to_zip:
        if os.path.exists(f):
            zipf.write(f)
        else:
            print(f"警告：文件 {f} 不存在，跳过")

5. 进阶技巧：批量压缩目录中的所有文件

使用 os.listdir() 或 os.walk() 可以遍历指定目录，将其中所有文件打包进 ZIP。

import os
import zipfile

def zip_directory(folder_path, output_zip):
    with zipfile.ZipFile(output_zip, 'w') as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                file_path = os.path.join(root, file)
                arcname = os.path.relpath(file_path, os.path.dirname(folder_path))
                zipf.write(file_path, arcname)

6. 性能优化：压缩级别与压缩算法

从 Python 3.7 开始，zipfile.ZipFile 支持指定压缩算法和压缩级别。例如使用 zipfile.ZIP_DEFLATED 提高压缩率。

with zipfile.ZipFile('compressed.zip', 'w', zipfile.ZIP_DEFLATED) as zipf:
    zipf.write('large_file.txt')

7. 流程图：打包多个文件的完整流程

graph TD
A[开始] --> B[创建 ZIP 文件]
B --> C{检查文件是否存在}
C -->|是| D[添加文件到 ZIP]
C -->|否| E[打印警告信息]
D --> F[继续处理下一个文件]
E --> F
F --> G{是否还有更多文件}
G -->|是| C
G -->|否| H[关闭 ZIP 文件]
H --> I[结束]