blob: 61f2458dd0a429b653ac3546a6053fe80e36c402 [file] [log] [blame]
Fred Drakec19425d2000-06-28 15:07:31 +00001"""
2atexit.py - allow programmer to define multiple exit functions to be executed
3upon normal program termination.
4
Tim Peters146965a2001-01-14 18:09:23 +00005One public function, register, is defined.
Fred Drakec19425d2000-06-28 15:07:31 +00006"""
7
Skip Montanaroe99d5ea2001-01-20 19:54:20 +00008__all__ = ["register"]
9
Fred Drakec19425d2000-06-28 15:07:31 +000010_exithandlers = []
11def _run_exitfuncs():
12 """run any registered exit functions
13
14 _exithandlers is traversed in reverse order so functions are executed
15 last in, first out.
16 """
Tim Peters146965a2001-01-14 18:09:23 +000017
Fred Drakec19425d2000-06-28 15:07:31 +000018 while _exithandlers:
Tim Peters384fd102001-01-21 03:40:37 +000019 func, targs, kargs = _exithandlers.pop()
Fred Drakec19425d2000-06-28 15:07:31 +000020 apply(func, targs, kargs)
Fred Drakec19425d2000-06-28 15:07:31 +000021
22def register(func, *targs, **kargs):
23 """register a function to be executed upon normal program termination
24
25 func - function to be called at exit
26 targs - optional arguments to pass to func
27 kargs - optional keyword arguments to pass to func
28 """
29 _exithandlers.append((func, targs, kargs))
30
31import sys
32try:
33 x = sys.exitfunc
34except AttributeError:
35 sys.exitfunc = _run_exitfuncs
36else:
37 # if x isn't our own exit func executive, assume it's another
38 # registered exit function - append it to our list...
39 if x != _run_exitfuncs:
40 register(x)
41del sys
42
43if __name__ == "__main__":
44 def x1():
45 print "running x1"
46 def x2(n):
47 print "running x2(%s)" % `n`
48 def x3(n, kwd=None):
49 print "running x3(%s, kwd=%s)" % (`n`, `kwd`)
50
51 register(x1)
52 register(x2, 12)
53 register(x3, 5, "bar")
54 register(x3, "no kwd args")