使用 Python 将 zip 文件和嵌套的 zip 文件提取到目标目录中

Question

我有一个类似这样的文件结构：

/a.zip

/not_a_zip/

内容

/b.zip

内容

我想创建一个目录 a 并将 a.zip 和所有嵌套的压缩文件提取到其中，所以我得到这样的东西：

/一个/

/not_a_zip/

内容

/b/

内容

我尝试了这个解决方案，但我遇到了错误，因为在我的主目录中我有子目录以及 zip 文件。

我希望能够将主 zip 文件提取到同名目录中，然后能够提取其中的所有嵌套文件，无论它们嵌套多深。

编辑：我当前的代码是这个

archive = zipfile.ZipFile(zipped, 'r')
for file in archive.namelist():
    archive.extract(file, resultDirectory)

for f in [filename for filename in archive.NameToInfo if filename.endswith(".zip")]:
    # get file name and path to extract
    fileToExtract = resultDirectory + '/' + f
    # get directory to extract new file to
    directoryToExtractTo = fileToExtract.rsplit('/', 1)
    directoryToExtractTo = directoryToExtractTo[0] + '/'
    # extract nested file
    nestedArchive = zipfile.ZipFile(fileToExtract, 'r')
    for file in nestedArchive.namelist():
        nestedArchive.extract(fileToExtract, directoryToExtractTo)

但我收到此错误：

KeyError: "There is no item named 'nestedFileToExtract.zip' in the archive"

即使它存在于文件系统中

Answer 1

基于此其他解决方案： this和this 。

import os
import io
import sys
import zipfile


def extract_with_structure(input_file, output):
    with zipfile.ZipFile(input_file) as zip_file:
        print(f"namelist: {zip_file.namelist()}")
        for obj in zip_file.namelist():
            filename = os.path.basename(obj)

            if not filename:
                # Skip folders
                continue

            if 'zip' == filename.split('.')[-1]:
                # extract a zip
                content = io.BytesIO(zip_file.read(filename))
                f = zipfile.ZipFile(content)
                dirname = os.path.splitext(os.path.join(output, filename))[0]
                for i in f.namelist():
                    f.extract(i, dirname)
            else:
                # extract a file
                zip_file.extract(obj, os.path.join(output))


if __name__ == "__main__":
    if len(sys.argv) < 3:
        print("No zipfile specified or output folder.")
        exit(1)
    
    extract_with_structure(sys.argv[1], sys.argv[2])