Question

我正在为库存系统编写Python CGI脚本。它需要通过pickle存储对象的列表（称为locations）。这是我正在使用的代码：

try:
    with open(".config/autosave.bin", "rb") as dataFile:
        locations = pickle.load(dataFile)
except (FileNotFoundError, PermissionError):
    dispHTML("p", contents="Error in load:  Save file incorrectly configured!")
    locations = []

然而，这导致：

 /Applications/MAMP/cgi-bin/ic/main.py in ()
     16 try:
     17         with open(".config/autosave.bin", "rb") as dataFile:
=>   18                 locations = pickle.load(dataFile)
     19 except (FileNotFoundError, PermissionError):
     20         dispHTML("p", contents="Error in load:  Save file incorrectly configured!")
AttributeError: Can't get attribute 'Location' on <module '__main__' from '/Applications/MAMP/cgi-bin/ic/main.py'> 
      args = ("Can't get attribute 'Location' on <module '__main__' from '/Applications/MAMP/cgi-bin/ic/main.py'>",) 
      with_traceback = <built-in method with_traceback of AttributeError object>

如您所见，保存文件存储在.config/autosave.bin。写入似乎工作正常，但我无法检查。

我该如何解决这个问题？

Answer 1

pickle阅读代码需要Location类的定义。如果不是，它将无法重建该类的自定义对象。

# config_writer.py

import pickle

class Location:
    def __init__(self, store, aisle):
        self.store = store
        self.aisle = aisle

locations = [Location(i, i) for i in range(10)]
with open('.config/autosave.bin', 'wb') as f:
    pickle.dump(locations, f)

这是一个尝试在没有Location的类定义的情况下读取pickle文件的示例（在另一个终端/会话中运行此代码）：

>>> import pickle
>>> with open('.config/autosave.bin','rb') as f:
...     data = pickle.load(f)
... 
Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
AttributeError: Can't get attribute 'Location' on <module '__main__' (built-in)>

但是，可以访问类定义：

>>> from config_writer import Location
>>> with open('.config/autosave.bin','rb') as f:
...     data = pickle.load(f)
>>> data
[<config_writer.Location object at 0x7f8b472111d0>, <config_writer.Location object at 0x7f8b41ad6e48>, <config_writer.Location object at 0x7f8b41adb0f0>, <config_writer.Location object at 0x7f8b41adb128>, <config_writer.Location object at 0x7f8b41adb160>, <config_writer.Location object at 0x7f8b41adb198>, <config_writer.Location object at 0x7f8b41adb1d0>, <config_writer.Location object at 0x7f8b41adb208>, <config_writer.Location object at 0x7f8b41adb240>, <config_writer.Location object at 0x7f8b41adb278>]

希望读取pickle文件的代码能够从我的示例中导入Location的其他模块的类定义。

unpickling时的AttributeError：无法获取属性'Location'

1 个答案: