python - msgpack 将 dict 键字符串反序列化为字节

Question

我在 python 中遇到 msgpack 问题。似乎在序列化 dict 时，如果键是字符串 str，则它们未正确反序列化并导致引发 KeyError 异常。

例子:

>>> import msgpack
>>> d = dict()
>>> value = 1234
>>> d['key'] = value
>>> binary = msgpack.dumps(d)
>>> new_d = msgpack.loads(binary)
>>> new_d['key']
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
KeyError: 'key'

这是因为调用 loads() 后键不是字符串，而是反序列化为 bytes 对象。

>>> d.keys()
dict_keys(['key'])
>>> new_d.keys()
dict_keys([b'key'])

这似乎与提到的未实现功能有关 in github

我的问题是，是否有解决此问题的方法或变通方法以确保在反序列化时可以使用相同的 key ？

我想使用 msgpack 但如果我不能用 str 键构建一个 dict 对象并期望能够使用相同的对象反序列化后的 key ，它变得无用。

Answer 1

调用dumps 或packb 时会设置默认编码

:param str encoding:
 |      Convert unicode to bytes with this encoding. (default: 'utf-8')

但在调用 loads 或 unpackb 时默认情况下未设置，如下所示:

Help on built-in function unpackb in module msgpack._unpacker:

unpackb(...)
    unpackb(... encoding=None, ... )

因此更改反序列化的编码可以解决问题，例如:

>>> d['key'] = 1234
>>> binary = msgpack.dumps(d)
>>> msgpack.loads(binary, encoding = "utf-8")
{'key': 1234}
>>> msgpack.loads(binary, encoding = "utf-8") == d
True