问题引入:在阅读《Python语言及其应用》(2016年出版)一书时,在第七章 7.2.3节中介绍使用第三方库construct处理二进制数据时,给出了一个例子:使用construct库处理PNG图片的二进制数据,获取PNG图片的尺寸。书上给出的代码示例如下:
from construct import Struct, Magic, UBInt32, Const, String
#基于https://github.com/construct上的代码修改而来
fmt = Struct('png',
Magic(b'\x89PNG\r\n\x1a\n'),
UBInt32('length'),
Const(String('type', 4), b'IHDR'),
UBInt32('width'),#获取4字节的宽度
UBInt32('height')#获取4字节的高度
)
#b'\x89PNG\r\n\x1a\n\x00\x00\x00\r',这标志着这是一个有效的PNG格式文件
#PNG图片的30字节数据,width存放在第16~20字节(\x00\x00\x00\x9a),height存放在第21~24字节(\x00\x00\x00\x8d)
data = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR' + \
b'\x00\x00\x00\x9a\x00\x00\x00\x8d\x08\x02\x00\x00\x00\xc0'
result = fmt.parse(data)
print(result)
运行结果如下:
但是,我在自己的的环境(系统:Mac OS,python版本:3.7,construct版本:2.9.51)中运行时,运行出错,错误信息如下:
后来通过查阅construct官方文档(链接:construct官网)才知道,construct 2.9对之前的版本进行了大量修改,用之前的版本写的代码如果用2.9版本需要进行代码修改。
问题解决:针对上面的问题,通过参考官方文档,我对代码进行了修改,修改后的代码如下:
from construct import *
fmt = Struct(
Const(b'\x89PNG\r\n\x1a\n'),
"length" / Int32ub,
"type" / Const(b'IHDR'),
"width" / Int32ub,#获取4字节的宽度
"height" / Int32ub,
)
#Struct结构体的顺序要和PNG的二进制数据对应,不然容易出错
data = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR' + \
b'\x00\x00\x00\x9a\x00\x00\x00\x8d\x08\x02\x00\x00\x00\xc0'
result = fmt.parse(data)
print(result)
print(result.width, result.height)
运行结果如下: