python设置函数超时正常执行

Question

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

Answer 1

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

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秒")

输出结果如下图：

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

---

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()执行直到超时。

---

为了解决该问题，这里答主尝试启动多线程执行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")

---

程序运行结果，如下图：

---

---

发现程序并没有阻塞主线程成功在控制台打印了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")

---

输出结果如下图：

---

如果解答有用，请采纳~