atexit模块

最新推荐文章于 2024-04-17 23:34:27 发布

易爻64

最新推荐文章于 2024-04-17 23:34:27 发布

阅读量493

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python

本文链接：https://blog.youkuaiyun.com/weixin_42176112/article/details/124520004

python 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了Python的atexit模块，用于在程序退出时执行回调函数，进行资源清理。讲解了atexit的实现原理，即通过sys.exitfunc来按先进后出顺序调用注册函数。文章还展示了register的标准语法和应用例子，强调了即使有异常，其他回调也会执行完毕。并提供了两个参考链接供进一步阅读。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

前言

atexit模块很简单，只定义了一个register函数用于注册程序退出时的回调函数，我们可以在这个回调函数中做一些资源清理的操作。
如果程序是非正常crash，或者通过os._exit()退出，注册的回调函数将不会被调用。

实现原理

我们也可以通过sys.exitfunc来注册回调，但通过它只能注册一个回调，而且还不支持参数。
所以建议大家使用atexit来注册回调函数。但千万不要在程序中同时使用这两种方式，否则通过atexit注册的回调可能不会被正常调用。
其实通过查阅atexit的源码，你会发现原来它内部是通过sys.exitfunc来实现的，
它先把注册的回调函数放到一个列表中，当程序退出的时候，按先进后出的顺序调用注册的回调。
如果回调函数在执行过程中抛出了异常，atexit会打印异常的文字信息，并继续执行下一下回调，直到所有的回调都执行完毕，它会重新抛出最后接收到的异常。

常用方法

register

标准语法

register(func [, args [, kwargs]])

应用举例

import atexit


def exit0(*args, **kwarg):
    print('exit0')
    for arg in args:
        print(' ' * 4, arg)
    for item in kwarg.items():
        print(' ' * 4, item)


def exit1():
    print('exit1')
    raise Exception('exit1')


def exit2():
    print('exit2')


@atexit.register
def exit3():
    print('exit3')


if __name__ == '__main__':
    atexit.register(exit0, *[1, 2, 3], **{"a": 1, "b": 2, })
    atexit.register(exit1)
    atexit.register(exit2)