Python处理函数调用超时方法的详细教学

 更新时间:2025年09月05日 10:10:29   作者:Python资讯站  
限制函数调用的最大时间是一种非常实用的技术手段,能够帮助开发者更好地控制程序的行为,本文将介绍四种Python处理函数调用超时的方法,大家可以根据需要进行选择

限制函数调用的最大时间是一种非常实用的技术手段,能够帮助开发者更好地控制程序的行为,提升用户体验,同时确保系统的稳定性和可靠性。

func-timeout

func-timeout 是一个 Python 库,允许为函数设置超时时间,防止代码长时间运行或无限阻塞。它适用于需要强制限制执行时间的场景,例如网络请求、计算密集型任务或可能出现死循环的代码。

1. 安装 func-timeout

可以使用 pip 安装:

pip install func-timeout

2. 基本用法

最常用的方式是 func_timeout,它允许在指定的时间内运行一个函数,超时则抛出异常。

from func_timeout import func_timeout, FunctionTimedOut
import time

def long_running_task():
    time.sleep(5)  # 模拟长时间运行的任务
    return "Task completed"
try:
    result = func_timeout(3, long_running_task)  # 设置3秒超时
    print(result)
except FunctionTimedOut:
    print("Function execution timed out!")

解释:

  • func_timeout(3, long_running_task):尝试在 3 秒内运行 long_running_task
  • FunctionTimedOut 异常表示函数超时未完成

也可以使用装饰器方式为函数设定超时:

from func_timeout import func_set_timeout
import time

@func_set_timeout(2)  # 限制该函数的运行时间为2秒
def long_task():
    time.sleep(5)  # 任务实际需要5秒
    return "Finished"

try:
    print(long_task())
except FunctionTimedOut:
    print("Function execution timed out!")
  • 这种方式适用于需要多次调用的函数,避免每次调用都手动设置超时。
  • func-timeout 本质上还是依赖 多线程 或 多进程 实现超时控制,在某些情况下可能不适用于主线程(如 Jupyter Notebook)。它也不能用于 main 线程内的 while True 死循环,因为 Python 的 GIL 可能会影响信号处理。

自定义进程

除了使用上面的库,也可以自己使用一个进程来计时和检测超时,另一个进程来调用 Python 函数。

以下是具体实现代码:

import time
from itertools import count
from multiprocessing import Process

def inc_forever():
    print('Starting function inc_forever()...')
    while True:
        time.sleep(1)
        print(next(counter))

def return_zero():
    print('Starting function return_zero()...')
    return 0

if __name__ == '__main__':
    # counter 是一个无限迭代器
    counter = count(0)

    p1 = Process(target=inc_forever, name='Process_inc_forever')
    p2 = Process(target=return_zero, name='Process_return_zero')
    p1.start()
    p2.start()
    p1.join(timeout=5)
    p2.join(timeout=5)
    p1.terminate()
    p2.terminate()

if p1.exitcode is None:
    print(f'Oops, {p1} timeouts!')

if p2.exitcode == 0:
    print(f'{p2} is luck and finishes in 5 seconds!')

运行结果如下:

Starting function inc_forever()...
Starting function return_zero()...
0
1
2
3
4
Oops, <Process(Process_inc_forever, started)> timeouts!
<Process(Process_return_zero, stopped)> is luck and finishes in 5 seconds!

从退出码可以看出,inc_forever() 函数超时了(退出码为 None),而 return_zero() 函数在 5 秒内成功完成。

subprocess

参数设置超时

从 Python 3.5 开始,subprocess 模块提供了一个便捷且推荐使用的 run() API,它内置了超时支持。

以下是示例代码:

import subprocess

r = subprocess.run(['echo', 'hello timeout'], timeout=5)
print(
    f'''type(r)={type(r)},
    r.args={r.args},
    r.returncode={r.returncode},
    r.stdout={r.stdout},
    r.stderr={r.stderr}''')
try:
    r = subprocess.run(['ping', 'www.google.com'], timeout=5)
except subprocess.TimeoutExpired as e:
    print(e)

运行结果如下:

hello timeout
type(r)=<class 'subprocess.CompletedProcess'>,
    r.args=['echo', 'hello timeout'],
    r.returncode=0,
    r.stdout=None,
    r.stderr=None
PING www.google.com (216.58.194.164) 56(84) bytes of data.
64 bytes from ...: icmp_seq=1 ttl=54 time=10.4 ms
64 bytes from ...: icmp_seq=2 ttl=54 time=5.90 ms
64 bytes from ...: icmp_seq=3 ttl=54 time=6.19 ms
64 bytes from ...: icmp_seq=4 ttl=54 time=9.04 ms
64 bytes from ...: icmp_seq=5 ttl=54 time=16.7 ms
Command '['ping', 'www.google.com']' timed out after 5 seconds

当超时时,会抛出一个 TimeoutExpired 异常。

信号(Signals)

对于 UNIX 系统,还可以使用 signal 模块,通过在 5 秒后向处理器发送信号来引发异常。不过,这种方法相对底层且不够直观。

import signal
def handler(signum, frame):
    raise TimeoutError("函数超时")
def my_function():
    pass
signal.signal(signal.SIGALRM, handler)
signal.alarm(5)
try:
    my_function()
except TimeoutError:
    print("函数超时")
finally:
    signal.alarm(0)

知识补充

Python函数超时后不抛出异常的方法

1.func_timeout库的使用及异常处理

func_timeout 库简介:使用python第三方 func_timeout 模块中提供的 func_set_timeout 装饰器可以非常简单的设置python程序的超时时间。

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秒"),进行异常处理,输出结果如下图:

2.func_timeout阻塞主线程问题

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

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

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

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

3.启动多线程解决阻塞问题

Python的多线程:Python提供两个模块进行多线程的操作,分别是thread和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()函数超时抛出了超时异常。

4.try语句封装函数,实现全程捕获异常

异常处理是编程语言或计算机硬件里的一种机制,用于处理软件或信息系统中出现的异常状况(即超出程序正常执行流程的某些特殊条件)。

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

总结

在开发中,限制函数执行时间是提升程序稳定性和用户体验的重要手段。本文介绍了几种实现方法:

  • func-timeout 库:通过 func_timeout 或装饰器 func_set_timeout,可为函数设置超时时间,超时则抛出异常。适用于网络请求或计算密集型任务。
  • 自定义进程:利用 multiprocessing 模块创建子进程执行函数,通过 join(timeout) 控制超时,超时后终止进程。
  • subprocess 模块:从 Python 3.5 起,subprocess.run() 支持超时参数,超时会抛出 TimeoutExpired 异常,适合外部命令调用。
  • 信号机制:在 UNIX 系统中,使用 signal 模块设置超时信号,超时后触发异常,但实现较底层。

这些方法各有优劣,开发者可根据实际需求选择合适的方案。

到此这篇关于Python处理函数调用超时方法的详细教学的文章就介绍到这了,更多相关Python处理函数调用超时内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python探索之实现一个简单的HTTP服务器

    Python探索之实现一个简单的HTTP服务器

    这篇文章主要介绍了Python探索之实现一个简单的HTTP服务器,具有一定参考价值,需要的朋友可以了解下。
    2017-10-10
  • 基于python编写一个车型识别小程序

    基于python编写一个车型识别小程序

    基于Python制作一个车型识别小程序,可用于拍照识车,根据拍摄照片,快速识别图片中车辆的品牌型号,文章通过代码示例介绍的非常详细,具有一定的参考价值,需要的朋友可以参考下
    2023-10-10
  • 协程Python 中实现多任务耗资源最小的方式

    协程Python 中实现多任务耗资源最小的方式

    协程是 Python 中另外一种实现多任务的方式,只不过比线程更小,占用更小执行单元(理解为需要的资源)。这篇文章主要介绍了协程Python 中实现多任务耗资源最小的方式,需要的朋友可以参考下
    2020-10-10
  • Python munch包 /Munch() 的用法详解

    Python munch包 /Munch() 的用法详解

    这篇文章主要介绍了Python munch包 /Munch() 的用法,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-09-09
  • Python进行PostgreSQL数据库连接的详细使用指南

    Python进行PostgreSQL数据库连接的详细使用指南

    在Python中连接PostgreSQL数据库,最常用的库是psycopg2,本文为大家详细介绍了Python使用psycopg2库操作PostgreSQL的详细步骤,需要的可以了解下
    2025-05-05
  • python安装第三方包的三种方法图文详解

    python安装第三方包的三种方法图文详解

    安装Python第三方包有多种方法,下面这篇文章主要给大家介绍了关于python安装第三方包的三种方法,文中通过图文的非常详细,需要的朋友可以参考下
    2024-03-03
  • el-table 多表格弹窗嵌套数据显示异常错乱问题解决方案

    el-table 多表格弹窗嵌套数据显示异常错乱问题解决方案

    使用vue+element开发报表功能时,需要列表上某列的超链接按钮弹窗展示,在弹窗的el-table列表某列中再次使用超链接按钮点开弹窗,以此类推多表格弹窗嵌套,本文以弹窗两次为例,需要的朋友可以参考下
    2023-11-11
  • pygame播放视频并实现音视频同步的解决方案

    pygame播放视频并实现音视频同步的解决方案

    本篇提供了pygame播放视频的两个方案,第一个方案是网上找的现成的,第二个方案则是参考了opencv(主流方案)自己逆向思维做的,还未经过实际验证,感兴趣的朋友跟随小编一起看看吧
    2023-11-11
  • Python语言描述连续子数组的最大和

    Python语言描述连续子数组的最大和

    这篇文章主要介绍了Python语言描述连续子数组的最大和,具有一定借鉴价值,需要的朋友可以参考下
    2018-01-01
  • python里反向传播算法详解

    python里反向传播算法详解

    在本篇文章了小编给大家整理的是一篇关于python里反向传播算法详解内容,有兴趣的朋友们可以学习下。
    2020-11-11

最新评论