Python中如何实现文件压缩？zipfile模块使用

import zipfile
import os

# 假设我们有一些文件需要压缩
file_to_compress_1 = 'example_file_1.txt'
file_to_compress_2 = 'example_file_2.log'
# 创建一些示例文件
with open(file_to_compress_1, 'w') as f:
    f.write("这是第一个示例文件的内容。\n")
    f.write("多行文本测试。")
with open(file_to_to_compress_2, 'w') as f:
    f.write("这是第二个示例文件的日志内容。\n")
    f.write("一些运行记录。")

output_zip_name = 'my_archive.zip'

try:
    with zipfile.ZipFile(output_zip_name, 'w', zipfile.ZIP_DEFLATED) as zf:
        # 压缩单个文件
        zf.write(file_to_compress_1, arcname='renamed_file_in_zip.txt') # arcname可以指定在zip中的路径和名称
        # 压缩另一个文件
        zf.write(file_to_to_compress_2) # 不指定arcname，则保留原文件名

    print(f"文件 '{file_to_compress_1}' 和 '{file_to_to_compress_2}' 已成功压缩到 '{output_zip_name}'。")

except Exception as e:
    print(f"压缩过程中发生错误: {e}")
finally:
    # 清理示例文件
    if os.path.exists(file_to_compress_1):
        os.remove(file_to_compress_1)
    if os.path.exists(file_to_to_compress_2):
        os.remove(file_to_to_to_compress_2)
    # 如果压缩失败，可能没有生成zip文件，这里不做强制删除
    # if os.path.exists(output_zip_name):
    #     os.remove(output_zip_name)

import zipfile
import os

# 假设有这些文件
files_to_pack = ['document.txt', 'image.jpg', 'data.csv']
# 创建示例文件
for f_name in files_to_pack:
    with open(f_name, 'w') as f:
        f.write(f"这是 {f_name} 的内容。")

output_zip = 'my_documents.zip'

try:
    with zipfile.ZipFile(output_zip, 'w', zipfile.ZIP_DEFLATED) as zf:
        for file_path in files_to_pack:
            if os.path.exists(file_path):
                # 直接写入，文件在zip中保持原名
                zf.write(file_path)
            else:
                print(f"文件 '{file_path}' 不存在，跳过。")
    print(f"文件已成功打包到 '{output_zip}'。")
except Exception as e:
    print(f"打包文件时出错: {e}")
finally:
    for f_name in files_to_pack:
        if os.path.exists(f_name):
            os.remove(f_name)

import zipfile
import os

def compress_folder(folder_path, output_zip_path):
    if not os.path.isdir(folder_path):
        print(f"错误：'{folder_path}' 不是一个有效的文件夹路径。")
        return

    try:
        with zipfile.ZipFile(output_zip_path, 'w', zipfile.ZIP_DEFLATED) as zf:
            for root, dirs, files in os.walk(folder_path):
                # root 当前目录路径
                # dirs 当前目录下的子目录列表
                # files 当前目录下的文件列表

                # 计算当前文件或目录在zip文件中的相对路径
                # 这一步很关键，确保zip内部结构与源文件夹结构一致
                relative_path = os.path.relpath(root, folder_path)
                if relative_path == '.': # 如果是根目录本身
                    relative_path = ''

                for file in files:
                    file_full_path = os.path.join(root, file)
                    # 构建文件在zip中的路径
                    arcname = os.path.join(relative_path, file) if relative_path else file
                    zf.write(file_full_path, arcname=arcname)
                    print(f"正在压缩: {file_full_path} -> {arcname}")

                # 也可以选择性地添加空目录，zipfile默认不会添加空目录
                # for dir_name in dirs:
                #     dir_full_path = os.path.join(root, dir_name)
                #     arcname_dir = os.path.join(relative_path, dir_name) if relative_path else dir_name
                #     if not os.listdir(dir_full_path): # 如果是空目录
                #         zf.writestr(arcname_dir + '/', '') # 添加空目录项

        print(f"文件夹 '{folder_path}' 已成功压缩到 '{output_zip_path}'。")

    except Exception as e:
        print(f"压缩文件夹时发生错误: {e}")

# --- 示例用法 ---
# 创建一个示例文件夹结构
test_folder = 'my_project_data'
os.makedirs(os.path.join(test_folder, 'src'), exist_ok=True)
os.makedirs(os.path.join(test_folder, 'docs'), exist_ok=True)
os.makedirs(os.path.join(test_folder, 'empty_dir'), exist_ok=True) # 示例空目录

with open(os.path.join(test_folder, 'main.py'), 'w') as f:
    f.write("print('Hello from main.py')")
with open(os.path.join(test_folder, 'src', 'helper.py'), 'w') as f:
    f.write("def help_func(): pass")
with open(os.path.join(test_folder, 'docs', 'README.md'), 'w') as f:
    f.write("# My Project Readme")

output_zip_file = 'project_archive.zip'
compress_folder(test_folder, output_zip_file)

# 清理示例文件夹
import shutil
if os.path.exists(test_folder):
    shutil.rmtree(test_folder)
if os.path.exists(output_zip_file):
    print(f"生成的ZIP文件 '{output_zip_file}' 存在。")
    # os.remove(output_zip_file) # 可以选择删除

import zipfile
import os

# 示例文件
test_file = 'large_text_file.txt'
with open(test_file, 'w') as f:
    f.write("This is a test line for compression level testing.\n" * 1000)

# 压缩到不同的级别
zip_no_compression = 'no_compression.zip'
zip_default_compression = 'default_compression.zip'
zip_best_compression = 'best_compression.zip'

try:
    # 1. 无压缩 (仅存储)
    with zipfile.ZipFile(zip_no_compression, 'w', zipfile.ZIP_STORED) as zf:
        zf.write(test_file, arcname='stored_content.txt')
    print(f"'{test_file}' (无压缩) -> '{zip_no_compression}' (大小: {os.path.getsize(zip_no_compression)} 字节)")

    # 2. 默认压缩 (ZIP_DEFLATED)
    with zipfile.ZipFile(zip_default_compression, 'w', zipfile.ZIP_DEFLATED) as zf:
        zf.write(test_file, arcname='default_compressed_content.txt')
    print(f"'{test_file}' (默认压缩) -> '{zip_default_compression}' (大小: {os.path.getsize(zip_default_compression)} 字节)")

    # 3. 最高压缩级别 (ZIP_DEFLATED, compresslevel=9)
    # 注意：compresslevel只对ZIP_DEFLATED有效
    with zipfile.ZipFile(zip_best_compression, 'w', zipfile.ZIP_DEFLATED, compresslevel=9) as zf:
        zf.write(test_file, arcname='best_compressed_content.txt')
    print(f"'{test_file}' (最高压缩) -> '{zip_best_compression}' (大小: {os.path.getsize(zip_best_compression)} 字节)")

    print(f"原始文件大小: {os.path.getsize(test_file)} 字节")

except Exception as e:
    print(f"控制压缩级别时发生错误: {e}")
finally:
    if os.path.exists(test_file):
        os.remove(test_file)
    if os.path.exists(zip_no_compression):
        os.remove(zip_no_compression)
    if os.path.exists(zip_default_compression):
        os.remove(zip_default_compression)
    if os.path.exists(zip_best_compression):
        os.remove(zip_best_compression)