如何识别字节字符串是否是腌制对象？

Question

如何识别字节字符串对象是否为腌制对象。

例如，在下面的字典中，我如何识别b 是腌制对象而其他对象不是？

example = {'a':1, 'b':b'\x80\x03K\x02.', 'c':b'3'}

Answer 1

虽然有一些方法可以判断一个流是否可能是一个泡菜流（例如第一个字节总是\x80，后面是协议号，这里是\x03，最后一个字节也总是b'.'），通常很难确定（第一个字节是随机字符串中的\x80可能是巧合）。所以你最好简单地尝试解开它，看看是否可行：

import pickle

def is_pickle_stream(stream):
    try:
        pickle.loads(stream)
        return True
    except UnpicklingError:
        return False

如果流是pickle流，上述方法将返回True，如果不是，则返回False，如果loads抛出异常（内存不足等），则抛出异常。