pickle,dill ImportError:没有名为moduleA.moduleB.moduleC的模块

2024-06-16 11:55:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我对文件中的pickle.load()有问题。转储和加载在dill_read_write.py中完成:

dill_read_write.py

import os
import dill
from contact_geometry import ContactGeometry

def write_pickle(obj, filename):
    os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__))))
    filename = os.path.join(os.getcwd(), filename)
    with open(filename, 'wb') as output_:
        dill.dump(obj, output_)

def read_pickle(filename):
    with open(filename, 'rb') as input_:
        return dill.load(input_)

if __name__ == "__main__":
    read_pickle("ground_.pkl")

将对象ContactGeometry数据保存到pickle文件是在PyQt应用程序(项目)运行时完成的。函数write()moduleC.py中调用:

模.py

from contact_geometry import ContactGeometry
from moduleA.moduleB import dill_read_write

class Foo(FooItem):
    def __init__(self,...):
        ...
    def createGeometry(self):
        contact_geometry_ = ContactGeometry()
        #   save object to pickle file
        dill_read_write.write_pickle(contact_geometry_, "object_data.pkl") 

保存对象并创建pickle文件。但是,当我只运行文件dill_read_write.py从pickle文件读取(加载)对象数据时,会得到以下错误:

    Traceback (most recent call last):
File "C:\projectName\moduleA\moduleB\dill_read_write.py", line 29, in <module>
read("ground_.pkl")
File "C:\projectName\moduleA\moduleB\dill_read_write.py", line 24, in read
return dill.load(input_)
File "C:\Python27\lib\site-packages\dill-0.2.2-py2.7.egg\dill\dill.py", line 199, in load
obj = pik.load()
File "C:\Python27\Lib\pickle.py", line 858, in load
dispatch[key](self)
File "C:\Python27\Lib\pickle.py", line 1090, in load_global
klass = self.find_class(module, name)
File "C:\Python27\lib\site-packages\dill-0.2.2-py2.7.egg\dill\dill.py", line 278, in find_class
return StockUnpickler.find_class(self, module, name)
File "C:\Python27\Lib\pickle.py", line 1124, in find_class
__import__(module)
ImportError: No module named moduleA.moduleB.contact_geometry

我搜索了一下,发现dill对于类的性能比pickle要好,但是我在实现它时遇到了问题。我还发现我必须在文件contact_geometry.py中的类ContactGeometry中实现__reduce__()

接触几何.py

class ContactGeometry(object):
    def __init__(self, ...):
        ...    
    def __reduce__(self):
        return (self.__class__, (os.path.realpath(__file__))

但我不知道该怎么回报这种方法?如何从当前情况成功加载pickle文件?

下面是项目结构,如果有帮助的话。

enter image description here


Tags: 文件inpyimportselfreadosline