Python模块——struct（字节流，组包拆包实现）

我们知道python只定义了6种数据类型，字符串，整数，浮点数，列表，元组，字典。但是C语言中有些字节型的变量，在python中该如何实现呢？这点颇为重要，特别是要在网络上进行数据传输的话。

   python提供了一个struct模块来提供转换。下面就介绍这个模块中的几个方法。

   struct.pack():

   struct.pack用于将Python的值根据格式符，转换为字符串（因为Python中没有字节(Byte)类型，可以把这里的字符串理解为字节流，或字节数组）。其函数原型为：struct.pack(fmt,v1, v2, ...)，参数fmt是格式字符串，关于格式字符串的相关信息下面有所介绍。v1, v2,...表示要转换的python值。下面的例子将两个整数转换为字符串（字节流）:

>>> importstruct
>>> a=20
>>> b=400
>>>str=struct.pack('ii',a,b)   #转换成字节流，虽然还是字符串，但是可以在网络上传输
>>>printlen(str)              #ii表示两个int
8                               #可以看到长度为8个字节，正好是两个int型数据的长度
>>>print str
                              #二进制是乱码
>>>print repr(str)
'\x14\x00\x00\x00\x90\x01\x00\x00'  #其中十六进制的 0x00000014,0x00001009分别表示20和400
>>>

  由此我们就可以任意的进行组包了，比如下面一个打包的例子，只介绍其中的pack

 

format = "!HH%ds" % len(data)
buffer = struct.pack(format,opcode,blocknumber,data)

 我们要对一个数据进行打包，加上一些个包头，我们根据下面的格式符信息，知道H是unsignedshort是2个字节，而s是char型。所以这个buffer就是2个字节的opcode，2个字节的blocknumber，和len长的char。

 

   struct.unpack():

  我们接着上面的例子运行：

>>>a1,a2=struct.unpack('ii',str)
>>> print'a1',a1
a1 20
>>> print'a2=',a2
a2= 400

可以看到 “ii”以四个字节为分界，把8个字节的str分成了两个int型的整数。

 struct.calcsize():用来计算特定格式的输出的大小，是几个字节，比如：

>>>struct.calcsize('HH4s')
8
>>>struct.calcsize('ii')
8
>>>

>>>format='!HH%ds' % len('hello python')
>>>struct.calcsize(format)
16
>>>