从Python中读取字符串中的字节

Question

我有一个表示从网络读取的字节的python字符串，我需要从该字符串中连续读取几个字节。例如，我在一个字符串中有9个字节，我需要读取4个字节作为整数，2个字节作为短，3个字节的自定义数据类型。

python中是否有读者可以执行以下操作：

reader = reader(my_string)    

integer = int.from_bytes(reader.read(4), 'big')
short = int.from_bytes(reader.read(2), 'big')
custom = customType.Unpack(reader.read(3))

我认为使用struct.unpack，但我不知道如何处理非基本类型。

有什么想法吗？

感谢。

Answer 1

我想你想要这个：

import struct
integer, short = struct.unpack('>ih', my_string)
custom = customType.Unpack(my_string[6:9])

或许这个：

from StringIO import StringIO
reader = StringIO(my_string)
integer, short = struct.unpack('>ih', reader.read(6))
custom = customType.Unpack(reader.read(3))