Python 办公神器：教你使用 Python 自动化处理压缩文件

在这篇文章中，我将教你如何用python编写一个简单的程序，实现自动化处理压缩文件的功能。你只需要运行这个程序，它就会定时检测你指定的文件夹（比如 download 文件夹），如果发现有新的压缩文件，就会自动将它们解压缩到当前文件夹内的新文件夹中，并删除原来的压缩文件。

你是否经常需要下载一些压缩文件，然后手动解压缩它们，再删除原来的压缩文件？这个过程是不是很繁琐，而且容易出错？有没有想过用python来自动化这个过程，让你的办公效率提高呢？

这个程序主要涉及到以下几个方面的知识：

• os 模块：这个模块提供了一些操作系统相关的功能，比如创建、删除、重命名文件和文件夹，获取文件的属性，切换工作目录等。
• glob 模块：这个模块提供了一些通配符匹配的功能，比如可以用 *.zip 来匹配所有以 .zip 结尾的文件，或者用 ??.txt 来匹配两个字符的文本文件等。
• gzip zipfile rarfile tarfile 模块：这些模块分别提供了对不同格式的压缩文件的读写操作，比如可以用 gzip 模块来处理 .gz 格式的文件，用 zipfile 模块来处理 .zip 格式的文件，用 rarfile 模块来处理 .rar 格式的文件，用 tarfile 模块来处理 .tar .tgz .bz2 等格式的文件。

下面，我们就来一步一步地编写这个程序，你可以跟着我一起动手试试。

第一步：导入需要的模块

首先，我们需要导入上面提到的几个模块，以及 time 模块，用于控制程序的运行间隔。我们可以在程序的开头写下如下代码：

import os
import glob
import gzip
import zipfile
import rarfile
import tarfile
import time

第二步：定义需要处理的文件夹和压缩文件的格式

接下来，我们需要定义一些常量，用于指定需要处理的文件夹的路径，以及需要处理的压缩文件的格式。我们可以在导入模块的下面写下如下代码：

# 需要处理的文件夹的路径，你可以根据自己的实际情况修改
FOLDER = 'download'

# 需要处理的压缩文件的格式，你可以根据自己的实际情况添加或删除
FORMATS = ['.gz', '.zip', '.rar', '.tar', '.tgz', '.bz2']

第三步：定义解压缩文件的函数

然后，我们需要定义一个函数，用于根据不同的压缩文件格式，调用不同的模块来解压缩文件。我们可以在定义常量的下面写下如下代码：

def unzip_file(file):
    # 获取文件的扩展名，用于判断文件的格式
    ext = os.path.splitext(file)[1]

    # 根据不同的格式，调用不同的模块来解压缩文件
    if ext == '.gz':
        # 如果是 .gz 格式的文件，用 gzip 模块来解压缩
        with gzip.open(file, 'rb') as f_in:
            # 获取解压缩后的文件名，去掉 .gz 后缀
            out_file = file[:-3]
            # 以二进制写入模式打开输出文件
            with open(out_file, 'wb') as f_out:
                # 将输入文件的内容复制到输出文件
                f_out.write(f_in.read())
    elif ext == '.zip':
        # 如果是 .zip 格式的文件，用 zipfile 模块来解压缩
        with zipfile.ZipFile(file, 'r') as f_in:
            # 获取解压缩后的文件夹名，去掉 .zip 后缀
            out_folder = file[:-4]
            # 创建输出文件夹
            os.mkdir(out_folder)
            # 将输入文件中的所有文件解压缩到输出文件夹
            f_in.extractall(out_folder)
    elif ext == '.rar':
        # 如果是 .rar 格式的文件，用 rarfile 模块来解压缩
        with rarfile.RarFile(file, 'r') as f_in:
            # 获取解压缩后的文件夹名，去掉 .rar 后缀
            out_folder = file[:-4]
            # 创建输出文件夹
            os.mkdir(out_folder)
            # 将输入文件中的所有文件解压缩到输出文件夹
            f_in.extractall(out_folder)
    elif ext in ['.tar', '.tgz', '.bz2']:
        # 如果是 .tar .tgz .bz2 等格式的文件，用 tarfile 模块来解压缩
        with tarfile.open(file, 'r') as f_in:
            # 获取解压缩后的文件夹名，去掉扩展名
            out_folder = os.path.splitext(file)[0]
            # 创建输出文件夹
            os.mkdir(out_folder)
            # 将输入文件中的所有文件解压缩到输出文件夹
            f_in.extractall(out_folder)
    else:
        # 如果是其他格式的文件，不做处理，直接返回
        return

    # 解压缩完成后，删除原来的压缩文件
    os.remove(file)

这个函数的主要逻辑是：

• 首先，获取文件的扩展名，用于判断文件的格式。
• 然后，根据不同的格式，调用不同的模块来解压缩文件。这里，我们分别处理了 .gz .zip .rar .tar .tgz .bz2 等常见的压缩文件格式，你可以根据自己的需要添加或删除其他格式。
• 最后，解压缩完成后，删除原来的压缩文件，释放空间。

第四步：定义主函数

最后，我们需要定义一个主函数，用于定时检测指定的文件夹，如果发现有新的压缩文件，就调用上面定义的函数来解压缩它们。我们可以在定义解压缩文件的函数的下面写下如下代码：

def main():
    # 无限循环，直到用户终止程序
    while True:
        # 切换到指定的文件夹
        os.chdir(FOLDER)

        # 遍历需要处理的压缩文件的格式
        for format in FORMATS:
            # 用 glob 模块来匹配当前文件夹下的所有符合格式的文件
            files = glob.glob('*' + format)

            # 遍历匹配到的文件
            for file in files:
                # 调用解压缩文件的函数来处理文件
                unzip_file(file)

        # 等待一段时间（单位为秒），你可以根据自己的需要修改
        # 这里我们设置为 10 秒，你可以根据自己的需要修改
        time.sleep(10)

这个函数的主要逻辑是：

• 首先，无限循环，直到用户终止程序。
• 然后，切换到指定的文件夹，遍历需要处理的压缩文件的格式，用 glob 模块来匹配当前文件夹下的所有符合格式的文件。
• 最后，遍历匹配到的文件，调用解压缩文件的函数来处理文件，等待一段时间，然后重复上述过程。

至此，我们已经完成了这个程序的编写，你可以将所有的代码保存为一个 .py 文件，比如 unzip.py，然后在命令行中运行它，就可以看到效果了。

python unzip.py

你可以尝试在 download 文件夹中放入一些不同格式的压缩文件，看看程序是否能正确地解压缩它们，并删除原来的压缩文件。

这个程序虽然简单，但是却能实现一个非常实用的功能，帮助你自动化处理压缩文件，提高你的办公效率。你可以根据自己的需要，修改或扩展这个程序，比如增加日志记录，增加异常处理，增加其他格式的支持等。