如何在 python 3 中读取在 python 2.7 或 3 中复制的泡菜？

Question

• I have read that I can read pickles that were dumped in python 2.7 in python 3 using

  content = pickle.load(o, encoding='latin1')

• Obviously, I can read pickles that were dumped in python 3 using

  content = pickle.load(o)

---

My problem is, I can't know the source of my pickle. It could be either one.

How can I test which type of pickle I am trying to read in order to use the correct method?

Answer 1

inspired by @DeepSpace:

def load(path):
    print(f"loading pkl: {os.path.abspath(path)}")
    assert os.path.isfile(path)
    try:
        with open(path, "rb") as f:
            content = pickle.load(f)
    except UnicodeDecodeError:  # pickle was created with python 2.7
        with open(path, "rb") as f:
            content = pickle.load(f, encoding='latin1')

    return content

Answer 2

使用 if else: 尝试在没有编码的情况下进行正常加载并进行一些读取，如果成功则使用 encoding='latin1' 进行 else 加载