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)