Python 8行代码实现input询问键盘输入超时自动跳过选择默认值

在编写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捕捉异常,并进行后续操作。

此方法对于多线程中的子线程内的超时问题也能提供很好的处理,有兴趣的同学可以尝试一下,欢迎讨论~

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章