从 Python3 中的 zip 存档中提取特定文件夹的内容

Question

我有一个 zip 存档，其内部结构如下所示：

file.zip
  |
   --- foo/
  |
   --- bar/
        |
         --- file1.txt
        |
         --- dir/
              |
               --- file2.txt

我想使用 python3 将bar 的内容提取到输出目录，得到如下所示的内容：

output-dir/
    |
     --- file1.txt
    |
     --- dir/
          |
           --- file2.txt

但是，当我在bar 下面运行代码时，它的内容被提取到output-dir

import zipfile

archive = zipfile.ZipFile('path/to/file.zip')

for archive_item in archive.namelist():
    if archive_item.startswith('bar/'):
        archive.extract(archive_item, 'path/to/output-dir')

我该如何解决这个问题？ 谢谢！

Answer 1

不要使用ZipFile.extract，而是使用ZipFile.open、open 和shutil.copyfileobj 以便将文件准确地放在您想要的位置，使用路径操作来创建输出你需要的路径。

archive = zipfile.ZipFile('path/to/file.zip')
PREFIX = 'bar/'
out = pathlib.Path('path/to/output-dir')
for archive_item in archive.namelist():
    if archive_item.startswith(PREFIX):
        # strip out the leading prefix then join to `out`, note that you 
        # may want to add some securing against path traversal if the zip
        # file comes from an untrusted source
        destpath = out.joinpath(archive_item[len(PREFIX):])
        # make sure destination directory exists otherwise `open` will fail
        os.makedirs(destpath.parent, exist_ok=True)
        with archive.open(archive_item) as source,
             open(destpath, 'wb') as dest:
            shutil.copyfileobj(source, dest)

类似的东西。