你是否经常需要下载一些压缩文件,然后手动解压缩它们,再删除原来的压缩文件?这个过程是不是很繁琐,而且容易出错?有没有想过用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 文件夹中放入一些不同格式的压缩文件,看看程序是否能正确地解压缩它们,并删除原来的压缩文件。
这个程序虽然简单,但是却能实现一个非常实用的功能,帮助你自动化处理压缩文件,提高你的办公效率。你可以根据自己的需要,修改或扩展这个程序,比如增加日志记录,增加异常处理,增加其他格式的支持等。