How to find exit code or reason when atexit callback is called in Python?

by

You can solve this using sys.excepthook and by monkey-patching sys.exit():

import atexit
import sys

class ExitHooks(object):
    def __init__(self):
        self.exit_code = None
        self.exception = None

    def hook(self):
        self._orig_exit = sys.exit
        sys.exit = self.exit
        sys.excepthook = self.exc_handler

    def exit(self, code=0):
        self.exit_code = code
        self._orig_exit(code)

    def exc_handler(self, exc_type, exc, *args):
        self.exception = exc

hooks = ExitHooks()
hooks.hook()

def foo():
    if hooks.exit_code is not None:
        print("death by sys.exit(%d)" % hooks.exit_code)
    elif hooks.exception is not None:
        print("death by exception: %s" % hooks.exception)
    else:
        print("natural death")
atexit.register(foo)

# test
sys.exit(1)

More Related Contents:

  1. Why can’t I call read() twice on an open file?
  2. Convert hex string to int in Python
  3. Converting a list to a set changes element order
  4. pip how to remove incorrectly installed package with a leading dash: “-pkgname”
  5. stack bar plot in matplotlib and add label to each section
  6. Pandas: Chained assignments [duplicate]
  7. How to pull a random record using Django’s ORM?
  8. How to run Spyder in virtual environment?
  9. How to print Unicode character in Python?
  10. How do I perform HTML decoding/encoding using Python/Django?
  11. Making an asynchronous task in Flask
  12. converting currency with $ to numbers in Python pandas
  13. Lookup Values by Corresponding Column Header in Pandas 1.2.0 or newer
  14. Removing list of words from a string
  15. Is there a way with python-selenium to wait until all elements of a page has loaded?
  16. Python Error: “ValueError: need more than 1 value to unpack”
  17. Week number of the month?
  18. Re-read an open file Python
  19. ImportError: No module named – Python
  20. Run BASH built-in commands in Python?
  21. “SSLError: The read operation timed out” when using pip
  22. How do I ‘lock the keyboard’ to prevent any more keypresses being sent on X11/Linux/Gnome?
  23. Multiple variables in SciPy’s optimize.minimize
  24. Where do I define the domain to be used by url_for() in Flask?
  25. Executing an SQL query over a pandas dataset
  26. Generating movie from python without saving individual frames to files
  27. How to install xgboost in Anaconda Python (Windows platform)?
  28. pyspark parse fixed width text file
  29. How to pass additional parameters to numba cfunc passed as LowLevelCallable to scipy.integrate.quad
  30. Selenium / Python – Selecting via css selector

Leave a Comment