python的struct模块

  了解c语言的人,必定会知道struct结构体在c语言中的做用,它定义了一种结构,里面包含不一样类型的数据(int,char,bool等等),方便对某一结构对象进行处理。而在网络通讯当中,大多传递的数据是以二进制流(binary data)存在的。当传递字符串时,没必要担忧太多的问题,而当传递诸如int、char之类的基本数据的时候,就须要有一种机制将某些特定的结构体类型打包成二进制流的字符串而后再网络传输,而接收端也应该能够经过某种机制进行解包还原出原始的结构体数据。python中的struct模块就提供了这样的机制,该模块的主要做用就是对python基本类型值与用python字符串格式表示的C struct类型间的转化。再者, 有的时候须要用python处理二进制数据,好比,存取文件,socket操做时.这时候,可使用python的struct模块来完成.能够用 struct来处理c语言中的结构体. struct模块中最重要的三个函数是pack(), unpack(), calcsize()python


#  按照给定的格式(fmt),把数据封装成字符串(其实是相似于c结构体的字节流) 网络

pack(fmt, v1, v2, ...) app

 # 按照给定的格式(fmt)解析字节流string,返回解析出来的tuple socket

unpack(fmt, string)      函数

# 计算给定的格式(fmt)占用多少字节的内存 ui

calcsize(fmt)google


struct中支持的format格式

注1.q和Q只在机器支持64位操做时有意思spa

注2.每一个格式前能够有一个数字,表示个数指针

注3.s格式表示必定长度的字符串,4s表示长度为4的字符串,可是p表示的是pascal字符串code

注4.P用来转换一个指针,其长度和机器字长相关

注5.最后一个能够用来表示指针类型的,占4个字节

一个format字符串是用数字和上述定义的字符来表示的。如 5i 表示5个 int 型数,H 表示一个 unsigned short 类型的数, 0Q 则表示0个 unsigned long long 类型的数。在上面的表格中,若是表示了字符表明的位数,则这个字符前面再加数字,表示多个变量,如 3Q 表示3个变量;若是没有指明字符表明的位数,则前面加数据表明一个变量,如 3s 表示一个长度为3的字符串,在pack或者unpack的时候须要注意这一点。

struct支持的数据对齐方式

在与其它语言进行数据交互时,一般还要考虑的事情是:编译器是否有字节对齐?大端存储仍是小端存储?在format字段串的第一个字符能够用来定义数据的对齐方式:

例如:

struct.unpack('>I', a)[0]  

format=">I"是什么意思,从google找了一下,有人说这个东西,可都是比较笼统,没能让我明白,因而硬着头皮看API:
By default, C numbers are represented in the machine’s native format and byte order, and properly aligned by skipping pad bytes if necessary (according to the rules used by the C compiler).
一般,C语言下数字都是机器语言的格式而且按照字节排序,同时在须要的状况下会利用跳过填补的字节来进行适当的调整

Alternatively, the first character of the format string can be used to indicate the byte order, size and alignment of the packed data。
非此即彼:字符串的第一个字符要么被用于表示字符串的字节的排序,或者是字符串的size,还有就是数据是否对准。

Native byte order is big-endian or little-endian, depending on the host system. For example, Motorola and Sun processors are big-endian; Intel and DEC processors are little-endian.
计算机的字节序要么是高位顺序,要么是低位的,这依赖于主机自己。好比,摩托罗拉和sun的处理器是高位的,可是intel和DEC的是低位的。

这样子就明白了上面的format=">I"的意思,也就是说按照高位顺序来格式化取得一个int或long值。下面问题就又来了,你怎么知道读取的就是一个int或long值呢?

经过看struct的文档,能够看到struct经过两张表制定了必定的format规则,我按照本身的观察,给他概括为两类,一个是和C当中类型的对照,另外一个就是选择按照高位仍是低位来解释字节。上面已经说了高低字节顺序,那么观察和C对照的表格,发现I 表明的就是integer or long ,详细的能够去看python的API。

使用示例

示例1: C结构体打包和解包

一个C结构体:

struct Header {    
 unsigned short id;   
 char[4] tag;    
 unsigned int version;  
 unsigned int count; }

如今接收到一个上述结果体数据,能够用unpack函数解析:

import struct (id, tag, version, count) = struct.unpack('!H4s2I', s)

在上述代码中,H对应id,4s对应tag,2I对应version和count。!表示是由网络字节顺序传输。

这样,经过unpack,就解出对应的信息。一样,经过pack也能够将信息打包也对应的二进制格式:

import struct ss = struct.pack("!H4s2I", id, tag, version, count);

pack函数将信息打包成一个字符串,其实是类C结果体字节流,表示的就是一个Header结构体。

示例2: 格式化类型和二进制的转换

import struct 
a=12.34 
#将a变为二进制 
bytes=struct.pack('i',a)

pack后,bytes就是一个str字符串,内容与a的二进制存储内容相同

能够用以下代码进行反转换:

(a,) = struct.unpack('i', bytes)

注意:unpack返回的是一个tuple。


# 使用struck.unpack获取子字符串 ,取前5个字符,跳过4个字符华,再取3个字符 
format = '5s 4x 3s'

print struct.unpack(format, 'Test astring') 
#('Test', 'ing') 

来个简单的例子吧,有一个字符串'He is not very happy',处理一下,把中间的not去掉,而后再输出。 
import struct 
theString = 'He is not very happy' 
format = '2s 1x 2s 5x 4s 1x 5s' 
print ' '.join(struct.unpack(format, theString)) 
输出结果: 
He is very happy


随后是关于网络字节的东东,从网上看来的,感受有用:

Python的socket库采用string类型来发送和接收数据,这样当咱们用
i = socket.recv(4)
来接收一个4字节的整数时,该整数其实是以二进制的形式保存在字符串 i 的前4个字节中;大多数的时候咱们须要的是一个真正的integer/long型,而不是一个用string型表示的整型。这时咱们可使用struct库:Interpret

strings as packed binary data. 对上面的状况,咱们能够写
t = unpack("I", i)
第一个参数是格式化字符串,I指明字符串 i 包含的头一个数据项是一个以C语言的unsigned integer表示的整数,这里 i 只包含了一个数据项,实际上这个被解释的字符串也能够包含多个数据项,只要在格式化字符串里为每项数据指明一个格式便可;天然地,unpack返回的就是一个tuple类型了。




可能遇到的错误

1. struct.pack时可能会遇到:> struct.error: pack requires exactly 2 arguments

这一错误说明format中的参数个数与实际输入的参数个数不符。如 bytes = struct.pack('2I', a) 须要2个参数而只输入了一个

2. struct.unpack时可能会遇到:> struct.error: unpack requires a string argument of length *

这一错误说明format中参数表明的数据占用内存数,与输入的数据占用的内存长度不符。这时候能够用 struct.calcsize(fmt) 函数检查format字符表明的长度,及用len(string)函数来检测数据的长度。

相关文章
相关标签/搜索