时间戳

首页 > 解决方案 > 如何在内部转储带有导入的函数

python - 如何在内部转储带有导入的函数

问题描述

我正在尝试将函数转储到文件中，以便可以在其他地方使用此文件/函数。我选择dill而不是pickle，因为我需要依赖项。但是，dill如果函数内部有导入，则不起作用。例如：

def func():
    import numpy

import dill
dill.settings['recurse'] = True 
with open("test.pickle","wb") as f:
    dill.dump(func,f)

当我重新启动并重新加载函数时，出现此错误，

import dill 
func = dill.load(open("test.pickle"))
func()
---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input> in <module>()
      1 import dill
      2 func = dill.load(open("test.pickle"))
----> 3 func()

<ipython-input> in func()

ImportError: __import__ not found

如果我使用pickle转储，此示例有效，但pickle似乎不能递归保存依赖项，因此我无法保存def fun1(): return fun2(). 有没有办法转储具有导入和依赖项的函数？我觉得pickle还是dill只做了一半。

标签： pythonpickledumpdill

解决方案

我是dill作者。我相信dill也应该为你工作：

$ python
Python 3.6.10 (default, Dec 21 2019, 11:39:07) 
[GCC 4.2.1 Compatible Apple LLVM 9.0.0 (clang-900.0.39.2)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> 
>>> def func():
...   import numpy
... 
>>> import dill
>>> 
>>> with open('XXX.pkl', 'wb') as f:
...   dill.dump(func, f)
... 
>>> 

$ python
Python 3.6.10 (default, Dec 21 2019, 11:39:07) 
[GCC 4.2.1 Compatible Apple LLVM 9.0.0 (clang-900.0.39.2)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import dill
>>> func = dill.load(open('XXX.pkl', 'rb'))
>>> func()
>>>

这些recurse设置意味着通过全局字典递归跟踪引用，但不存储整个全局字典。的默认设置dill是在酸洗函数时存储所有全局字典。因此，recurse可以使泡菜更小，但它们也可能由于缺少参考而失败。

推荐阅读