Python字节缓冲对象?

时间:2011-07-31 21:54:27

标签: python buffer

Python中是否有一个字节缓冲区对象可以附加特定类型的值? (最好具有可指定的字节顺序)

例如:

buf.add_int(4)    # should add a 4 byte integer
buf.add_short(10) # should add a 2 byte short
buf.add_byte(24)  # should add a byte

我知道我可以使用struct.pack,但这种方法似乎更容易。理想情况下,它应该与 Java 中的DataOutputStreamDataInputStream对象完成相同的任务。

3 个答案:

答案 0 :(得分:10)

您始终可以使用bitstring。它能够完成你要求的所有事情以及更多。

>>> import bitstring
>>> stream=bitstring.BitStream()
>>> stream.append("int:32=4")
>>> stream.append("int:16=10")
>>> stream.append("int:8=24")
>>> stream
BitStream('0x00000004000a18')
>>> stream.bytes
'\x00\x00\x00\x04\x00\n\x18'

以下是documentation的链接。

答案 1 :(得分:5)

正如Kark Knechtel建议你必须制作自己的类型来处理这个问题。这是bytearray的快速扩展:

import struct

class DataStream(bytearray):

    def append(self, v, fmt='>B'):
        self.extend(struct.pack(fmt, v))

>>> x = DataStream()
>>> x.append(5)
>>> x
bytearray(b'\x05')
>>> x.append(-1, '>i')
>>> x
bytearray(b'\x05\xff\xff\xff\xff')

答案 2 :(得分:1)

只需将struct.pack逻辑包装在您自己的类中。