atexit模块的主要作用是在程序即将结束之间执行的代码。atexit模块使用register函数用于注册程序退出时的回调函数。
主要函数:
atexit.register(func, *args, **kargs)
atexit.unregister(func)
例子:
import atexit
def atexitFunc_1():
print("这是测试一号")
def atexitFunc_2(name, age):
print('这是测试二号, %s is %d' % (name, age))
atexit.register(atexitFunc_1)
atexit.register(atexitFunc_2, 'tom', 20)
@atexit.register
def atexitFunc_3():
print('这是测试三号')
程序输出:
调用顺序和注册顺序是相反的。
这是测试三号
这是测试二号, tom is 20
这是测试一号
注意:
-
如果程序是非正常退出,或者通过os._exit()退出,注册的回调函数是不会被调用的。
-
也可以通过sys.exitfunc来注册回调,但是它只能注册一个回调,而且不支持参数。
-
建议使用atexit来注册回调函数。但是不要在程序中同时使用这两种方法,否则通过atexit定义的回调函数不会被正常调用。atexit内部是通过sys.exitfunc来实现的,它把注册的回调函数放到一个列表中,当车灯徐退出时,按照先进后出的顺序调用。如果回调函数在执行过程中抛出异常,atexit会打印异常的文字信息,并继续执行下一个回调,直到所有会的回调执行完毕,它会重新重新抛出最后接收到的异常。
-
如果使用的python版本是2.6及以后的版本,可以通过修饰器来注册回调函数。