如何转储内部导入的函数

Question

我正在尝试将函数转储到一个文件中，以便我可以在其他地方使用这个 file/function。我选择 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只做了一半。

Answer 1

我是 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 的默认设置是在 pickle 函数时存储所有全局字典。因此，recurse 可以使 pickle 变小，但它们也可能由于缺少引用而失败。

如何转储内部导入的函数

How to dump a function with import inside

python

dump

pickle

dill