Python压缩文件夹完全指南：从基础到高级应用

在数据管理和传输中，压缩文件夹是常见的操作。Python提供了强大的内置库，如zipfile和shutil，使得文件夹压缩变得简单而高效。本文将系统介绍如何使用Python压缩文件夹，涵盖基础方法到高级技巧。

一、基础压缩：使用zipfile模块

zipfile模块是Python标准库的一部分，专门用于处理ZIP归档文件。它允许逐个添加文件，适合需要精细控制压缩过程的场景。

import zipfile
import os

def compress_folder_with_zipfile(folder_path, output_zip):
    with zipfile.ZipFile(output_zip, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                file_path = os.path.join(root, file)
                arcname = os.path.relpath(file_path, folder_path)
                zipf.write(file_path, arcname)
    print(f"文件夹已压缩到: {output_zip}")

# 示例使用
compress_folder_with_zipfile('my_folder', 'archive.zip')

此代码遍历指定文件夹，将每个文件添加到ZIP归档中。ZIP_DEFLATED表示使用DEFLATE压缩算法，平衡压缩率和速度。

二、快速压缩：使用shutil.make_archive

对于简单的一键压缩需求，shutil模块的make_archive函数是更高效的选择。它自动处理整个文件夹的递归压缩。

import shutil

shutil.make_archive('output_archive', 'zip', 'my_folder')
print("压缩完成，文件为 output_archive.zip")

该函数支持多种格式（如zip、tar、gztar），通过修改第二个参数即可切换。适用于快速备份或数据分发。

三、高级技巧：大文件处理与加密

1. 分割压缩包

当文件夹过大时，可以分割压缩包以适应传输限制。结合zipfile模块，可以实现分卷压缩：

import zipfile
import os

def split_compress(folder_path, max_size_mb=100):
    part_num = 0
    current_size = 0
    zipf = None
    
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            file_path = os.path.join(root, file)
            arcname = os.path.relpath(file_path, folder_path)
            file_size = os.path.getsize(file_path)
            
            if current_size + file_size > max_size_mb * 1024 * 1024:
                zipf.close()
                part_num += 1
                current_size = 0
            
            if not zipf or current_size == 0:
                zipf = zipfile.ZipFile(f"archive_part{part_num}.zip", 'w', zipfile.ZIP_DEFLATED)
            
            zipf.write(file_path, arcname)
            current_size += file_size
    
    if zipf:
        zipf.close()
    print(f"分割压缩完成，共 {part_num + 1} 部分")

2. 密码保护ZIP文件

zipfile模块支持AES加密，但需安装pyzipper库。以下为加密压缩示例：

import pyzipper

def encrypt_compress(folder_path, output_zip, password):
    with pyzipper.AESZipFile(output_zip, 'w', compression=pyzipper.ZIP_DEFLATED, encryption=pyzipper.WZ_AES) as zf:
        zf.setpassword(password.encode())
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                zf.write(os.path.join(root, file), arcname=os.path.relpath(os.path.join(root, file), folder_path))
    print(f"加密压缩完成: {output_zip}")

四、自动化场景：定时备份与云集成

结合调度库（如schedule），可以实现定时压缩备份。例如，每日压缩日志文件夹：

import schedule
import time

def daily_backup():
    compress_folder_with_zipfile('logs', f'logs_{time.strftime("%Y%m%d")}.zip')

schedule.every().day.at('23:00').do(daily_backup)

while True:
    schedule.run_pending()
    time.sleep(60)

对于云存储，可以将压缩后的文件上传至AWS S3或Google Cloud Storage，实现远程备份。

五、性能优化与最佳实践

• 选择压缩算法：根据数据类型选择ZIP_LZMA（高压缩率）或ZIP_STORE（无压缩，速度快）。
• 排除无关文件：在压缩前过滤临时文件（如__pycache__）或大型日志。
• 错误处理：添加异常捕获，确保文件句柄正确关闭，防止数据损坏。
• 并行处理：对于超大文件夹，可使用multiprocessing模块并行压缩子目录。

结论

Python的内置库为文件夹压缩提供了灵活而强大的工具。从简单的shutil.make_archive到可定制的zipfile方法，开发者可以根据需求选择合适方案。通过扩展加密、分卷和自动化，可以构建适应复杂场景的数据管理流程。掌握这些技巧，将显著提升文件处理效率与数据安全性。

提示：实际应用中，建议结合日志记录和校验（如MD5哈希）确保压缩数据的完整性。