65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
"""
|
|
atexit.py - allow programmer to define multiple exit functions to be executed
|
|
upon normal program termination.
|
|
|
|
One public function, register, is defined.
|
|
"""
|
|
|
|
__all__ = ["register"]
|
|
|
|
import sys
|
|
|
|
_exithandlers = []
|
|
def _run_exitfuncs():
|
|
"""run any registered exit functions
|
|
|
|
_exithandlers is traversed in reverse order so functions are executed
|
|
last in, first out.
|
|
"""
|
|
|
|
exc_info = None
|
|
while _exithandlers:
|
|
func, targs, kargs = _exithandlers.pop()
|
|
try:
|
|
func(*targs, **kargs)
|
|
except SystemExit:
|
|
exc_info = sys.exc_info()
|
|
except:
|
|
import traceback
|
|
print >> sys.stderr, "Error in atexit._run_exitfuncs:"
|
|
traceback.print_exc()
|
|
exc_info = sys.exc_info()
|
|
|
|
if exc_info is not None:
|
|
raise exc_info[0], exc_info[1], exc_info[2]
|
|
|
|
|
|
def 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.
|
|
"""
|
|
_exithandlers.append((func, targs, kargs))
|
|
return func
|
|
|
|
if hasattr(sys, "exitfunc"):
|
|
# Assume it's another registered exit function - append it to our list
|
|
register(sys.exitfunc)
|
|
sys.exitfunc = _run_exitfuncs
|
|
|
|
if __name__ == "__main__":
|
|
def x1():
|
|
print "running x1"
|
|
def x2(n):
|
|
print "running x2(%r)" % (n,)
|
|
def x3(n, kwd=None):
|
|
print "running x3(%r, kwd=%r)" % (n, kwd)
|
|
|
|
register(x1)
|
|
register(x2, 12)
|
|
register(x3, 5, "bar")
|
|
register(x3, "no kwd args")
|