Come estrarre file zip in modo ricorsivo in Python

Question

Ho un file zip che contiene tre file zip in questo modo:

zipfile.zip\ 
    dirA.zip\ 
     a 
    dirB.zip\ 
     b 
    dirC.zip\ 
     c 

voglio estrarre tutti i file zip interiori che sono all'interno del file zip nella directory con questi nomi (dirA, dirB, dirC).
Fondamentalmente, io voglio finire con il seguente schema:

output\ 
    dirA\ 
     a 
    dirB\ 
     b 
    dirC\ 
     c 

Ho provato quanto segue:

import os, re 
from zipfile import ZipFile 

os.makedirs(directory) # where directory is "\output" 
with ZipFile(self.archive_name, "r") as archive: 
    for id, files in data.items(): 
     if files: 
      print("Creating", id) 
      dirpath = os.path.join(directory, id) 

      os.mkdir(dirpath) 

      for file in files: 
       match = pattern.match(filename) 
       new = match.group(2) 
       new_filename = os.path.join(dirpath, new) 

       content = archive.open(file).read() 
      with open(new_filename, "wb") as outfile: 
       outfile.write(content) 

Ma estrae solo il file zip e io alla fine con:

output\ 
    dirA\ 
     dirA.zip 
    dirB\ 
     dirB.zip 
    dirC\ 
     dirC.zip 

Qualsiasi suggerimento compresi i segmenti di codice sarà molto apprezzato perché ho provato così tante diverse t le ali e leggere i documenti senza successo.

Answer 1

Durante l'estrazione del file zip, è necessario scrivere i file zip interni in memoria anziché su disco. Per fare questo, ho usato BytesIO.

Partenza di questo codice:

import os 
import io 
import zipfile 

def extract(filename): 
    z = zipfile.ZipFile(filename) 
    for f in z.namelist(): 
     # get directory name from file 
     dirname = os.path.splitext(f)[0] 
     # create new directory 
     os.mkdir(dirname) 
     # read inner zip file into bytes buffer 
     content = io.BytesIO(z.read(f)) 
     zip_file = zipfile.ZipFile(content) 
     for i in zip_file.namelist(): 
      zip_file.extract(i, dirname) 

Se si esegue extract("zipfile.zip") con zipfile.zip come:

zipfile.zip/ 
    dirA.zip/ 
     a 
    dirB.zip/ 
     b 
    dirC.zip/ 
     c 

uscita dovrebbe essere:

dirA/ 
    a 
dirB/ 
    b 
dirC/ 
    c 

Answer 2

Per una funzione che estrae una zip nidificato file (qualsiasi livello di nidificazione) e pulisce i file zip originali:

import zipfile, re, os 

def extract_nested_zip(zippedFile, toFolder): 
    """ Extract a zip file including any nested zip files 
     Delete the zip file(s) after extraction 
    """ 
    with zipfile.ZipFile(zippedFile, 'r') as zfile: 
     zfile.extractall(path=toFolder) 
    os.remove(zippedFile) 
    for root, dirs, files in os.walk(toFolder): 
     for filename in files: 
      if re.search(r'\.zip$', filename): 
       fileSpec = os.path.join(root, filename) 
       extract_nested_zip(fileSpec, root)