Python 强制停止多线程运行

Posted 2021-02-17 授客的博客

强制停止多线程运行

by:授客 QQ：1033553122

 

#!/usr/bin/env python

# -*- coding:utf-8 -*-

 

 

__author__ = \'shouke\'

 

import threading

import time

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 stop_thread(thread):

    _async_raise(thread.ident, SystemExit)

 

 

def test():

    try:

        while True:

            print(\'-------\')

            time.sleep(0.5)

    except Exception as e:

        print(\'Exception：%s\' % e)

    finally:

        print(\'stop running test function\')

 

 

if __name__ == "__main__":

    t = threading.Thread(target=test)

    t.start()

    time.sleep(5)

    stop_thread(t)

    print("main thread running")

    print("main thread running")

    print("main thread running")

    print("main thread running")

 

运行结果：

 

 

 

结论：

按上述方法是可以停止多线程的，但是需要注意的地方是，线程退出前，会执行try...finally中的代码，如果代码包含了多层try...finally，每一层的finally中的语句都会被执行，如下：

 

修改代码如下

def test():

    try:

        try:

            while True:

                print(\'-------\')

                time.sleep(0.5)

        except Exception as e:

            print(\'Exception：%s\' % e)

        finally:

            print(\'stop running test function\')

        print(\'outer try\')

    except Exception:

        pass

    finally:

        print(\'outer try finally\')

 

再次运行，结果