使用Python编写helper函数来处理文件压缩和解压缩

压缩和解压缩是常见的文件处理操作，Python提供了多个库来处理文件的压缩和解压缩。在本篇文章中，我将展示如何使用Python编写helper函数来处理文件的压缩和解压缩，并提供使用例子。

1. 使用zipfile库进行文件压缩和解压缩

zipfile库是Python的内置库，提供了对ZIP格式文件的压缩和解压缩操作。

首先，我们需要导入zipfile库：

import zipfile

a. 压缩文件：

我们可以使用zipfile.ZipFile()函数来创建一个ZIP文件对象，并使用write()方法来将文件添加到ZIP文件中。下面是一个例子：

def compress_files(file_list, zip_name):
    with zipfile.ZipFile(zip_name, 'w') as zip_obj:
        for file in file_list:
            zip_obj.write(file)

这个函数接受一个文件列表file_list和一个ZIP文件名zip_name作为参数。函数会遍历文件列表，将每个文件添加到ZIP文件中，并以zip_name为名字保存。

b. 解压缩文件：

我们可以使用zipfile.ZipFile()函数来创建一个ZIP文件对象，并使用extractall()方法将ZIP文件中的所有文件解压缩到指定目录。下面是一个例子：

def extract_files(zip_name, extract_path):
    with zipfile.ZipFile(zip_name, 'r') as zip_obj:
        zip_obj.extractall(extract_path)

这个函数接受一个ZIP文件名zip_name和一个解压缩路径extract_path作为参数。函数会将ZIP文件中的所有文件解压缩到指定路径extract_path下。

2. 使用tarfile库进行文件压缩和解压缩

tarfile库是Python的内置库，提供了对TAR格式文件的压缩和解压缩操作。

首先，我们需要导入tarfile库：

import tarfile

a. 压缩文件：

我们可以使用tarfile.open()函数创建一个TAR文件对象，并使用add()方法将文件添加到TAR文件中。下面是一个例子：

def compress_files(file_list, tar_name):
    with tarfile.open(tar_name, 'w') as tar_obj:
        for file in file_list:
            tar_obj.add(file)

这个函数接受一个文件列表file_list和一个TAR文件名tar_name作为参数。函数会遍历文件列表，将每个文件添加到TAR文件中，并以tar_name为名字保存。

b. 解压缩文件：

我们可以使用tarfile.open()函数创建一个TAR文件对象，并使用extractall()方法将TAR文件中的所有文件解压缩到指定目录。下面是一个例子：

def extract_files(tar_name, extract_path):
    with tarfile.open(tar_name, 'r') as tar_obj:
        tar_obj.extractall(extract_path)

这个函数接受一个TAR文件名tar_name和一个解压缩路径extract_path作为参数。函数会将TAR文件中的所有文件解压缩到指定路径extract_path下。

3. 使用shutil库进行文件压缩和解压缩

shutil库是Python的内置库，提供了高级的文件操作功能，包括文件的压缩和解压缩操作。

首先，我们需要导入shutil库：

import shutil

a. 压缩文件：

我们可以使用shutil.make_archive()函数来创建一个指定格式的压缩文件。下面是一个例子：

def compress_files(file_list, zip_name):
    shutil.make_archive(zip_name, 'zip', *file_list)

这个函数接受一个文件列表file_list和一个ZIP文件名zip_name作为参数。函数会将文件列表中的所有文件压缩到一个ZIP文件中，并以zip_name为名字保存。

b. 解压缩文件：

我们可以使用shutil.unpack_archive()函数来解压缩指定格式的压缩文件。下面是一个例子：

def extract_files(zip_name, extract_path):
    shutil.unpack_archive(zip_name, extract_path)

这个函数接受一个压缩文件名zip_name和一个解压缩路径extract_path作为参数。函数会将压缩文件解压缩到指定路径extract_path下。

使用例子：

下面是一个示例代码，演示了如何使用上述的helper函数来处理文件的压缩和解压缩：

# 导入库
import zipfile
import tarfile
import shutil

# 使用zipfile压缩和解压缩
def compress_files_zip(file_list, zip_name):
    with zipfile.ZipFile(zip_name, 'w') as zip_obj:
        for file in file_list:
            zip_obj.write(file)

def extract_files_zip(zip_name, extract_path):
    with zipfile.ZipFile(zip_name, 'r') as zip_obj:
        zip_obj.extractall(extract_path)

# 使用tarfile压缩和解压缩
def compress_files_tar(file_list, tar_name):
    with tarfile.open(tar_name, 'w') as tar_obj:
        for file in file_list:
            tar_obj.add(file)

def extract_files_tar(tar_name, extract_path):
    with tarfile.open(tar_name, 'r') as tar_obj:
        tar_obj.extractall(extract_path)

# 使用shutil压缩和解压缩
def compress_files_shutil(file_list, zip_name):
    shutil.make_archive(zip_name, 'zip', *file_list)

def extract_files_shutil(zip_name, extract_path):
    shutil.unpack_archive(zip_name, extract_path)

# 使用示例
file_list = ['file1.txt', 'file2.txt', 'file3.txt']

compress_files_zip(file_list, 'files.zip')
extract_files_zip('files.zip', 'extracted_zip')

compress_files_tar(file_list, 'files.tar')
extract_files_tar('files.tar', 'extracted_tar')

compress_files_shutil(file_list, 'files_shutil')
extract_files_shutil('files_shutil.zip', 'extracted_shutil')

使用上述代码，你可以在压缩和解压缩文件时灵活选择使用zipfile、tarfile还是shutil库。