home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / atexit.py < prev    next >
Text File  |  2003-12-30  |  1KB  |  51 lines
  1. """
  2. atexit.py - allow programmer to define multiple exit functions to be executed
  3. upon normal program termination.
  4.  
  5. One public function, register, is defined.
  6. """
  7.  
  8. __all__ = ["register"]
  9.  
  10. _exithandlers = []
  11. def _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.     """
  17.  
  18.     while _exithandlers:
  19.         func, targs, kargs = _exithandlers.pop()
  20.         func(*targs, **kargs)
  21.  
  22. def 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.  
  31. import sys
  32. if hasattr(sys, "exitfunc"):
  33.     # Assume it's another registered exit function - append it to our list
  34.     register(sys.exitfunc)
  35. sys.exitfunc = _run_exitfuncs
  36.  
  37. del sys
  38.  
  39. if __name__ == "__main__":
  40.     def x1():
  41.         print "running x1"
  42.     def x2(n):
  43.         print "running x2(%s)" % `n`
  44.     def x3(n, kwd=None):
  45.         print "running x3(%s, kwd=%s)" % (`n`, `kwd`)
  46.  
  47.     register(x1)
  48.     register(x2, 12)
  49.     register(x3, 5, "bar")
  50.     register(x3, "no kwd args")
  51.