主动杀死python线程

发表于 更新于 分类于 Valine： 本文字数： 862 阅读时长 ≈ 1 分钟

Python的线程没有提供从外部主动终止线程的方法，当然从程序设计上应该避免这种方式，但有些时候我们希望把线程超时当做一种异常来处理，我们需要提前终止它，下面的代码提供了一种可参考方案

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import threading
import inspect
import ctypes

def _async_raise(tid, exctype):
    """raises the exception, performs cleanup if needed"""
    tid = ctypes.c_long(tid)
    if not inspect.isclass(exctype):
        exctype = type(exctype)
    res = ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, ctypes.py_object(exctype))
    if res == 0:
        raise ValueError("invalid thread id")
    elif res != 1:
        # """if it returns a number greater than one, you're in trouble,
        # and you should call it again with exc=NULL to revert the effect"""
        ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, None)
        raise SystemError("PyThreadState_SetAsyncExc failed")

def kill_thread(tid):
    _async_raise(tid, SystemExit)

实际测试下来大多数情况是可以工作的，除了线程长时间陷在time.sleep的情况，也就是说上面的代码无法把一个线程从sleep状态唤醒并终止。