Python, запись в память zip в файл
Как записать в память zipfile в файл?
# 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()
3 ответов
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())
изменить ответ falsetru для python3
1) использовать io.StringIO
вместо StringIO.StringIO
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 с помощью gzip?
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-файл считывает поврежденные и вы замечаете отсутствующие символические ссылки или поврежденные файлы с неправильными метками времени, это может быть тот факт, что свойства файла не копируются.
приведенный выше фрагмент кода-это то, как я решил проблему.