mirror of https://github.com/python/cpython
Bug #1597824: return the registered function from atexit.register()
to facilitate usage as a decorator.
This commit is contained in:
parent
adff8eb8d6
commit
540821183b
|
@ -44,6 +44,10 @@ If an exception is raised during execution of the exit handlers, a
|
|||
traceback is printed (unless \exception{SystemExit} is raised) and the
|
||||
exception information is saved. After all exit handlers have had a
|
||||
chance to run the last exception to be raised is re-raised.
|
||||
|
||||
\versionchanged[This function now returns \var{func} which makes it
|
||||
possible to use it as a decorator without binding the
|
||||
original name to \code{None}]{2.6}
|
||||
\end{funcdesc}
|
||||
|
||||
|
||||
|
@ -92,3 +96,15 @@ atexit.register(goodbye, 'Donny', 'nice')
|
|||
# or:
|
||||
atexit.register(goodbye, adjective='nice', name='Donny')
|
||||
\end{verbatim}
|
||||
|
||||
Usage as a decorator:
|
||||
|
||||
\begin{verbatim}
|
||||
import atexit
|
||||
|
||||
@atexit.register
|
||||
def goodbye():
|
||||
print "You are now leaving the Python sector."
|
||||
\end{verbatim}
|
||||
|
||||
This obviously only works with functions that don't take arguments.
|
||||
|
|
|
@ -40,8 +40,11 @@ def register(func, *targs, **kargs):
|
|||
func - function to be called at exit
|
||||
targs - optional arguments to pass to func
|
||||
kargs - optional keyword arguments to pass to func
|
||||
|
||||
func is returned to facilitate usage as a decorator.
|
||||
"""
|
||||
_exithandlers.append((func, targs, kargs))
|
||||
return func
|
||||
|
||||
if hasattr(sys, "exitfunc"):
|
||||
# Assume it's another registered exit function - append it to our list
|
||||
|
|
Loading…
Reference in New Issue