一种设置python函数执行超时时间 | 函数超时后不抛出异常的方法

  • 问题来源❓: 写该文章的缘由是因为最近回答CSDN用户一个关于函数超时的问题。当时用户提问:最近想用python实现如下的一个功能,即设置函数的超时时间,当超过这个时间后函数退回到父函数而不会引发异常或导致整个程序结束。当时用户还贴出了另一篇文章的地址 https://blog.csdn.net/qq_43994782/article/details/119033086,但没有解决他的问题。
  • 问题场景一:文中还提到,最近碰到一个场景,用opencv链接网络摄像头,但是摄像头📸的地址不一定是正确的,或者是打开的,当错误的时候会直接报错,但是当地址正确却没有打开,或者网络限制的时候,会等上半天才返回,并且不报错。
  • 问题场景二:通常我们在编写好一个函数后,会调用函数来完成我们想要的功能。例如网络爬虫程序🖥️,发送请求后服务器会在指定时间内响应,通常情况这个相应时间很短,但有时服务器没有返回任何数据,或服务器已经识别爬虫不予返回数据,亦或者是服务器繁忙等,导致返回时间不可控。这时,程序就一直等待服务器响应,这种情况非常浪费资源

文章目录

  • 前言
  • 一、func_timeout库的使用及异常处理
  • 二、func_timeout阻塞主线程问题
  • 三、启动多线程解决阻塞问题
  • 四、try语句封装函数,实现全程捕获异常
  • 总结

前言

通过网络搜索相关资料,然后进行几次程序测试后,博主找到了解决该问题的一个方法,该方法使用func_timeout库和threading库,最终实现了函数的超时处理,同时实现不阻塞主线程和异常不终止程序。

一、func_timeout库的使用及异常处理

  • func_timeout 库简介:使用python第三方 func_timeout 模块中提供的 func_set_timeout 装饰器可以非常简单的设置python程序的超时时间。
  • 参考阅读:Python module to support running any existing function with a given timeout.

首先,根据该CSDN问题提供的文章链接内容,如果只是解决程序不报错,捕获func_timeout.exceptions.FunctionTimedOut异常,进行异常处理即可。如以下代码:

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

上述代码使用@func_set_timeout(3),设置了函数超时时间为3秒,同时使用except func_timeout.exceptions.FunctionTimedOut: print("执行已超时3秒"),进行异常处理,输出结果如下图:

二、func_timeout阻塞主线程问题

  • 什么是线程阻塞:在某一时刻某一个线程在运行一段代码的时候,这时候另一个线程也需要运行,但是在运行过程中的那个线程执行完成之前,另一个线程无法获取到CPU执行权,这时就会造成线程阻塞。

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

mport 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()

怎么验证呢,我们在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()执行完毕,直到超时报错。

三、启动多线程解决阻塞问题

  • Python的多线程:Python提供两个模块进行多线程的操作,分别是thread和threading,前者是比较低级的模块,用于更底层的操作,一般应用级别的开发不常用。
  • 参考阅读:【python第三方库】python多线程编程—threading库

为了解决阻塞主线程的问题,这里尝试使用threading库,启动多线程来执行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语句封装函数,实现全程捕获异常

  • 异常处理:是编程语言或计算机硬件里的一种机制,用于处理软件或信息系统中出现的异常状况(即超出程序正常执行流程的某些特殊条件)。
  • 参考阅读:python异常处理(十分钟彻底搞定!)

到这里,其实还是有一个问题没有解决。那就是刚才程序输出的异常还没被捕获和处理,为了得到完整的解决方案,代码中还需要加上异常处理的部分,同时还需要另外写一个函数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")

最终程序输出结果如下图,实现了不阻塞主线程,3秒后超时抛出异常并被正常捕获。

总结

之前,由于想拿到CSDNpython入门徽章需要去问答区回答5个问题,一开始觉得好难😂,但在坚持作答几天后发现,在回答别人问题的时候自己也会学到很多知识,而且感觉边解决问题边学习成长的更快,理解的更深。

虽然自己的解决方案不一定很完美,但最终能得到问题提出用户的采纳还是挺有成就感的,以后还会坚持常常去问答区看看,学习学习✍️✍️✍️,现在已经累计回答了23个问题,被采纳8个,采纳率34.7%,换个角度看就是自己也学习了23个知识点,继续加油吧。

如果文章对你有帮助,欢迎点赞、评论、收藏哦!

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年11月30日
下一篇 2023年11月30日

相关推荐