一、需求描述
編寫一個Python程序，每次下載壓縮包形式的文件后，自動將內(nèi)部文件解壓到當(dāng)前文件夾后將壓縮包刪除，通過本案例可以學(xué)到的知識點(diǎn)：
- os 模塊綜合應(yīng)用
- glob 模塊綜合應(yīng)用
- 利用 gzip zipfile rarfile tarfile 模塊解壓文件
二、步驟分析和前置知識
碼代碼之前需要將復(fù)雜問題解釋成多個明確的要求，即這個程序?qū)崿F(xiàn)的邏輯為：
- 定時檢測某個文件夾(如 download 文件夾)是否存在壓縮文件
- 如果有新出現(xiàn)的壓縮文件則將其解壓到當(dāng)前文件夾內(nèi)的新文件夾中 刪除壓縮文件
提到壓縮文件就要分不同壓縮格式進(jìn)行討論，主要有以下4種：
- gz: 即gzip，通常只能壓縮一個文件
- tar: 本質(zhì)上不是壓縮工具而是打包工具，可以跟.gz配合形成.tar.gz的打包壓縮格式
- zip: 和.tar.gz很類似但可以打包壓縮多個文件
- rar: 打包壓縮文件，最初用于DOS
因此，文件是否為壓縮文件的判斷邏輯可以如下：
- 建立一個含壓縮包后綴的列表 compressed_lst = ['gz', 'tar', 'zip', 'rar']
- 目標(biāo)文件夾下的所有文件名運(yùn)行filename.split(.)[-1]獲取后綴名
- 判斷后綴名是否在compressed_lst中，如果存在則運(yùn)行后續(xù)的解壓代碼
- 如果是以gz結(jié)尾的文件，解壓后需要再次判斷是否是以.tar結(jié)尾，并進(jìn)行相應(yīng)處理
不同壓縮文件的解壓代碼各不相同，在后續(xù)代碼操作中具體展開。
三、代碼實(shí)現(xiàn)
首先是獲取 download 文件夾下所有文件的文件名
import glob
import os
path = r'C:\xxxx\download'
file_lst = glob.glob(path + '/*')
# 列表推導(dǎo)式
filename_lst = [os.path.basename(i) for i in file_lst]
print(filename_lst)
接下來根據(jù)文件的后綴名判斷是否需要壓縮，先看一下代碼框架
for i in filename_lst:
if '.' in i:
# 獲取后綴名
suffix = i.split('.')[-1]
# 逐個將后綴名和各格式的壓縮后綴匹配
if suffix == 'gz':
pass
if ...: # 再次判斷新產(chǎn)生的文件名是否以 .tar 結(jié)尾
pass
if suffix == 'tar':
pass
if suffix == 'zip':
pass
if suffix == 'rar':
pass
這里有2個注意點(diǎn)：
- 有些文件沒有后綴名，因此可能不存在 .，這時用 filename.split('.')[-1]會報錯，因此需要先判斷
- 以.gz 結(jié)尾的文件，解壓后需要再次判斷是否是以 .tar 結(jié)尾，然后我們就可以分別將四種壓縮文件的解壓代碼寫成函數(shù)，符合時候單獨(dú)調(diào)用即可
1. 處理.gz 文件
import gzip
def ungz(filename):
filenamefilename = filename[:-3] # gz文件的單文件解壓就是去掉 filename 后面的 .gz
gz_file = gzip.GzipFile(filename)
with open(filename, "w+") as file:
file.write(gz_file.read())
return filename # 這個gzip的函數(shù)需要返回值以進(jìn)一步配合untar函數(shù)
前面反復(fù)提及的gz文件有可能跟tar文件配合，因此解壓gz文件后需要再判斷是否有必要解開tar文件
這里就可以先把tar文件的函數(shù)寫出來
