web-dev-qa-db-fra.com

Extraire un fichier zip en mémoire?

Comment extraire un Zip en mémoire?

Ma tentative (retour de None sur .getvalue()):

from zipfile import ZipFile
from StringIO import StringIO

def extract_Zip(input_Zip):
    return StringIO(ZipFile(input_Zip).extractall())
pythonmemoryzipzipfilestringio
39
user1438003

extractall extrait dans le système de fichiers, vous n'obtiendrez donc pas ce que vous voulez. Pour extraire un fichier en mémoire, utilisez la méthode ZipFile.read() .

Si vous avez vraiment besoin du contenu complet en mémoire, vous pouvez faire quelque chose comme:

def extract_Zip(input_Zip):
    input_Zip=ZipFile(input_Zip)
    return {name: input_Zip.read(name) for name in input_Zip.namelist()}
63
mata

Travailler fréquemment avec des archives en mémoire dans Python 2 Je recommanderais de créer un outil. Quelque chose comme ceci:

import zipfile
import StringIO

class InMemoryZip(object):
   def __init__(self):
       # Create the in-memory file-like object for working w/imz
       self.in_memory_Zip = StringIO.StringIO()

   # Just Zip it, Zip it
   def append(self, filename_in_Zip, file_contents):
       # Appends a file with name filename_in_Zip and contents of
       # file_contents to the in-memory Zip.
       # Get a handle to the in-memory Zip in append mode
       zf = zipfile.ZipFile(self.in_memory_Zip, "a", zipfile.Zip_DEFLATED, False)

       # Write the file to the in-memory Zip
       zf.writestr(filename_in_Zip, file_contents)

       # Mark the files as having been created on Windows so that
       # Unix permissions are not inferred as 0000
       for zfile in zf.filelist:
           zfile.create_system = 0       

       return self

   def read(self):
       # Returns a string with the contents of the in-memory Zip.
       self.in_memory_Zip.seek(0)
       return self.in_memory_Zip.read()

   # Zip it, Zip it, Zip it
   def writetofile(self, filename):
       # Writes the in-memory Zip to a file.
       f = file(filename, "wb")
       f.write(self.read())
       f.close()

if __== "__main__":
   # Run a test
   imz = InMemoryZip()
   imz.append("testfile.txt", "Make a test").append("testfile2.txt", "And another one")
   imz.writetofile("testfile.Zip")
13
Deviacium