Python模块教程之atexit

发布时间:2020-03-30编辑:脚本学堂
介绍下python中atexit模块的用法,atexit模块使用register函数用于注册程序退出时的回调函数,然后在回调函数中做一些资源清理的操作,有需要的朋友参考下。

本节内容:
python atexit模块的用法

在python编程中,atexit模块很简单,只定义了一个register函数用于注册程序退出时的回调函数,可以在这个回调函数中做一些资源清理的操作。

注意:如果程序是非正常crash,或通过os._exit()退出,注册的回调函数将不会被调用。

也可以通过sys.exitfunc来注册回调,但通过它只能注册一个回调,而且还不支持参数。
建议使用atexit来注册回调函数。

不过请特别注意:
不要在程序中同时使用这两种方式,否则通过atexit注册的回调可能不会被正常调用。
通过查阅atexit的源码,原来它内部是通过sys.exitfunc来实现的,它先把注册的回调函数放到一个列表中,当程序退出时,按先进后出的顺序调用注册的回调。如果回调函数在执行过程中抛出了异常,atexit会打印异常的文字信息,并继续执行下一下回调,直到所有的回调都执行完毕,它会重新抛出最后接收到的异常。

如果使用的python版本是2.6,还可以用装饰器的语法来注册回调函数。

例子,展示了python atexit模块的使用:
 

复制代码 代码示例:
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(exit0, *[1, 2, 3], **{ "a": 1, "b": 2, }) 
atexit.register(exit1) 
atexit.register(exit2) 
 
@atexit.register 
def exit3(): 
    print 'exit3' 
 
if __name__ == '__main__': 
    pass 
 

从程序运行的结果,可以看到回调函数执行的顺序与它们被注册的顺序刚才相反。

以上通过实例介绍了Python atexit模块的用法,希望对大家有所帮助。