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

Posted 神奇的代码在哪里

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了一种设置python函数执行超时时间 | 函数超时后不抛出异常的方法相关的知识,希望对你有一定的参考价值。

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

文章目录

前言

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

一、func_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()执行完毕,直到超时报错。


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

为了解决阻塞主线程的问题,这里尝试使用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个知识点,继续加油吧。

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

Python设置函数调用超时

http://blog.sina.com.cn/s/blog_63041bb80102uy5o.html

背景:

       最近写的Python代码不知为何,总是执行到一半卡住不动,为了使程序能够继续运行,设置了函数调用超时机制。

代码:
import time
import signal

def test(i):
    time.sleep(i%4)
    print "%d within time"%(i)
    return i

if __name__ == __main__:
    def handler(signum, frame):
        raise AssertionError

    i = 0
    for i in range(1,10):
        try:
            signal.signal(signal.SIGALRM, handler)
            signal.alarm(3)
            test(i)
            i = i + 1
            signal.alarm(0)
        except AssertionError:
            print "%d timeout"%(i)
说明:
        1、调用test函数超时监控,使用sleep模拟函数执行超时
        2、引入signal模块,设置handler捕获超时信息,返回断言错误
        3、alarm(3),设置3秒闹钟,函数调用超时3秒则直接返回
        4、捕获异常,打印超时信息
 
程序执行结果:
1 within time
2 within time
3 timeout
4 within time
5 within time
6 within time
7 timeout
8 within time
9 within time

以上是关于一种设置python函数执行超时时间 | 函数超时后不抛出异常的方法的主要内容,如果未能解决你的问题,请参考以下文章

小程序开发 云函数超时设置了,但还是3s警报

python调用函数设置超时机制

如何为查询执行设置语句超时

php中pcntl_alarm为函数设置超时限制

如何在X毫秒后超时python函数?

关于winsock2中的connect函数