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

注意回调函数只有正常退出的时候才会调用,如果程序是被信号杀死或者因为严重错误导致退出,或者使用os._exit()退出程序的时候,回调函数不会调用。

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

  1. atexit.register(func[,*args[,**kargs]])
    注册func函数,当程序退出时将调用这个函数。func需要的参数可以先传递给register函数。
       示例:计数的简单实现,从文件中获取计数值,当程序正常退出的时候自动加1。
  1. try:
    _count = int(open("/tmp/counter").read())
    exceptIOError:
    _count =0 def incrcounter(n):
    global _count
    _count = _count + n def savecounter():
    open("/tmp/counter","w").write("%d"% _count)
    import atexit
    atexit.register(savecounter)
    还可以给register函数传递位置参数和关键字参数。 def goodbye(name, adjective):
    print'Goodbye, %s, it was %s to meet you.'%(name, adjective)
    import atexit
    atexit.register(goodbye,'Donny','nice')
    # or:
    atexit.register(goodbye, adjective='nice', name='Donny')
 
        除此之外,还可以用作修饰器函数。
  1. import atexit
    
    @atexit.register
    def goodbye():
    print"You are now leaving the Python sector."
05-02 09:56