Python 3 - 打印回溯并在工作线程异常时停止执行



在我的应用程序中,我使用了几个工作线程和一个主循环,它是外部模块的一部分,无法修改。当工作线程引发异常时,应用应停止主循环,执行一些清理,打印工作线程异常回溯并退出。主循环应通过引发 ExitMainLoop 异常来停止。

下面是一个示例代码:

from time import sleep
from threading import Thread

class MyException(BaseException):
    pass

class ExitMainLoop(BaseException):
    pass

def main_loop():
    # this is a part of external module
    print('started main loop')
    try:
        while True:
            print('.')
            sleep(1)
    except ExitMainLoop:
        pass

def do_work():
    print('started worker thread')
    sleep(5)
    raise MyException

def main():
    worker = Thread(target=do_work)
    worker.daemon = True
    worker.start()
    main_loop()

if __name__ == '__main__':
    main()

编辑:我真正的问题是我的应用程序正在使用 urwid 库进行 curses UI,所以我看不到工人异常。

我有一个奇怪的想法,也许可以工作。

   def kill(ex_cls, ex, tb):
       os.kill(os.getpid(), signal.SIGTERM)
    sys.excepthook = kill

发生任何异常时,获取进程 ID 并终止它。

最新更新