python - 取消 pickle 错误: Invalid load key'(heart)'

Question

我已经花了几天时间摆弄 pickle，试图将其应用到“猜数字”练习计划的高分系统中。我以为我已经正确理解了这个概念，但现在出现了这个错误，我不知道为什么。

相关代码如下:

def EnterHighScore(score,scoresList):
    name = input("Enter your name: ")
    newPlayer = player(name,score)
    scoresList.append(newPlayer)
    scoresFile = open('scores','wb')
    pickle.dump(scoresList,scoresFile)
    scoresFile.close()

    for i in scoresList:
        print(i.name + ' - ' + str(i.score))

def CheckHighScores(score):
    try:
        scoresFile = open('scores','rb')
    except:
        scoresFile = open('scores','wb+')

    if not scoresFile.read(1):
        scoresList = []
    else:
        scoresList = pickle.load(scoresFile)
    scoresFile.close()

    if not scoresList:
        EnterHighScore(score,scoresList)
    else:
        for counter,i in enumerate(scoresList):
            if counter == 3:
                break
            if score >= i.score:
                EnterHighScore(score,scoresList)
                break

当我运行它时，第一次运行顺利。也就是说，当“分数”文件根本不存在时。它被正确创建，scoresList被创建为空，然后填充一个玩家对象，并且它被转储到scoresFile中，没有任何错误。但是，当我尝试使用新的“分数”文件数据加载 ScoreList 时，出现以下错误:

UnpicklingError: Invalid load key'(heart)'

(心)代表真正的心形角色。

我读到其他人也遇到过这个问题，但在这些情况下，他们试图在不同的操作系统中打开文件，或者在 pickle 之后但在取消 pickle 之前以某种方式修改了文件。在这种情况下，文件根本没有被修改，只是写入并关闭。

我尝试在其他更简单的场景中使用 pickle，并且没有引起其他错误。

任何帮助将不胜感激。

Answer 1

查看文件是否为空的测试将文件读取指针推进到文件开头:

if not scoresFile.read(1):

你必须回到起点:

if not scoresFile.read(1):
    scoresList = []
else:
    scoresFile.seek(0)
    scoresList = pickle.load(scoresFile)

更好的测试是捕获 pickle.load() 在文件为空时抛出的 EOFError 异常:

try:
    scoresList = pickle.load(scoresFile)
except EOFError:
    # File empty
    scoresList = []

或者当文件不存在时您可以捕获IOError:

try:
    with open('scores','rb') as scoresFile:
        scoresList = pickle.load(scoresFile)
except IOError:
    scoresList = []

并且只是不打开文件进行写入。