在Python的主线程中捕获线程的异常

Question

Hi I'm new to Python and multithreaded programming. I have a script running through some test steps. Meanwhile this is running I want to create a thread that polls with an interval and reads if any processes have crashed during execution. I have a function which acquires this information. My problem is how I can throw an exception from that thread whenever I get that a process has crashed. My current thread looks something like this:

class process_thread(threading.Thread):
  def __init__(self):
  threading.Thread.__init__(self)
  self.run_thread = True

  def run(self):
    while self.run_thread:
      if has_any_processes_crashed() == -1:
        self.run_thread = False
        raise MyStopException("A process has crashed") # This is the problematic line. It is only raised in this thread, not in main thread.
      else:
        time.sleep(3)

The problem is however that the exception is only raised in that thread and not in the main thread. I want to throw the same exception there and exit the script. What I do with this thread is that I create an object before I start with all my test steps and set it as a Daemon thread. When my test steps are done I want to kill this thread before shutting down the simulation. All this is done by:

p_thread = process_thread()
p_thread.setDaemon(True)
p_thread.start()

# Executing all test steps here
# test_step_1
do_something_1()
# test_step_n
do_something_n()

p_thread.run_thread = False
p_thread.join()

I do not think that a Queue is possible in my scenario as I still need to execute my test steps in the main thread.

Any help is appreciated!

Answer 1

You can save the exception as a variable using the traceback module. In your thread that you are running and expect the exception, try:

import traceback

class process_thread(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
        self.run_thread = True

    def run(self):
        try:
            process that may crash()
        except:
            self.exception_var = traceback.format_exc()

The you access the variable in the main thread:

print(self.exception_var)

Or whatever you want to do with it here. Bear in mind, without seeing how your processes crash, i'm not sure this is exactly what you want. This does require that the processes that crash actually cause an exception, hence using traceback. If they don't, then you may have to manually raise an exception. There is a great answer on doing that here: Manually raising (throwing) an exception in Python