python用zipfile压缩文件夹

Python使用zipfile压缩文件夹的实现方法

简介

在Python开发中，有时候我们需要将一个文件夹或者多个文件压缩成一个zip文件，以便于传输或者保存。Python中的zipfile模块提供了压缩和解压缩文件的功能，可以轻松地实现这个需求。

本文将介绍如何使用Python的zipfile模块来压缩文件夹，并逐步引导刚入行的小白完成这个任务。

总体流程

下面是实现“Python用zipfile压缩文件夹”的整体流程：

erDiagram
    开始 --> 检查文件夹是否存在
    检查文件夹是否存在 --> 压缩文件夹
    压缩文件夹 --> 结束

步骤详解

步骤1：检查文件夹是否存在

在开始压缩文件夹之前，我们首先需要检查待压缩的文件夹是否存在。如果文件夹不存在，我们将无法进行压缩操作。

import os

def check_folder_exist(folder_path):
    if not os.path.exists(folder_path):
        print("文件夹不存在！")
        return False
    else:
        return True

以上代码中，我们使用os.path.exists()函数来检查文件夹是否存在。如果文件夹不存在，则会输出提示信息，并返回False；如果文件夹存在，则返回True。

步骤2：压缩文件夹

在确认文件夹存在后，我们可以开始压缩文件夹了。我们将使用zipfile模块中的ZipFile类来完成压缩操作。

import zipfile

def compress_folder(folder_path, zip_path):
    with zipfile.ZipFile(zip_path, 'w') as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                zipf.write(os.path.join(root, file), file)

以上代码中，我们首先创建一个ZipFile对象，并指定了要生成的zip文件的路径和模式（'w'表示写入模式）。然后，我们使用os.walk()函数遍历文件夹中的所有文件和子文件夹，并将每个文件写入到zip文件中。

完整代码

下面是完整的Python代码，将以上两个函数整合在一起：

import os
import zipfile

def check_folder_exist(folder_path):
    if not os.path.exists(folder_path):
        print("文件夹不存在！")
        return False
    else:
        return True

def compress_folder(folder_path, zip_path):
    with zipfile.ZipFile(zip_path, 'w') as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                zipf.write(os.path.join(root, file), file)

if __name__ == "__main__":
    folder_path = "待压缩文件夹路径"
    zip_path = "生成的zip文件路径"

    if check_folder_exist(folder_path):
        compress_folder(folder_path, zip_path)
        print("文件夹压缩成功！")

总结

本文介绍了使用Python的zipfile模块来压缩文件夹的方法。首先，我们需要检查待压缩的文件夹是否存在，然后才能进行压缩操作。在压缩过程中，我们使用了ZipFile类来创建zip文件，并使用os.walk()函数遍历文件夹中的所有文件。

希望本文能够帮助刚入行的小白掌握使用Python压缩文件夹的方法。如有任何问题，请随时提问。