Python实现一键批量解压文件夹内的所有ZIP/RAR并自动分类

# ZIP/TAR 使用内置库，无需安装
pip install rarfile py7zr

import os
import zipfile
import tarfile
from pathlib import Path
from datetime import datetime

# ==================== ZIP 解压 ====================
def extract_zip(zip_path, extract_dir=None):
    """
    解压 ZIP 文件

    参数:
        zip_path: ZIP 文件路径
        extract_dir: 解压目录（默认在 ZIP 同级目录创建同名文件夹）
    """
    zip_file = Path(zip_path)
    if extract_dir is None:
        extract_dir = zip_file.parent / zip_file.stem

    extract_dir.mkdir(parents=True, exist_ok=True)

    with zipfile.ZipFile(zip_file, 'r') as zf:
        # 检查是否有密码
        has_password = any(zinfo.flag_bits & 0x1 for zinfo in zf.infolist())

        if has_password:
            password = input(f"请输入 {zip_file.name} 的密码: ").encode()
        else:
            password = None

        zf.extractall(extract_dir, pwd=password)

    print(f"[ZIP解压] {zip_file.name} → {extract_dir}/")
    return extract_dir


# ==================== TAR/GZ 解压 ====================
def extract_tar(tar_path, extract_dir=None):
    """解压 TAR/TAR.GZ/TAR.BZ2 文件"""
    tar_file = Path(tar_path)
    if extract_dir is None:
        extract_dir = tar_file.parent / tar_file.stem

    extract_dir.mkdir(parents=True, exist_ok=True)

    # 自动检测压缩类型
    mode = 'r:*'  # 自动检测
    with tarfile.open(tar_file, mode) as tf:
        tf.extractall(extract_dir)

    print(f"[TAR解压] {tar_file.name} → {extract_dir}/")
    return extract_dir


# ==================== 批量解压 ====================
def batch_extract_all(search_dir, extract_base=None):
    """
    批量解压目录中的所有压缩包

    参数:
        search_dir: 包含压缩包的目录
        extract_base: 解压根目录（默认在源目录下创建"已解压"文件夹）
    """
    search_path = Path(search_dir)

    if extract_base is None:
        extract_base = search_path / "已解压"
    extract_base.mkdir(parents=True, exist_ok=True)

    stats = {'success': 0, 'failed': 0, 'skipped': 0}

    for file_path in search_path.rglob('*'):
        if not file_path.is_file():
            continue

        suffix = file_path.suffix.lower()
        extract_dir = extract_base / file_path.stem

        # 跳过已经解压过的
        if extract_dir.exists() and any(extract_dir.iterdir()):
            stats['skipped'] += 1
            continue

        try:
            if suffix == '.zip':
                extract_zip(file_path, extract_dir)
                stats['success'] += 1

            elif suffix in ('.tar', '.gz', '.bz2', '.tgz'):
                extract_tar(file_path, extract_dir)
                stats['success'] += 1

            elif suffix == '.rar':
                try:
                    import rarfile
                    with rarfile.RarFile(file_path, 'r') as rf:
                        extract_dir.mkdir(parents=True, exist_ok=True)
                        rf.extractall(extract_dir)
                    print(f"[RAR解压] {file_path.name} → {extract_dir}/")
                    stats['success'] += 1
                except ImportError:
                    print(f"[跳过] {file_path.name}: 需要安装 rarfile 库")
                    stats['failed'] += 1

            elif suffix == '.7z':
                try:
                    import py7zr
                    with py7zr.SevenZipFile(file_path, 'r') as sz:
                        extract_dir.mkdir(parents=True, exist_ok=True)
                        sz.extractall(extract_dir)
                    print(f"[7Z解压] {file_path.name} → {extract_dir}/")
                    stats['success'] += 1
                except ImportError:
                    print(f"[跳过] {file_path.name}: 需要安装 py7zr 库")
                    stats['failed'] += 1

        except Exception as e:
            print(f"[失败] {file_path.name}: {e}")
            stats['failed'] += 1

    # 汇总
    print(f"\n{'='*50}")
    print(f"批量解压完成!")
    print(f"成功: {stats['success']}")
    print(f"失败: {stats['failed']}")
    print(f"跳过: {stats['skipped']}")


# ==================== 解压后文件分类 ====================
def classify_extracted_files(extract_dir):
    """
    对解压后的文件按类型分类整理
    """
    from shutil import move

    CLASSIFY_RULES = {
        '.jpg': '图片', '.jpeg': '图片', '.png': '图片', '.gif': '图片',
        '.doc': '文档', '.docx': '文档', '.pdf': '文档', '.txt': '文档',
        '.xls': '表格', '.xlsx': '表格', '.csv': '表格',
        '.mp4': '视频', '.avi': '视频', '.mkv': '视频',
        '.mp3': '音频', '.wav': '音频',
    }

    extract_path = Path(extract_dir)

    for file_path in extract_path.rglob('*'):
        if file_path.is_file():
            suffix = file_path.suffix.lower()
            category = CLASSIFY_RULES.get(suffix)

            if category:
                category_dir = extract_path / category
                category_dir.mkdir(exist_ok=True)

                dest = category_dir / file_path.name
                if not dest.exists():
                    move(str(file_path), str(dest))

    print(f"文件分类完成: {extract_dir}/")


# ==================== 使用示例 ====================
if __name__ == "__main__":
    # 批量解压
    batch_extract_all(r"D:\Downloads\压缩包")

    # 分类解压后的文件
    # classify_extracted_files(r"D:\Downloads\压缩包\已解压")

import rarfile
rarfile.UNRAR_TOOL = r"C:\Program Files\WinRAR\UnRAR.exe"

zf.extractall(extract_dir, pwd=b'your_password')