ipython 的 atexit 替代品答案

【问题标题】：atexit alternative for ipythonipython 的 atexit 替代品
【发布时间】：2016-10-21 23:11:46
【问题描述】：

python 有atexit 模块来注册函数以在关闭解释器之前运行。 This question 很好，为什么不调用 atexit。

我想知道ipython 是否可以在退出使用%run <name> 运行的东西之前注册一个函数？理想情况下，我想创建一个装饰器，可以在任何一个中使用寄存器模块取决于解释器。

【问题讨论】：

在%run 之后没有特别触发的东西，但你可以define a custom magic 执行%run 加上一些操作，或者provide a post_run_cell callback 在每个输入执行后运行。跨度>

【解决方案1】：

感谢 Thomas K 的好评。如果他写了一个答案，我会接受他的。否则这段代码可能会让其他人受益：

# exit_register runs at the end of ipython %run or the end of the python interpreter
try:
    def exit_register(fun, *args, **kwargs):
        """ Decorator that registers at post_execute. After its execution it
        unregisters itself for subsequent runs. """
        def callback():
            fun()
            ip.events.unregister('post_execute', callback)
        ip.events.register('post_execute', callback)


    ip = get_ipython()
except NameError:
    from atexit import register as exit_register


@exit_register
def callback():
    print('I\'m done!')


print('Running')

【讨论】：

这很好用。要补充的一件事。 atexit.register 可以接受参数和关键字参数。考虑通过 def exit_register(fun, *args, **kwargs) 和 fun(*args, **kwargs) 更改 exit_register 以允许此操作。