如何使用 Python 创建目录的 Zip 存档

在日常工作中，经常需要将一些文件或目录打包成 Zip 存档进行传输或备份。Python 作为一门强大的编程语言，可以方便地实现这一功能。本文将介绍如何使用 Python 创建包含目录的 Zip 存档，以及相关注意事项。

一、创建 Zip 存档

Python 的标准库中提供了 zipfile 模块，可以用于创建和解压 Zip 存档。下面是一个简单的示例，演示如何使用 zipfile 模块创建一个空的 Zip 存档：

import zipfile

with zipfile.ZipFile('example.zip', 'w') as zip_file:
    pass

这段代码中，我们使用 with 语句创建了一个 ZipFile 对象，并指定了存档文件的名称和模式。模式参数 ‘w’ 表示创建一个新的 Zip 存档，如果存档文件已经存在，则会被覆盖。

在 with 语句块中，我们没有向 Zip 存档中添加任何文件，而是直接执行了一个 pass 语句。这是因为在创建一个空的 Zip 存档时，我们不需要添加任何文件。

如果要向 Zip 存档中添加文件，可以使用 ZipFile 对象的 write() 方法。下面是一个示例，演示如何将一个文件添加到 Zip 存档中：

import zipfile

with zipfile.ZipFile('example.zip', 'w') as zip_file:
    zip_file.write('file.txt')

这段代码中，我们使用 with 语句创建了一个 ZipFile 对象，并指定了存档文件的名称和模式。在 with 语句块中，我们调用了 ZipFile 对象的 write() 方法，将文件 ‘file.txt’ 添加到 Zip 存档中。

如果要向 Zip 存档中添加目录，可以使用 ZipFile 对象的 write() 方法，并指定目录名称和 arcname 参数。下面是一个示例，演示如何将一个目录添加到 Zip 存档中：

import zipfile

with zipfile.ZipFile('example.zip', 'w') as zip_file:
    zip_file.write('dir', arcname='dir')

这段代码中，我们使用 with 语句创建了一个 ZipFile 对象，并指定了存档文件的名称和模式。在 with 语句块中，我们调用了 ZipFile 对象的 write() 方法，将目录 ‘dir’ 添加到 Zip 存档中，并指定了 arcname 参数为 ‘dir’，表示在 Zip 存档中的名称为 ‘dir’。

二、创建包含目录的 Zip 存档

在上面的示例中，我们演示了如何将一个目录添加到 Zip 存档中。但是，如果这个目录下还包含子目录和文件，我们该怎么办呢？

我们可以使用 os 模块的 walk() 函数来遍历一个目录下的所有文件和子目录。下面是一个示例，演示如何遍历一个目录下的所有文件和子目录：

import os

for root, dirs, files in os.walk('dir'):
    print(root, dirs, files)

这段代码中，我们调用了 os 模块的 walk() 函数，传入目录名称 ‘dir’。walk() 函数返回一个生成器对象，可以用于遍历目录下的所有文件和子目录。在遍历过程中，我们可以获取到当前目录的路径、子目录列表和文件列表。

有了遍历目录的功能，我们就可以将一个目录下的所有文件和子目录添加到 Zip 存档中。下面是一个示例，演示如何创建一个包含目录的 Zip 存档：

import os
import zipfile

def add_to_zip(zip_file, path):
    for root, dirs, files in os.walk(path):
        for file in files:
            zip_file.write(os.path.join(root, file), arcname=os.path.join(root[len(path)+1:], file))
        for dir in dirs:
            add_to_zip(zip_file, os.path.join(root, dir))

with zipfile.ZipFile('example.zip', 'w') as zip_file:
    add_to_zip(zip_file, 'dir')

这段代码中，我们定义了一个名为 add_to_zip() 的函数，用于将一个目录下的所有文件和子目录添加到 Zip 存档中。在函数内部，我们使用 os 模块的 walk() 函数遍历目录下的所有文件和子目录。对于每个文件，我们调用 ZipFile 对象的 write() 方法将其添加到 Zip 存档中，并指定 arcname 参数为相对路径。对于每个子目录，我们递归调用 add_to_zip() 函数，将其下的所有文件和子目录添加到 Zip 存档中。

在主程序中，我们使用 with 语句创建了一个 ZipFile 对象，并指定了存档文件的名称和模式。然后，我们调用 add_to_zip() 函数，将目录 ‘dir’ 下的所有文件和子目录添加到 Zip 存档中。

三、注意事项

在使用 Python 创建 Zip 存档时，需要注意以下几点：

1. Zip 存档中的文件路径应该使用正斜杠 / 分隔，而不是反斜杠 \。这是因为在不同的操作系统中，路径分隔符不同，使用正斜杠可以保证在任何系统中都能正确解析路径。
2. 在将目录添加到 Zip 存档中时，应该指定 arcname 参数，以保证在解压缩时能够正确还原目录结构。
3. 如果 Zip 存档中已经存在同名文件，使用 ‘w’ 模式创建 ZipFile 对象时会覆盖原有文件。如果想要将新的文件添加到 Zip 存档中而不覆盖原有文件，可以使用 ‘a’ 模式创建 ZipFile 对象。
4. 在使用 with 语句创建 ZipFile 对象时，不需要调用 close() 方法，因为 with 语句会自动调用 ZipFile 对象的 close() 方法。
5. 在使用 ZipFile 对象的 write() 方法添加文件或目录时，应该使用绝对路径或相对路径，而不是使用相对于当前工作目录的路径。可以使用 os 模块的 abspath() 函数将相对路径转换为绝对路径。
6. 在使用 ZipFile 对象的 write() 方法添加文件或目录时，应该使用 utf-8 编码，以保证在不同的操作系统中都能正确解析文件名。可以使用 encode(‘utf-8’) 方法将字符串转换为 utf-8 编码。

综上所述，使用 Python 创建包含目录的 Zip 存档并不复杂，只需要掌握一些基本的知识和技巧即可。在实际应用中，我们可以根据需要对代码进行优化和封装，以提高程序的可读性和可维护性。

火焰兔

海报 链接