mirror of
https://github.com/python/cpython.git
synced 2024-11-24 10:24:35 +08:00
012b69cb30
existed at the time atexit first got imported. That's a bug, and this fixes it. Also reworked test_atexit.py to test for this too, and to stop using an "expected output" file, and to test what actually happens at exit instead of just simulating what it thinks atexit will do at exit. Bugfix candidate, but it's messy so I'll backport to 2.2 myself.
51 lines
1.2 KiB
Python
51 lines
1.2 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"]
|
|
|
|
_exithandlers = []
|
|
def _run_exitfuncs():
|
|
"""run any registered exit functions
|
|
|
|
_exithandlers is traversed in reverse order so functions are executed
|
|
last in, first out.
|
|
"""
|
|
|
|
while _exithandlers:
|
|
func, targs, kargs = _exithandlers.pop()
|
|
apply(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")
|