| """ |
| 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") |
