4

当使用“ ipython”或“ code.interact(local=locals())”时,我希望有一种方法可以将整个程序地址空间保存到一个 pickle 文件中,以及类似的一种方法来加载这样的文件,然后在该上下文中开始执行。

由于虚拟内存,这对于简单的脚本应该是完全可能的。

所有定义的名称(以及它们指向的非孤立对象),从本地到全局和全局函数,都将被腌制。在 unpickling 时,这些名称将在其原始范围内再次分配。

假设程序不使用网络,因此避免了那里的状态不连续性。

假设程序对于感知到的系统时钟的不连续性是容错的,所以也没有问题。

唯一的挑战似乎是如何处理文件描述符。理想情况下,如果它不存在,应该有一个简单的“文件描述符pickle”函数来获取它打开的模式位,文件光标的位置,以及文件内容的校验和(如果尝试 unpickle 时校验和不匹配)。

是否有几行可以完成这种“腌制整个会话”?

4

1 回答 1

4

为此,我会使用dill,它可以在 python 中序列化几乎任何东西。

>>> import dill
>>> 
>>> def foo(a):
...   def bar(x):
...     return a*x
...   return bar
... 
>>> class baz(object):
...   def __call__(self, a,x):
...     return foo(a)(x)
... 
>>> b = baz()
>>> b(3,2)
6
>>> c = baz.__call__
>>> c(b,3,2)
6
>>> g = dill.loads(dill.dumps(globals()))
>>> g
{'dill': <module 'dill' from '/Library/Frameworks/Python.framework/Versions/7.2/lib/python2.7/site-packages/dill-0.2a.dev-py2.7.egg/dill/__init__.pyc'>, 'c': <unbound method baz.__call__>, 'b': <__main__.baz object at 0x4d61970>, 'g': {...}, '__builtins__': <module '__builtin__' (built-in)>, 'baz': <class '__main__.baz'>, '_version': '2', '__package__': None, '__name__': '__main__', 'foo': <function foo at 0x4d39d30>, '__doc__': None}

Dill 将它的类型注册到pickle注册表中,因此如果您有一些使用的黑盒代码pickle并且您无法真正编辑它,那么只需导入 dill 就可以神奇地使其工作而无需猴子修补第 3 方代码。

您还想腌制整个解释器会话... dill 也可以这样做。

>>> # continuing from above
>>> dill.dump_session('foobar.pkl')
>>>
>>> ^D
dude@sakurai>$ python
Python 2.7.5 (default, Sep 30 2013, 20:15:49) 
[GCC 4.2.1 (Apple Inc. build 5566)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import dill
>>> dill.load_session('foobar.pkl')
>>> c(b,3,2)
6

Dill 还提供了一些很好的工具,可以帮助您了解在代码失败时导致酸洗失败的原因。

于 2013-10-19T15:15:37.097 回答