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