在编写Python程序时,经常需要设置临时选择项,如出现异常时询问后续操作,或程序开始执行时询问必要参数等,这时候往往需要用到input
或弹窗等方式向用户询问。然而,当程序并不是自己用,或者需要分享给他人的时候,出于稳健性的考虑,我们可能需要为此类输入命令设置超时时间,如询问选择后若5秒内未得到返回值,就选择默认参数,并继续执行后续进程。
然而,通过一段时间的搜索,网上提供了不少超时终止的方法,如采取多线程的方法,或使用包eventlet
,或者使用signal
。经过测试,发现其各自都有一些问题。
- 对于使用
threading
创建多线程的方法来说,实现方法较为复杂,需要编写装饰器,而且若要获取返回值还需要为Thread撰写子类来实现,而这里只是需要实现一个简单的超时跳过,大可不必如此大动干戈。 - 对于使用
eventlet
方法实现超时的方法,介绍的文章大部分都是针对爬虫或大量计算的情况,经过测试发现该方法对input
无效,对于input
来说无论超时多久都不会自动跳过 - 对于使用
signal
的方法,只有Linux\MacOS
平台下可用,Windows
无法使用
本文通过调用一个简单的库实现装饰器,能够通过短短8行代码,实现获取input询问输入,并计时超时后自动抛出异常或设置默认参数继续执行后续进程。
首先pip install func_timeout
包,将需要询问的内容封装进一个函数,并为其添加func_set_timeout(2)
的装饰器,参数2
表示超时时间,此处设置为2s。
import func_timeout
@func_timeout.func_set_timeout(2)
def askChoice():
return input('yes or no:')
try:
s = askChoice()
except func_timeout.exceptions.FunctionTimedOut as e:
s = 'f'
print(s)
程序执行时先调用askChoice
函数,并开始计时,若用户在计时期间内输入,则正常传参,若用户超时,则触发func_timeout.exceptions.FunctionTimedOut
异常,try...except
捕捉异常,并进行后续操作。
此方法对于多线程中的子线程内的超时问题也能提供很好的处理,有兴趣的同学可以尝试一下,欢迎讨论~