使用Python解压各种压缩文件的代码实现

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

def detect_compression(file_path):
    """通过文件头识别压缩类型"""
    with open(file_path, 'rb') as f:
        # ZIP文件检测
        if f.read(4) == b'PK\x03\x04':
            return 'zip'
        # TAR文件检测
        if f.read(261)[257:262] == b'ustar':
            return 'tar'
        # GZ文件检测
        if f.read(2) == b'\x1f\x8b':
            return 'gz'
        # BZ2文件检测
        if f.read(3) == b'BZh':
            return 'bz2'
    return 'unknown'

def extract_archive(archive_path, extract_to='.'):
    """自动解压函数"""
    comp_type = detect_compression(archive_path)
    
    # 根据类型调用对应解压方法
    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')

pip install rarfile py7zr

# 添加在detect_compression函数中
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函数中
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)