python设置函数超时正常执行

最近想用python实现如下的一个功能,即设置函数的超时时间,当超过这个时间后函数退回到父函数而不会引发异常(导致整个程序结束)
如下的方式会导致整个程序停止执行
https://blog.csdn.net/qq_43994782/article/details/119033086

①首先,我认真看了题主的链接,如果只是要不报错,加上异常处理就可以了,如下代码:

import time
import func_timeout
from func_timeout import func_set_timeout

@func_set_timeout(3)
def task():
    while True:
        print('hello world')
        time.sleep(1)

if __name__ == '__main__':
    try:
        task()
    except func_timeout.exceptions.FunctionTimedOut:
        print("执行已超时3秒")

输出结果如下图:

img

②这里发现另外一个问题,链接中的如下代码,其实还是会阻塞主线程。


import time
from func_timeout import func_set_timeout
​
@func_set_timeout(1)
def task():
    while True:
        print('hello world')
        time.sleep(1)
​
if __name__ == '__main__':
    task()


怎么验证呢,我们在task()后面加一句print("a"),如下代码:


import time
from func_timeout import func_set_timeout

@func_set_timeout(3)
def task():
    while True:
        print('hello world')
        time.sleep(1)

if __name__ == '__main__':
    task()
    print("a")


输出结果如下,发现控制台并没有打印字母a,所以题主按这个代码,还是会导致整个程序等待task()执行直到超时。

img


为了解决该问题,这里答主尝试启动多线程执行task(),代码如下:


import time
from func_timeout import func_set_timeout
from threading import Thread

@func_set_timeout(3)
def task():
    while True:
        print('hello world')
        time.sleep(1)

if __name__ == '__main__':
    t1 = Thread(target=task, args=())
    t1.start()
    print("a")


程序运行结果,如下图:


img


发现程序并没有阻塞主线程成功在控制台打印了a,同时3秒后task()超时,抛出异常。
但这里还是有一个问题,就是异常还没捕获和处理,这里加上异常处理代码,同时需要另外再写一个函数try_task()。
完整代码如下:

import time
import func_timeout
from func_timeout import func_set_timeout
from threading import Thread


def try_task():
    try:
        task()
    except func_timeout.exceptions.FunctionTimedOut:
        print("执行已超时3秒")


@func_set_timeout(3)
def task():
    while True:
        print('hello world')
        time.sleep(1)
      

if __name__ == '__main__':
    t1 = Thread(target=try_task, args=())
    t1.start()
    print("a")



输出结果如下图:

img


如果解答有用,请采纳~

望采纳,timeout 不行的花就用,
signal模块来实现超时时间的设置,当超时时间到达时,可以设置一个信号处理函数,在信号处理函数中调用函数的返回语句,从而实现函数的超时处理