1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
atexit_callbacks = []
def register(func, *args, **kwargs):
"""Register a function to be executed upon normal program termination.
func - function to be called at exit
args - optional arguments to pass to func
kwargs - optional keyword arguments to pass to func
func is returned to facilitate usage as a decorator."""
if not callable(func):
raise TypeError("func must be callable")
atexit_callbacks.append((func, args, kwargs))
return func
def run_exitfuncs():
"Run all registered exit functions."
# Maintain the last exception
last_exc, last_tb = None, None
for (func, args, kwargs) in reversed(atexit_callbacks):
if func is None:
# unregistered slot
continue
try:
func(*args, **kwargs)
except BaseException as e:
if not isinstance(e, SystemExit):
import traceback
# obscure: we can't use sys.exc_info() here because this
# function is an appleveldef which marks its frame as
# hidden
last_exc = e
last_tb = e.__traceback__
traceback.print_exception(type(last_exc), last_exc, last_tb)
clear()
if last_exc is not None:
raise last_exc.with_traceback(last_tb)
def clear():
"Clear the list of previously registered exit functions."
del atexit_callbacks[:]
def unregister(func):
"""Unregister a exit function which was previously registered using
atexit.register"""
for i, (f, _, _) in enumerate(atexit_callbacks):
if f == func:
atexit_callbacks[i] = (None, None, None)
def ncallbacks():
return len(atexit_callbacks)
|
