Python, записывать в zip файл в файл

Как записать zip файл в память в файл?

# Create in memory zip and add files
zf = zipfile.ZipFile(StringIO.StringIO(), mode='w',compression=zipfile.ZIP_DEFLATED)
zf.writestr('file1.txt', "hi")
zf.writestr('file2.txt', "hi")

# Need to write it out
f = file("C:/path/my_zip.zip", "w")
f.write(zf)  # what to do here? Also tried f.write(zf.read())

f.close()
zf.close()

Ответ 1

StringIO.getvalue вернуть содержимое StringIO:

>>> import StringIO
>>> f = StringIO.StringIO()
>>> f.write('asdf')
>>> f.getvalue()
'asdf'

В качестве альтернативы вы можете изменить положение файла с помощью seek:

>>> f.read()
''
>>> f.seek(0)
>>> f.read()
'asdf'

Попробуйте следующее:

mf = StringIO.StringIO()
with zipfile.ZipFile(mf, mode='w', compression=zipfile.ZIP_DEFLATED) as zf:
    zf.writestr('file1.txt', "hi")
    zf.writestr('file2.txt', "hi")

with open("C:/path/my_zip.zip", "wb") as f: # use `wb` mode
    f.write(mf.getvalue())

Ответ 2

Изменить ответ falsetru для python3

1) используйте io.StringIO вместо StringIO.StringIO

StringIO в python3

2) используйте b"abc" вместо "abc" или

python 3.5: TypeError: необходим байтовый объект, а не "str" при записи в файл

3) кодировать в двоичную строку str.encode(s, "utf-8")

Лучший способ преобразования строки в байты в Python 3?

import zipfile
import io
mf = io.BytesIO()

with zipfile.ZipFile(mf, mode="w",compression=zipfile.ZIP_DEFLATED) as zf:

    zf.writestr('file1.txt', b"hi")

    zf.writestr('file2.txt', str.encode("hi"))
    zf.writestr('file3.txt', str.encode("hi",'utf-8'))


with open("a.txt.zip", "wb") as f: # use `wb` mode
    f.write(mf.getvalue())

Это также должно работать для gzip: Как мне сжать сжатие строки в Python?

Ответ 3

  with ZipFile(read_file, 'r') as zipread:
        with ZipFile(file_write_buffer, 'w', ZIP_DEFLATED) as zipwrite:
            for item in zipread.infolist():
                # Copy all ZipInfo attributes for each file since defaults are not preseved
                dest.CRC = item.CRC
                dest.date_time = item.date_time
                dest.create_system = item.create_system
                dest.compress_type = item.compress_type
                dest.external_attr = item.external_attr
                dest.compress_size = item.compress_size
                dest.file_size = item.file_size
                dest.header_offset = item.header_offset

В случае, когда zip файл читается поврежденным, и вы замечаете отсутствующие символические ссылки или поврежденные файлы с неправильными отметками времени, это может быть тот факт, что свойства файла не копируются.

Вышеупомянутый фрагмент кода - это то, как я решил проблему.