通过Python编写的文件压缩工具

文件压缩是一种常见的数据处理任务，它可以将一个或多个文件压缩成一个单独的压缩文件，以减小文件的大小并方便传输和存储。Python提供了多个库和模块来实现文件压缩功能，其中最常用的是zipfile和shutil模块。本文将介绍如何使用这两个模块来编写一个文件压缩工具，并提供详细的使用例子。

## 使用zipfile模块压缩文件

zipfile模块提供了压缩和解压缩zip文件的功能。可以使用该模块创建一个ZipFile对象，然后使用其方法来添加和压缩文件。

以下是使用zipfile模块编写的简单文件压缩工具的示例代码：

import zipfile

def compress_files(file_paths, output_path):
    with zipfile.ZipFile(output_path, 'w') as zip_obj:
        for file_path in file_paths:
            zip_obj.write(file_path)
    print("压缩完成！")

# 使用例子
file_paths = ["file1.txt", "file2.txt", "file3.txt"]  # 要压缩的文件路径列表
output_path = "compressed_files.zip"  # 压缩后的输出路径
compress_files(file_paths, output_path)

在上述代码中，我们首先导入zipfile模块。然后，我们定义一个compress_files函数，该函数接受两个参数：file_paths和output_path。file_paths是要压缩的文件的路径列表，output_path是压缩后的zip文件的输出路径。

在compress_files函数中，我们创建了一个ZipFile对象zip_obj，并将其打开以供写入。然后，我们使用write方法将每个文件添加到zip文件中。最后，我们使用close方法关闭zip文件。这样，所有指定的文件将被压缩到一个zip文件中。

请注意，在使用with块时，文件对象会在退出块的时候自动关闭。这样可以确保在操作结束时正确关闭文件。

## 使用shutil模块压缩文件

shutil模块提供了更高级的文件操作功能，包括压缩和解压缩文件。该模块还可以处理各种压缩格式，如zip、tar和tar.gz。

以下是使用shutil模块编写的文件压缩工具的示例代码：

import shutil

def compress_files(file_paths, output_path):
    with shutil.make_archive(output_path, 'zip', *file_paths) as zip_obj:
        print("压缩完成！")

# 使用例子
file_paths = ["file1.txt", "file2.txt", "file3.txt"]  # 要压缩的文件路径列表
output_path = "compressed_files"  # 压缩后的输出路径
compress_files(file_paths, output_path)

在上述代码中，我们首先导入shutil模块。然后，我们定义了一个compress_files函数，该函数接受两个参数：file_paths和output_path。file_paths是要压缩的文件的路径列表，output_path是压缩后的文件的输出路径。

在compress_files函数中，我们使用make_archive函数创建一个压缩文件。该函数接受三个参数：output_path、format和root_dir。output_path是输出的压缩文件路径，format指定所需的压缩格式（如'zip'、'gztar'等），root_dir是文件的根目录。我们使用*运算符来解压缩文件路径列表，以便将它们作为参数传递给make_archive函数。

这样，所有指定的文件将被压缩到一个zip文件中，并保存在指定的输出路径下。

## 总结

本文介绍了如何使用Python编写一个文件压缩工具，并提供了两个示例：使用zipfile模块和shutil模块。无论您选择使用哪个模块，都可以很方便地实现文件的压缩功能。通过对这些示例代码的理解和实践，您可以根据自己的需要扩展和改进这个文件压缩工具。祝您使用Python进行文件压缩工作的愉快！