Python压缩文件夹为zip

在工作中，我们经常需要将一些比较大的文件夹打包成zip文件来进行传输和存储，如何通过Python实现这个功能呢？本篇文章将介绍Python如何压缩文件夹为zip文件，并提供示例代码，让大家可以轻松快捷地实现这个功能。

确定要打包的文件夹

在实现Python压缩文件夹为zip文件之前，首先要确定好需要打包的文件夹。我们可以通过Python内置的os库来访问文件夹中的文件，并使用os.path模块来获取文件夹的路径。

以下是一个例子，代码会遍历整个文件夹，并输出所有文件的完整路径：

import os

def get_files_in_folder(folder_path):
    for root, dirs, files in os.walk(folder_path):
        for file_name in files:
            file_path = os.path.join(root, file_name)
            print(file_path)

上述代码中，os.walk函数会遍历指定的文件夹，返回一个三元组(root, dirs, files)，其中：

root是一个字符串，表示正在遍历的文件夹的路径
dirs是一个列表，包含所有在这个文件夹中的子文件夹
files是一个列表，包含所有在这个文件夹中的文件

在遍历的过程中，我们可以使用os.path.join函数将root和file_name拼接成完整的路径。

假设我们要打包的文件夹路径为folder_path = "/Users/xxx/Downloads/test"，使用get_files_in_folder(folder_path)函数可以输出所有文件的完整路径。

使用zipfile库进行压缩

Python内置的zipfile库可以帮助我们实现将文件夹打包为zip文件的功能。我们需要使用ZipFile类的实例来创建一个zip文件，并通过调用write函数将需要压缩的文件添加到zip文件中。

以下是示例代码，通过调用compress_folder_to_zip函数，可以将指定路径的文件夹压缩为zip文件，并保存到指定的目录下：

import zipfile
import os

def compress_folder_to_zip(folder_path, zip_file_path):
    with zipfile.ZipFile(zip_file_path, "w", zipfile.ZIP_DEFLATED) as zip_file:
        for root, dirs, files in os.walk(folder_path):
            for file_name in files:
                file_path = os.path.join(root, file_name)
                zip_file.write(file_path, os.path.relpath(file_path, folder_path))

在上述代码中，我们使用了with语句来创建一个ZipFile实例。通过指定”w”模式，我们可以创建一个新的zip文件。ZIP_DEFLATED参数表示使用zip的默认压缩算法进行压缩。

在遍历文件夹时，我们通过调用zip_file.write函数，将文件添加到zip文件中。write函数的第一个参数表示需要添加到zip文件中的文件的完整路径，第二个参数表示文件在zip文件中的相对路径。使用os.path.relpath函数可以计算出相对路径。

假设需要将/Users/xxx/Downloads/test文件夹压缩为/Users/xxx/Downloads/test.zip文件，使用compress_folder_to_zip函数即可实现这个功能：

compress_folder_to_zip("/Users/xxx/Downloads/test", "/Users/xxx/Downloads/test.zip")

测试压缩文件夹为zip功能

为了测试压缩文件夹为zip的功能是否正常，我们可以使用以下代码来解压zip文件并检查内容是否正确：

import zipfile

def verify_zip_file(zip_file_path, folder_path):
    with zipfile.ZipFile(zip_file_path, "r") as zip_file:
        for file_info in zip_file.infolist():
            extracted_file_path = zip_file.extract(file_info, folder_path)
            print("extractedfile: {}".format(extracted_file_path))

上述代码中，我们使用ZipFile类的实例来解压zip文件，并将文件解压到指定的文件夹中。infolist函数会返回zip文件中的所有文件信息，包括文件名、压缩前大小、压缩后大小等。在解压时，我们调用extract函数将文件解析并保存到指定的文件夹中，并输出每个文件在文件系统中的完整路径。

如果压缩和解压的过程都没有出现错误，那么我们可以确定Python压缩文件夹为zip的功能正常工作。

完整代码示例

下面是将文件夹压缩为zip文件的示例代码：

import zipfile
import os

def compress_folder_to_zip(folder_path, zip_file_path):
    with zipfile.ZipFile(zip_file_path, "w", zipfile.ZIP_DEFLATED) as zip_file:
        for root, dirs, files in os.walk(folder_path):
            for file_name in files:
                file_path = os.path.join(root, file_name)
                zip_file.write(file_path, os.path.relpath(file_path, folder_path))

def verify_zip_file(zip_file_path, folder_path):
    with zipfile.ZipFile(zip_file_path, "r") as zip_file:
        for file_info in zip_file.infolist():
            extracted_file_path = zip_file.extract(file_info, folder_path)
            print("extracted file: {}".format(extracted_file_path))

# 测试代码
compress_folder_to_zip("/Users/xxx/Downloads/test", "/Users/xxx/Downloads/test.zip")
verify_zip_file("/Users/xxx/Downloads/test.zip", "/Users/xxx/Downloads/test_extracted")