Python struct.unpack不工作

Question

我在试着做这个：

def ReadWord(fid,fmt,Addr):
    fid.seek(Addr)
    s = fid.readline(2)
    s = unpack(fmt + 'h', s)
    if(type(s) == tuple):
        return s[0]
    else:
        return s    

通过以下方式：

len(s) = 2
len(fmt) = 1
calcsize(fmt) = 0
calcsize(fmt + 'h') = 2

然而，Python返回：

struct.error:解包需要长度为4的字符串参数

根据python struct.unpack 文档的说法：

字符串必须准确地包含格式所需的数据量(len( string )必须等于calcsize(fmt))。

那么，如果我的字符串的长度是2，而fmt+'h'的煅烧大小也是2，那么为什么python说“解包需要长度为4的字符串参数”呢？

编辑:

谢谢你的回答。以下是完整的代码：

模块/一般/lecroy.py

因此，正如您在read_timetrace函数中看到的那样，fmt在if...else语句中被设置为'<'或'>'。打印证实了这一点。

但是你也应该知道我在做windowsx64 (为了工作)。

EDIT2

这是完整的回溯，很抱歉弄错了。

Traceback (most recent call last):
  File "C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Include\readLecroyTRCFile.py", line 139, in <module>
    read_timetrace("C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Traces\KL.ES.001.001.trc")
  File "C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Include\readLecroyTRCFile.py", line 60, in read_timetrace
    WAVE_ARRAY_1        = ReadLong(fid, fmt, aWAVE_ARRAY_1)
  File "C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Include\readLecroyTRCFile.py", line 100, in ReadLong
    s = unpack(fmt + 'l', s)
struct.error: unpack requires a string argument of length 4
[Finished in 0.2s]

EDIT3:

我将readline替换为read，并添加：

print "len(s) ", len(s)
print "len(fmt) ", len(fmt)
print "calcsize(fmt) ", calcsize(fmt)
print "calcsize(fmt + 'h') ", calcsize(fmt + 'h')
print "fmt ", fmt

转到ReadLong函数。

这是新的回溯：

len(s)  4
len(fmt)  1
calcsize(fmt)  0
calcsize(fmt + 'h')  2
fmt  <
len(s)  4
len(fmt)  1
calcsize(fmt)  0
calcsize(fmt + 'h')  2
fmt  <
len(s)  4
len(fmt)  1
calcsize(fmt)  0
calcsize(fmt + 'h')  2
fmt  <
len(s)  1
len(fmt)  1
calcsize(fmt)  0
calcsize(fmt + 'h')  2
fmt  <
Traceback (most recent call last):
  File "C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Include\readLecroyTRCFile.py", line 143, in <module>
    read_timetrace("C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Traces\KL.ES.001.001.trc")
  File "C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Include\readLecroyTRCFile.py", line 60, in read_timetrace
    WAVE_ARRAY_1        = ReadLong(fid, fmt, aWAVE_ARRAY_1)
  File "C:\Users\maxime.vast\Desktop\Test Campaign Template\Test Suite\Include\readLecroyTRCFile.py", line 104, in ReadLong
    s = unpack(fmt + 'l', s)
struct.error: unpack requires a string argument of length 4
[Finished in 0.2s]

Answer 1

FWIW，您应该使用read(2)，而不是readline(2)。如果fmt字符串确实是'>'，则不应该收到该错误。这里有一个简短的演示，如预期的那样执行。

from struct import unpack

fname = 'qbytes'

#Create a file of all byte values
with open(fname, 'wb') as f:
    f.write(bytearray(range(256)))

def ReadWord(fid, fmt, addr):
    fid.seek(addr)
    s = fid.read(2)
    s = unpack(fmt + 'h', s)
    return s[0]

fid = open(fname, 'rb')

for i in range(16):
    addr = i
    n = 256*i + i+1
    #Interpret file data as big-endian
    print i, ReadWord(fid, '>', addr), n

fid.close()

输出

0 1 1
1 258 258
2 515 515
3 772 772
4 1029 1029
5 1286 1286
6 1543 1543
7 1800 1800
8 2057 2057
9 2314 2314
10 2571 2571
11 2828 2828
12 3085 3085
13 3342 3342
14 3599 3599
15 3856 3856

顺便说一下，struct.unpack()总是返回一个元组，即使返回值是单个项。

在二进制文件上使用readline(2)可能会产生意想不到的结果。在我的测试文件中，在上面的代码中，文件中有一个(Linux风格的)换行符\xa0。因此，如果您将s = fid.read(2)更改为s = fid.readline(2)，那么一切一开始都很好，但是在第10行，它会崩溃，因为它只读取一个字节，这是因为换行符：

from struct import unpack

fname = 'qbytes'

#Create a file of all byte values
with open(fname, 'wb') as f:
    f.write(bytearray(range(256)))

def ReadWord(fid, fmt, addr):
    fid.seek(addr)
    s = fid.readline(2)
    print repr(s),
    s = unpack(fmt + 'h', s)
    return s[0]

with open(fname, 'rb') as fid:
    for i in range(16):
        addr = i
        n = 256*i + i+1
        #Interpret file data as big-endian
        print i, ReadWord(fid, '>', addr), n

输出

0 '\x00\x01' 1 1
1 '\x01\x02' 258 258
2 '\x02\x03' 515 515
3 '\x03\x04' 772 772
4 '\x04\x05' 1029 1029
5 '\x05\x06' 1286 1286
6 '\x06\x07' 1543 1543
7 '\x07\x08' 1800 1800
8 '\x08\t' 2057 2057
9 '\t\n' 2314 2314
10 '\n'
Traceback (most recent call last):
  File "./qtest.py", line 30, in <module>
    print i, ReadWord(fid, '>', addr), n
  File "./qtest.py", line 22, in ReadWord
    s = unpack(fmt + 'h', s)
struct.error: unpack requires a string argument of length 2

后记

您的代码中有几个函数几乎可以完成相同的任务。这违反了干的原则:不要重复自己。这里有一种修复方法，使用部分函数应用程序。有关更多信息，请参见函数式文档。

from functools import partial

def ReadNumber(fid, datalen=1, fmt='>', conv='b', addr=0):
    fid.seek(addr)
    s = fid.read(datalen)
    if len(s) != datalen:
        raise IOError('Read %d bytes but expected %d at %d' % (len(s), datalen, addr)) 
    return unpack(fmt+conv, s)[0]

ReadByte = partial(ReadNumber, datalen=1, conv='b') 
ReadWord = partial(ReadNumber, datalen=2, conv='h') 
ReadLong = partial(ReadNumber, datalen=4, conv='l') 
ReadFloat = partial(ReadNumber, datalen=4, conv='f') 
ReadDouble = partial(ReadNumber, datalen=8, conv='d') 

您需要使用关键字来调用这些新函数。例如，

ReadLong(fid, fmt='>', addr=addr)

的确，这稍微长了一点，但它使代码更加可读性。

Answer 2

格式的长度本身并不重要。重要的是您在那里指定了哪种格式。例如，有指定一个字节甚至八个字节的格式规范。因此，这实际上取决于s中应该有多少字符的格式。

例如：

>>> struct.unpack('b', 'A')
(65,)
>>> struct.unpack('L', 'A')

Traceback (most recent call last):
  File "<pyshell#3>", line 1, in <module>
    struct.unpack('L', 'A')
error: unpack requires a string argument of length 4
>>> struct.unpack('L', 'AAAA')
(1094795585,)

如果fmt真的如您所说的那样是>，那么它应该工作得很好：

>>> struct.unpack('>h', 'AA')
(16705,)

因此，我假设当错误出现时，fmt不仅仅是>，还会消耗额外的2个字节。尝试在fmt之前打印unpack。

Answer 3

当len(fmt) = 1时，它意味着fmt具有值。如果fmt = 'h'，那么fmt+'h'将是'hh‘。因此，unpack()需要4个字节的数据，因为每个'h‘都需要一个短整数(2个字节)。