一尘不染

Python-如何在raw_input上设置时间限制

python

在python中,有没有一种方法可以在等待用户输入时计数时间,以便在30秒后raw_input()自动跳过该函数?


阅读 1364

收藏
2020-02-15

共1个答案

一尘不染

@ jer推荐的解决方案所基于的signal.alarm函数仅适用于Unix。如果您需要跨平台或Windows特定的解决方案,则可以将其基于threading.Timer,使用thread.interrupt_mainKeyboardInterrupt从计时器线程将a发送到主线程。即:

import thread
import threading

def raw_input_with_timeout(prompt, timeout=30.0):
    print prompt,    
    timer = threading.Timer(timeout, thread.interrupt_main)
    astring = None
    try:
        timer.start()
        astring = raw_input(prompt)
    except KeyboardInterrupt:
        pass
    timer.cancel()
    return astring

无论30秒超时还是用户明确决定按Control-C放弃输入任何内容,这都将返回None,但是以相同的方式处理这两种情况似乎可以(如果需要区分,则可以使用定时器自己的功能,中断主线程之前,记录某处的事实超时已经发生了,在你的处理器的KeyboardInterrupt访问是“地方”来区分这两个案件发生)。

编辑:我可以发誓,这是工作,但我一定是错的-上面省略了代码显然需要的timer.start(),而且即使它,我不能做任何更多的工作。 select.select显然还有其他尝试,但是在Windows中的“普通文件”(包括stdin)上将不起作用-在Unix中,Windows中的所有文件都只能在套接字上工作。

所以我不知道如何做一个跨平台的“原始输入超时”。可以使用紧密循环轮询msvcrt.kbhit来构造特定于Windows的窗口,然后执行一次msvcrt.getche(并检查是否返回以指示输出已完成,在这种情况下,它会跳出循环,否则累积并一直等待)并检查如果需要,可以超时。我无法测试,因为我没有Windows机器(它们都是Mac和Linux机器),但是在这里,我建议使用未经测试的代码:

import msvcrt
import time

def raw_input_with_timeout(prompt, timeout=30.0):
    print prompt,    
    finishat = time.time() + timeout
    result = []
    while True:
        if msvcrt.kbhit():
            result.append(msvcrt.getche())
            if result[-1] == '\r':   # or \n, whatever Win returns;-)
                return ''.join(result)
            time.sleep(0.1)          # just to yield to other processes/threads
        else:
            if time.time() > finishat:
                return None

OP在评论中说他不想return None超时,但是还有什么选择?引发异常?返回不同的默认值?无论他想要什么选择,他都可以清楚地代替我的return None;-)。

如果您不希望仅由于用户输入速度慢而超时(而不是根本不输入!-),则可以在每次成功输入字符后重新计算finishat。

2020-02-15