Python 多个文件添加到zip包|极客笔记

Python 多个文件添加到zip包

在编程工作中，我们经常需要将多个文件打包成zip包进行传输或存储。Python提供了一个内置的zipfile模块，可以方便地实现这一功能。在本文中，我们将学习如何使用Python将多个文件添加到一个zip包中。

准备工作

在开始之前，我们需要创建一些示例文件以供打包。我们可以在Python脚本中创建这些文件，也可以手动创建。在本示例中，我们将创建三个文本文件来进行演示。

# 创建示例文件
file1 = open("file1.txt", "w")
file1.write("This is file 1.")
file1.close()

file2 = open("file2.txt", "w")
file2.write("This is file 2.")
file2.close()

file3 = open("file3.txt", "w")
file3.write("This is file 3.")
file3.close()

添加文件到zip包

下面我们将演示如何将这三个文件添加到一个zip包中。首先，我们需要导入zipfile模块。

import zipfile

接下来，我们创建一个ZipFile对象，并将示例文件添加到这个zip包中。

# 创建一个名为example.zip的zip包
with zipfile.ZipFile('example.zip', 'w') as zipf:
    # 添加示例文件到zip包中
    zipf.write('file1.txt')
    zipf.write('file2.txt')
    zipf.write('file3.txt')

运行以上代码后，将会在当前目录下生成一个名为example.zip的zip包，并将file1.txt、file2.txt以及file3.txt添加到其中。

添加文件夹到zip包

除了单个文件外，我们也可以将整个文件夹添加到zip包中。下面示例演示了如何将一个名为example_folder的文件夹添加到一个名为example_folder.zip的zip包中。

import os

# 创建一个文件夹example_folder，并在其中创建示例文件
os.mkdir('example_folder')
file4 = open("example_folder/file4.txt", "w")
file4.write("This is file 4.")
file4.close()

# 创建一个名为example_folder.zip的zip包
with zipfile.ZipFile('example_folder.zip', 'w') as zipf:
    # 将整个文件夹添加到zip包中
    for foldername, subfolders, filenames in os.walk('example_folder'):
        for filename in filenames:
            zipf.write(os.path.join(foldername, filename))

运行以上代码后，将会在当前目录下生成一个名为example_folder.zip的zip包，并将包含在文件夹example_folder中的所有文件添加到其中。