Python 压缩文件夹

介绍

在日常的文件处理中，我们经常需要对文件夹进行压缩以便于传输、存储或备份。Python 提供了丰富的库来处理文件和压缩操作，其中最常见的是 zipfile 模块。本文将详细介绍如何使用 Python 来压缩文件夹。

环境设置

在开始之前，我们需要确保已经安装了 Python。可以从官方网站 https://www.python.org/downloads/ 下载并安装最新版本的 Python。

导入模块

首先，我们需要导入 zipfile 模块，这是 Python 自带的用于处理压缩文件的模块。

import zipfile

压缩文件夹

使用 zipfile 模块可以非常方便地进行文件夹的压缩。下面是一个简单的示例，演示了如何压缩一个文件夹：

def compress_folder(folder_path, zip_path):
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for foldername, subfolders, filenames in os.walk(folder_path):
            for filename in filenames:
                file_path = os.path.join(foldername, filename)
                zipf.write(file_path, os.path.relpath(file_path, folder_path))

# 示例调用
folder_path = '/path/to/folder'
zip_path = '/path/to/compressed/file.zip'
compress_folder(folder_path, zip_path)

在上面的例子中，我们定义了一个名为 compress_folder 的函数，它接受文件夹路径和压缩文件路径作为参数。函数内部使用 zipfile.ZipFile 创建一个 Zip 文件对象，并使用 with 语句来自动关闭文件。

with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:

然后，我们通过遍历文件夹中的所有文件和文件夹，并使用 zipf.write 方法将它们写入 Zip 文件中。

for foldername, subfolders, filenames in os.walk(folder_path):
    for filename in filenames:
        file_path = os.path.join(foldername, filename)
        zipf.write(file_path, os.path.relpath(file_path, folder_path))

要注意的是，我们使用 os.path.relpath 函数来获取相对路径，这样在解压缩时可以保持文件夹结构。

解压缩文件夹

如果我们已经有一个压缩文件，想要将其解压缩到指定的文件夹中，也可以使用 zipfile 模块。

def extract_folder(zip_path, dest_folder):
    with zipfile.ZipFile(zip_path, 'r') as zipf:
        zipf.extractall(dest_folder)

# 示例调用
zip_path = '/path/to/compressed/file.zip'
dest_folder = '/path/to/destination/folder'
extract_folder(zip_path, dest_folder)

在上面的例子中，我们定义了一个名为 extract_folder 的函数，它接受压缩文件路径和目标文件夹路径作为参数。函数内部使用 zipfile.ZipFile 创建一个 Zip 文件对象，并使用 with 语句来自动关闭文件。

with zipfile.ZipFile(zip_path, 'r') as zipf:

然后，我们使用 zipf.extractall 方法将 Zip 文件中的所有文件解压缩到指定的文件夹中。

zipf.extractall(dest_folder)

进阶用法：加密压缩

zipfile 模块还支持对压缩文件进行加密，以提高文件的安全性。下面是一个加密压缩文件夹的示例：

def compress_folder_with_password(folder_path, zip_path, password):
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        zipf.setpassword(password.encode())
        for foldername, subfolders, filenames in os.walk(folder_path):
            for filename in filenames:
                file_path = os.path.join(foldername, filename)
                zipf.write(file_path, os.path.relpath(file_path, folder_path))

# 示例调用
folder_path = '/path/to/folder'
zip_path = '/path/to/compressed/file.zip'
password = 'password'
compress_folder_with_password(folder_path, zip_path, password)