Python中的二进制缓冲区

Question

在Python中,您可以将StringIO用于字符数据的类文件缓冲区.内存映射文件基本上对二进制数据做类似的事情,但它需要一个用作基础的文件.Python是否有一个用于二进制数据的文件对象,并且只是内存,相当于Java的ByteArrayOutputStream？

我的用例是我想在内存中创建一个ZIP文件,ZipFile需要一个类似文件的对象.

Answer 1

您可能正在寻找io.BytesIO课程.它的工作方式与StringIO完全相同,只是它支持二进制数据:

from io import BytesIO
bio = BytesIO(b"some initial binary data: \x00\x01")

StringIO将抛出TypeError:

from io import StringIO
sio = StringIO(b"some initial binary data: \x00\x01")

Answer 2

只要您不尝试将任何unicode数据放入您的手中StringIO,并且小心不要使用,cStringIO您应该没问题.

根据StringIO文档,只要你保持unicode或8位,一切都按预期工作.据推测,StringIO当有人做某事时f.write(u"asdf")(根据我的知识,ZipFile不做)会做一些特别的事情.无论如何;

import zipfile
import StringIO

s = StringIO.StringIO()
z = zipfile.ZipFile(s, "w")
z.write("test.txt")
z.close()
f = file("x.zip", "w")
f.write(s.getvalue())
s.close()
f.close()

按预期工作,生成的存档中的文件与原始文件之间没有区别.

如果您知道这种方法不起作用的特定情况,我最感兴趣的是听到它:)