作者： 陳熹

公眾號(hào)：早起Python

一、需求描述

編寫一個(gè)python程序，每次下載壓縮包形式的文件后，自動(dòng)將內(nèi)部文件解壓到當(dāng)前文件夾后將壓縮包刪除，通過(guò)本案例可以學(xué)到的知識(shí)點(diǎn)：

• os 模塊綜合應(yīng)用
• glob 模塊綜合應(yīng)用
• 利用 gzip zipfile rarfile tarfile 模塊解壓文件

二、步驟分析和前置知識(shí)

碼代碼之前需要將復(fù)雜問(wèn)題解釋成多個(gè)明確的要求，即這個(gè)程序?qū)崿F(xiàn)的邏輯為：

1. 定時(shí)檢測(cè)某個(gè)文件夾（如 download 文件夾）是否存在壓縮文件
2. 如果有新出現(xiàn)的壓縮文件則將其解壓到當(dāng)前文件夾內(nèi)的新文件夾中 刪除壓縮文件
3. 提到壓縮文件就要分不同壓縮格式進(jìn)行討論，主要有以下4種：

提到壓縮文件就要分不同壓縮格式進(jìn)行討論，主要有以下4種：

“

.gz: 即gzip，通常只能壓縮一個(gè)文件

.tar: 本質(zhì)上不是壓縮工具而是打包工具，可以跟.gz配合形成.tar.gz的打包壓縮格式

.zip: 和.tar.gz很類似但可以打包壓縮多個(gè)文件

.rar: 打包壓縮文件，最初用于DOS

”

因此，文件是否為壓縮文件的判斷邏輯可以如下：

1. 建立一個(gè)含壓縮包后綴的列表 compressed_lst = ['gz', 'tar', 'zip', 'rar']
2. 目標(biāo)文件夾下的所有文件名運(yùn)行filename.split(.)[-1]獲取后綴名
3. 判斷后綴名是否在compressed_lst中，如果存在則運(yùn)行后續(xù)的解壓代碼
4. 如果是以gz結(jié)尾的文件，解壓后需要再次判斷是否是以.tar結(jié)尾，并進(jìn)行相應(yīng)處理

不同壓縮文件的解壓代碼各不相同，在后續(xù)代碼操作中具體展開。

三、代碼實(shí)現(xiàn)

首先是獲取 download 文件夾下所有文件的文件名

import glob

import os

path = r'C:xxxxdownload'

file_lst = glob.glob(path + '/*')

# 列表推導(dǎo)式

filename_lst = [os.path.basename(i) for i in file_lst]

print(filename_lst)

接下來(lái)根據(jù)文件的后綴名判斷是否需要壓縮，先看一下代碼框架

for i in filename_lst:

if '.' in i:

# 獲取后綴名

suffix = i.split('.')[-1]

# 逐個(gè)將后綴名和各格式的壓縮后綴匹配

if suffix == 'gz':

pass

if ...: # 再次判斷新產(chǎn)生的文件名是否以 .tar 結(jié)尾

pass

if suffix == 'tar':

pass

if suffix == 'zip':

pass

if suffix == 'rar':

pass

這里有2個(gè)注意點(diǎn)：

1. 有些文件沒(méi)有后綴名，因此可能不存在 .，這時(shí)用 filename.split('.')[-1]會(huì)報(bào)錯(cuò)，因此需要先判斷
2. 以.gz 結(jié)尾的文件，解壓后需要再次判斷是否是以 .tar 結(jié)尾，然后我們就可以分別將四種壓縮文件的解壓代碼寫成函數(shù)，符合時(shí)候單獨(dú)調(diào)用即可

處理.gz 文件

import gzip

def ungz(filename):

filename = filename[:-3] # gz文件的單文件解壓就是去掉 filename 后面的 .gz

gz_file = gzip.GzipFile(filename)

with open(filename, "w+") as file:

file.write(gz_file.read())

return filename # 這個(gè)gzip的函數(shù)需要返回值以進(jìn)一步配合untar函數(shù)

前面反復(fù)提及的gz文件有可能跟tar文件配合，因此解壓gz文件后需要再判斷是否有必要解開tar文件

這里就可以先把tar文件的函數(shù)寫出來(lái)

——熱門課程推薦：

想學(xué)習(xí)PYTHON數(shù)據(jù)分析與金融數(shù)字化轉(zhuǎn)型精英訓(xùn)練營(yíng)，您可以點(diǎn)擊>>>“人才轉(zhuǎn)型”了解課程詳情；

想從事業(yè)務(wù)型數(shù)據(jù)分析師，您可以點(diǎn)擊>>>“數(shù)據(jù)分析師”了解課程詳情；

想從事大數(shù)據(jù)分析師，您可以點(diǎn)擊>>>“大數(shù)據(jù)就業(yè)”了解課程詳情；

想成為人工智能工程師，您可以點(diǎn)擊>>>“人工智能就業(yè)”了解課程詳情；

想了解Python數(shù)據(jù)分析，您可以點(diǎn)擊>>>“Python數(shù)據(jù)分析師”了解課程詳情；

想咨詢互聯(lián)網(wǎng)運(yùn)營(yíng)，你可以點(diǎn)擊>>>“互聯(lián)網(wǎng)運(yùn)營(yíng)就業(yè)班”了解課程詳情； 

想了解更多優(yōu)質(zhì)課程，請(qǐng)點(diǎn)擊>>>