diff options
Diffstat (limited to 'Lib/atexit.py')
-rw-r--r-- | Lib/atexit.py | 14 |
1 files changed, 5 insertions, 9 deletions
diff --git a/Lib/atexit.py b/Lib/atexit.py index 61f2458..b5929fc 100644 --- a/Lib/atexit.py +++ b/Lib/atexit.py @@ -29,15 +29,11 @@ def register(func, *targs, **kargs): _exithandlers.append((func, targs, kargs)) import sys -try: - x = sys.exitfunc -except AttributeError: - sys.exitfunc = _run_exitfuncs -else: - # if x isn't our own exit func executive, assume it's another - # registered exit function - append it to our list... - if x != _run_exitfuncs: - register(x) +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__": |