PYTHON如何在內存中生成ZIP文件

ybw8 9年前發布 | 14K 次閱讀 Python開發 Python

如題,代碼如下:

class MemoryZipFile(object):
   def __init__(self):
       #創建內存文件
       self._memory_zip = StringIO.StringIO()

   def append_content(self, filename_in_zip, file_content):
       """
       description: 寫文本內容到zip
       """
       zf = zipfile.ZipFile(self._memory_zip, "a", zipfile.ZIP_DEFLATED, False)
       zf.writestr(filename_in_zip, file_content)
       for zfile in zf.filelist: zfile.create_system = 0
       return self

   def append_file(self, filename_in_zip, local_file_full_path):
       """


      description:寫文件內容到zip

      注意這里的第二個參數是本地磁盤文件的全路徑(windows:c/demo/1.jpg | linux: /usr/local/test/1.jpg)

      """

      zf = zipfile.ZipFile(self._memory_zip, "a", zipfile.ZIP_DEFLATED, False)

      zf.write(local_file_full_path, filename_in_zip)

      for zfile in zf.filelist: zfile.create_system = 0      

      return self

     

  def read(self):

      """

      description: 讀取zip文件內容

      """

      self._memory_zip.seek(0)

      return self._memory_zip.read()

 

  def write_file(self, filename):

      """

      description:寫zip文件到磁盤

      """

      f = file(filename, "wb")

      f.write(self.read())

      f.close()





使用方法如下:


        mem_zip_file = MemoryZipFile()
        mem_zip_file.append_content('mimetype', "application/epub+zip")
        mem_zip_file.append_content('META-INF/container.xml', '''<?xml version="1.0" encoding="UTF-8" ?>
<container version="1.0" xmlns="urn:oasis:names:tc:opendocument:xmlns:container"> </container>''');
        #追加磁盤上的文件內容到內存,注意這里的第二個參數是本地磁盤文件的全路徑(windows:c/demo/1.jpg | linux: /usr/local/test/1.jpg)
        mem_zip_file.append_file("1.jpg", "c:\1.jpg")


        #將內存中的zip文件寫入磁盤
        mem_zip_file.write_file("c:test.zip")

        #獲取內存zip文件內容
        data = mem_zip_file.read()

        #上傳到fdfs
        my_fdfs_client.upload_by_buffer(data, 'zip')


 本文由用戶 ybw8 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!