1

我对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")

当 PyQt 应用程序（项目）运行时，将对象ContactGeometry数据保存到 pickle 文件。函数write()被调用moduleC.py：

模块C.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")

保存对象并创建泡菜文件。但是，当我仅运行文件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使用类执行得更好，但我在实现它时遇到了问题。我还发现我必须在文件__reduce__()中的类中实现。ContactGeometrycontact_geometry.py

contact_geometry.py

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

但我不确定应该返回什么方法？如何从当前情况成功加载泡菜文件？

以下是项目结构，如果有帮助的话。

在此处输入图像描述

4

2 回答 2

3

我是dill作者。很难说你是如​​何运行代码的，但看起来问题是你运行代码的一种方式和@Antti Haapala 的答案中的模块名称。他的建议也是很好的遵循。

我会添加这个...您需要确保 ( 1 )moduleA.moduleB.contact_geometry在）。PYTHONPATH__main__.moduleB.contact_geometrymoduleA.moduleB.contact_geometrydill__main__

但是，您不需要向__reduce__您的类添加方法。

于 2015-02-20T22:47:23.207 回答
1

你不能从这样的包中运行 python 文件；它不会找到顶级包名称。我会提出以下任何一项：

  1. 在顶层（在哪里main.py）编写一个启动脚本，导入并运行read_write_dillfrommoduleA.moduleB

  2. 相反，在顶级目录中main，您可以使用python -m moduleA.moduleB.dill_read_write.

  3. 或者，我首选的替代方案是setup.py为您的项目编写一个并为该实用程序编写一个脚本。

于 2015-02-20T22:25:40.410 回答