使用zipfile提取指定子目录遇异常：空文件夹及多余前置目录求助

问题根源 #

你的代码存在两个核心问题：

• 仅提取文件夹条目，未处理内部文件：Zip文件的namelist()中，文件夹和内部文件是独立条目。当前逻辑只匹配到了目标文件夹本身（如FolderA/FolderB/FolderC/ABC-FolderA/），并未处理文件夹内的文件条目，因此提取后是空文件夹。
• 保留原路径前置层级：zipfile.ZipFile.extract()会按条目原始路径在目标目录下重建完整层级，导致生成new_0/FolderA/FolderB/FolderC/ABC-FolderA/这类冗余结构，而非直接将目标文件夹内容放到new_0下。

修复后的代码 #

以下代码可提取目标文件夹内所有文件，同时去除FolderA/FolderB/FolderC前置层级：

import os
import sys
import zipfile

try:
    zip_file_path = sys.argv[1]
except IndexError:
    sys.exit('No zip file provided.')

# 定义需要剔除的前置路径前缀
prefix_to_remove = 'FolderA/FolderB/FolderC/'

with zipfile.ZipFile(zip_file_path) as archive:
    # 收集所有目标文件夹路径
    target_folders = set()
    for file in archive.namelist():
        # 筛选以ABC-开头的文件夹条目
        if file.endswith('/') and os.path.basename(file[:-1]).startswith('ABC-'):
            target_folders.add(file)
    
    # 遍历目标文件夹，提取内部文件
    for i, folder_path in enumerate(target_folders):
        new_dir = f'new_{i}'
        for file in archive.namelist():
            # 仅处理目标文件夹下的文件（排除文件夹本身）
            if file.startswith(folder_path) and not file.endswith('/'):
                # 重构目标路径：剔除前置前缀，拼接至新目录
                relative_path = file[len(prefix_to_remove):]
                dest_path = os.path.join(new_dir, relative_path)
                # 确保目标目录存在
                os.makedirs(os.path.dirname(dest_path), exist_ok=True)
                # 提取文件到指定路径
                archive.extract(file, os.path.dirname(dest_path))

代码说明 #

1. 收集目标文件夹：先遍历所有条目，筛选出所有以ABC-开头的文件夹，存入集合避免重复处理。
2. 提取文件内容：对每个目标文件夹，遍历所有条目，筛选出属于该文件夹的文件（排除文件夹本身条目）。
3. 重构路径：去掉FolderA/FolderB/FolderC/前置前缀，将文件直接放到new_i目录下。若需将原文件夹名（如ABC-FolderA）替换为new_i，可替换路径拼接逻辑：

   # 替换原文件夹名为new_dir
   relative_path = file.replace(folder_path, f'{new_dir}/')
   dest_path = relative_path
   

4. 安全处理：用with语句管理ZipFile，确保文件自动关闭；os.makedirs(exist_ok=True)避免目录已存在的报错。

测试验证 #

用你提供的测试代码生成testzip.zip后，运行修复后的脚本：

python extract.py testzip.zip

会生成new_0、new_1、new_2三个文件夹，每个文件夹下直接包含对应的somefile0.txt和somefile1.txt，无冗余的前置层级。

内容的提问来源于stack exchange，提问作者J.Doe