python - pickle 使用 importlib.util 导入的对象

Question

我在使用 Python 的 pickle 时遇到了问题。我需要通过将文件路径提供给 importlib.util 来加载一些 Python 模块，如下所示:

import importlib.util
spec = importlib.util.spec_from_file_location('custom', 'C:\path\to\.py\file.py')
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)

我想从加载的模块中实例化一些对象并将它们序列化以备后用，但是当我尝试时:

pickle.dump(module.ObjectFromModule(), open('C:\object\location\obj.p', 'wb'))

我明白了:_pickle.PicklingError:无法 pickle :模块“自定义”的导入失败

如果我尝试对通过 import 语句导入的对象进行 pickle ，则不会发生这种情况。我怎样才能绕过这个？

Answer 1

使其工作的最简单方法是通过以下方式手动将模块添加到 sys.module:

import importlib.util
import sys

spec = importlib.util.spec_from_file_location('custom', 'C:\path\to\.py\file.py')
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
sys.modules['custom'] = module