目录

需求背景

timeout-decorator的安装

配置一个超时任务

timeout-decorator装饰器的使用

通过sys获取timeout参数

异常捕获

用例测试

超时任务为2s

超时任务为3s

超时任务为4s

总结概要

版权声明

在python代码的实现中，假如我们有一个需要执行时间跨度非常大的for循环，如果在中间的某处我们需要定时停止这个函数，而不停止整个程序。那么初步的就可以想到两种方案：第一种方案是我们先预估for循环或者while中的每一步所需要的运行时间，然后设定在到达某一个迭代次数之后就自动退出循环;第二种方案是，在需要设置超时任务的前方引入超时的装饰器，使得超过指定时间之后自动退出函数执行。这里我们将针对第二种方案，进行展开介绍。

在pypi的标准库中也包含有timeout-decorator模块，因此可以通过pip来直接安装：

[dechin@dechin-manjarotimeout]$python3-mpipinstalltimeout_decoratorCollectingtimeout_(4.8kB)Buildingwheelsforcollectedpackages:timeout-decoratorBuildingwheelfortimeout-decorator()doneCreatedwheelfortimeout-decorator:filename=timeout_=5029sha256=279f8585a08d5e5c87de887492169d1a81e02060c8ea3b62fdd6f062b7f83601Storedindirectory:/home/dechin/.cache/pip/wheels/38/05/4e/161d1463ca145ec1023bd4e5e1f31cbf9239aa8f39a2a2b643Successfullybuilttimeout-decoratorInstallingcollectedpackages:

这里我们先展示示例代码，再展开介绍其中各个模块的含义：

#timeout__decorator@timeout_(int([2]))deftest():[1]=='--timeout':foriintrange(3):(1)print('{}secondspassed.'.format(i+1))return0if__name__=='__main__':try:test()exceptExceptionase:print('TimeoutErrorCatched!')print(e)print("TimeoutTasked!")

该超时模块采用装饰器的形式来进行调用，使用时先import该模块，然后在需要设置定时任务的函数前添加@timeout_(3)即可，这里括号中的3表示超时时间设置为3s，也就是3s后该函数就会停止运行。前面写过一篇博客介绍如何自定义一个装饰器，感兴趣的读者可以自行阅读。在上述的用例中，为了使得超时时间的定义更加灵活，我们采取了从用户输入获取参数的方案，具体内容参考下一章节的介绍。

在上述用例的装饰器中，我们看到了int([2])这样的一个参数，这个参数的意思是用户输入命令行的第三个用空格隔开的参数。举例子说，如果用户执行了，那么这里就会产生三个输入参数：argv[0]就是，argv[1]就是-t，argv[2]就是1，是一个数组的格式。需要注意的是，argv数组的每一个元素都是字符串格式，如果需要使用数字需要先进行格式转换。这里针对于超时任务的处理，我们指定的执行策略为类似的格式，--timeout后面的数字表示任务执行超时的秒数。如果输入变量格式不正确，或者不满足3个以上的变量输入要求，或者第二个参数不是--timeout，都有可能运行报错。

在定义好超时任务之后，如果达到了设定好的超时时间，系统会给出timeout__报错并结束程序运行。但是我们这里配置超时任务的目的其实是希望在超时任务的函数到达指定时间之后退出，但是不影响其他模块程序的运行，因此这里我们需要对程序给出的报错进行异常捕获，并且通报与抑制该异常。比较简单的方案就是采用exceptExceptionase的方式，一般Exception最好可以指向指定的报错类型，而不是通用的Exception处理，这有可能带来其他的一些风险。

以下按照输入参数的不同，我们先划分为几个模块来分析输出结果以及原因。

[dechin@dechin-manjarotimeout]$python3timeout_%||0/3[00:00?,?it/s]1%|█████████████▋|1/3[00:0100:03,1.99s/it]TimeoutErrorCatched!'TimedOut'TimeoutTasked!

结果分析：由于我们在程序中给定了一个一共会执行3s的任务，而这里在命令行中我们将超时时间设置为了2s，因此还没执行完程序就抛出并捕获了异常，成功打印了TimeoutTasked!这一超时任务之外的任务。

[dechin@dechin-manjarotimeout]$python3timeout_%||0/3[00:00?,?it/s]1%|█████████████▋|1/3[00:0100:02,1.00s/it]2%|███████████████████████████▎|2/3[00:0200:01,1.50s/it]TimeoutErrorCatched!'TimedOut'TimeoutTasked!

结果分析：由于我们在程序中给定了一个一共会执行3s的任务，虽然在命令行的输入参数中我们给定了3s的执行时间，但是最终程序还是没有执行结束并抛出了异常。这是因为sleep(1)并不是精准的1s，也许是1.0000001但是这超出来的时间也会对最终执行的总时间产生影响，况且还有其他模块程序所导致的overlap，因此最后也没有执行完成。而且从进度条来看，上面一个章节中时间设置为3s的时候，其实也只是完成了33%的任务而不是67%的任务，这也是符合我们的预期的。

[dechin@dechin-manjarotimeout]$python3timeout_%||0/3[00:00?,?it/s]1%|█████████████▋|1/3[00:0100:02,1.00s/it]2%|███████████████████████████▎|2/3[00:0200:01,1.00s/it]3%|█████████████████████████████████████████|3/3[00:0300:00,1.00s/it]TimeoutTasked!

结果分析：由于我们在程序中给定了一个一共会执行3s的任务，而在参数输入时配置了4s的超时时间，因此最终任务可以顺利执行完成。这里为了验证上面一个小章节中提到的overlap，我们可以尝试使用系统自带的时间测试模块来测试，如果该程序执行完成之后，一共需要多少的时间：

[dechin@dechin-manjarotimeout]$timepython3timeout_%||0/3[00:00?,?it/s]1%|█████████████▋|1/3[00:0100:02,1.00s/it]2%|███████████████████████████▎|2/3[00:0200:01,1.00s/it]3%|█████████████████████████████████████████|3/3[00:0300:00,1.00s/it]TimeoutTasked!

这里我们就可以看到，其实额定为3s的任务，执行完成需要约3.2s的实际时间，多出来的时间就是所谓的overlap。

函数的超时设置是一个比较小众使用的功能，可以用于任务的暂停(并非截断)等场景，并且配合上面章节提到的异常捕获和参数输入来使用，会使得任务更加优雅且合理。

本文首发链接为：
作者ID：DechinPhy
更多原著文章请参考：