快捷導(dǎo)航

解決Python復(fù)雜zip文件的解壓問題

更新時(shí)間：2021年12月11日 15:45:15 作者：Python_xiaowu

這篇文章主要介紹了Python復(fù)雜zip文件的解壓,通過配合 shutil 與 os 標(biāo)準(zhǔn)庫中的相關(guān)功能，實(shí)現(xiàn)將指定任意 zip 壓縮包，完好地解壓到指定的目錄中，需要的朋友可以參考下

廢話不多說，直接看問題，使用過?Python?中的標(biāo)準(zhǔn)庫?zipfile?解壓過?zip?格式壓縮包的朋友們，可能遇到過，當(dāng)壓縮文件中的目錄或文件名中包含中文等常見?unicode?字符時(shí)，典型如下面的例子：

使用?zipfile?的?extract()?或?extractall()?方法直接解壓時(shí)，產(chǎn)生的解壓結(jié)果名充斥著亂碼，這一點(diǎn)我們通過調(diào)用?namelist()?方法就可以看出來：

from zipfile import ZipFile

# 讀入壓縮包文件
file = ZipFile('示例壓縮包.zip')

# 查看壓縮包內(nèi)目錄、文件名稱
file.namelist()

這是因?yàn)?zipfile?中針對壓縮包內(nèi)容的編碼兼容性差，但我們可以通過下面的函數(shù)自行矯正：

def recode(raw: str) -> str:
    '''
    編碼修正
    '''
    
    try:
        return raw.encode('cp437').decode('gbk')
    
    except:
        return raw.encode('utf-8').decode('utf-8')
    
for file_or_path in file.namelist():
    
    print(file_or_path, ' -------> ' , recode(file_or_path))

解決了文件名亂碼的問題后，接下來我們就可以配合?shutil?與?os?標(biāo)準(zhǔn)庫中的相關(guān)功能，實(shí)現(xiàn)將指定任意?zip?壓縮包，完好地解壓到指定的目錄中，代碼如下：

def zip_extract_all(src_zip_file: ZipFile, target_path: str) -> None:
    
    # 遍歷壓縮包內(nèi)所有內(nèi)容
    for file_or_path in file.namelist():
        
        # 若當(dāng)前節(jié)點(diǎn)是文件夾
        if file_or_path.endswith('/'):
            try:
                # 基于當(dāng)前文件夾節(jié)點(diǎn)創(chuàng)建多層文件夾
                os.makedirs(os.path.join(target_path, recode(file_or_path)))
            except FileExistsError:
                # 若已存在則跳過創(chuàng)建過程
                pass
        
        # 否則視作文件進(jìn)行寫出
        else:
            # 利用shutil.copyfileobj，從壓縮包io流中提取目標(biāo)文件內(nèi)容寫出到目標(biāo)路徑
            with open(os.path.join(target_path, recode(file_or_path)), 'wb') as z:
                # 這里基于Zipfile.open()提取文件內(nèi)容時(shí)需要使用原始的亂碼文件名
                shutil.copyfileobj(src_zip_file.open(file_or_path), z)
            
# 向已存在的指定文件夾完整解壓當(dāng)前讀入的zip文件
zip_extract_all(file, '解壓測試')

可以看到，效果完美：