cpython/Lib/atexit.py

51 lines
1.2 KiB
Python
Raw Normal View History

"""
atexit.py - allow programmer to define multiple exit functions to be executed
upon normal program termination.
2001-01-14 18:09:23 +00:00
One public function, register, is defined.
"""
__all__ = ["register"]
_exithandlers = []
def _run_exitfuncs():
"""run any registered exit functions
_exithandlers is traversed in reverse order so functions are executed
last in, first out.
"""
2001-01-14 18:09:23 +00:00
while _exithandlers:
func, targs, kargs = _exithandlers.pop()
2003-02-27 20:14:51 +00:00
func(*targs, **kargs)
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
"""
_exithandlers.append((func, targs, kargs))
import sys
if hasattr(sys, "exitfunc"):
# Assume it's another registered exit function - append it to our list
register(sys.exitfunc)
sys.exitfunc = _run_exitfuncs
del sys
if __name__ == "__main__":
def x1():
print "running x1"
def x2(n):
print "running x2(%s)" % `n`
def x3(n, kwd=None):
print "running x3(%s, kwd=%s)" % (`n`, `kwd`)
register(x1)
register(x2, 12)
register(x3, 5, "bar")
register(x3, "no kwd args")