python atexit 模块(回调函数)

来源:互联网 发布:剑灵数据互通查询 编辑:程序博客网 时间:2024/06/05 20:26
>>>> import atexit>>> dir(atexit)['__all__', '__builtins__', '__doc__', '__file__', '__name__', '__package__', '_exithandlers', '_run_exitfuncs', 'register', 'sys']>>> help(atexit.register)Help on function register in module atexit:register(func, *targs, **kargs)    register a function to be executed upon normal program termination    func - function to be called at exit    targs - optional arguments to pass to func    kargs - optional keyword arguments to pass to func    func is returned to facilitate usage as a decorator.>>> 

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

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

下面是一个例子,展示了atexit模块的使用:

import atexitdef 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

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


示例2:

exit1.py文件:

#!usr/bin/env python#coding:utf-8import atexit  def exit0(*args, **kwarg):      print 'exit0'      for arg in args:          print ' ' * 4, arg      for item in kwarg.items():          print ' ' * 4, item    raise Exception,'exit0'     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__':    print '回调函数没有执行!!'    import os    os._exit(0) #通过os._exit()退出,注册的回调函数不会被调用
运行结果:

song@ubuntu:~$ python exit1.py
回调函数没有执行!!
song@ubuntu:~$


注意:在IDLE里面atexit的输出是不会显示的,需要在cmd窗口里面才有显示

IDLE是开发python程序的基本IDE(集成开发环境),具备基本的IDE的功能,是非商业Python开发的不错的选择。
当安装好python以后,IDLE就自动安装好了,不需要另外去找。



关于python中的sys.exit()  os._exit() 请参考:点击打开链接

原文链接:http://python.jobbole.com/81473/




1 0