快捷導(dǎo)航

使用Python解壓各種壓縮文件的代碼實(shí)現(xiàn)

更新時(shí)間：2025年09月12日 08:51:22 作者：detayun

本文介紹通過(guò)文件頭（Magic Number）識(shí)別壓縮類(lèi)型的方法,支持ZIP、TAR、GZ、BZ2、RAR、7Z等格式,需第三方庫(kù)輔助,并需處理路徑安全、大文件流式解壓及異常情況,實(shí)現(xiàn)更可靠的自動(dòng)解壓方案,需要的朋友可以參考下

核心原理

通過(guò)文件頭（Magic Number）判斷壓縮類(lèi)型，而非依賴(lài)文件擴(kuò)展名。支持格式包括：ZIP、TAR、GZ、BZ2、RAR、7Z等。

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

import os
import zipfile
import tarfile
import gzip
import bz2
import shutil

def detect_compression(file_path):
    """通過(guò)文件頭識(shí)別壓縮類(lèi)型"""
    with open(file_path, 'rb') as f:
        # ZIP文件檢測(cè)
        if f.read(4) == b'PK\x03\x04':
            return 'zip'
        # TAR文件檢測(cè)
        if f.read(261)[257:262] == b'ustar':
            return 'tar'
        # GZ文件檢測(cè)
        if f.read(2) == b'\x1f\x8b':
            return 'gz'
        # BZ2文件檢測(cè)
        if f.read(3) == b'BZh':
            return 'bz2'
    return 'unknown'

def extract_archive(archive_path, extract_to='.'):
    """自動(dòng)解壓函數(shù)"""
    comp_type = detect_compression(archive_path)
    
    # 根據(jù)類(lèi)型調(diào)用對(duì)應(yīng)解壓方法
    if comp_type == 'zip':
        with zipfile.ZipFile(archive_path, 'r') as zip_ref:
            zip_ref.extractall(extract_to)
    elif comp_type == 'tar':
        with tarfile.open(archive_path) as tar_ref:
            tar_ref.extractall(extract_to)
    elif comp_type == 'gz':
        with gzip.open(archive_path, 'rb') as gz_ref:
            with open(os.path.join(extract_to, os.path.basename(archive_path)[:-3]), 'wb') as out_file:
                shutil.copyfileobj(gz_ref, out_file)
    elif comp_type == 'bz2':
        with bz2.open(archive_path, 'rb') as bz2_ref:
            with open(os.path.join(extract_to, os.path.basename(archive_path)[:-4]), 'wb') as out_file:
                shutil.copyfileobj(bz2_ref, out_file)
    else:
        print(f"不支持的壓縮格式: {archive_path}")

# 使用示例
extract_archive('unknown_archive.xyz', './output')

特殊格式處理

對(duì)于RAR/7Z等格式，需安裝第三方庫(kù)：

pip install rarfile py7zr

需在代碼中添加：

# 添加在detect_compression函數(shù)中
elif f.read(7) == b'Rar!\x1a\x07\x00':
    return 'rar'
elif f.read(6) == b'7z\x00\x00\x00\x00':
    return '7z'

# 添加在extract_archive函數(shù)中
elif comp_type == 'rar':
    from rarfile import RarFile
    with RarFile(archive_path) as rf:
        rf.extractall(extract_to)
elif comp_type == '7z':
    import py7zr
    with py7zr.SevenZipFile(archive_path) as z:
        z.extractall(extract_to)