在Python程序中提供超时

时间:2019-05-26 16:18:21

标签: python interrupt

我正在用python编写一个接受两个输入的程序。

一旦用户输入了第一个输入,他将有10秒的时间输入第二个输入。 如果用户能够在这10秒钟内提供第二个值并按Enter键,则计时器将停止并转到程序的下一部分。

python中是否有任何功能可以让我在10秒后提供中断并停止接受第二个输入。如果提供了第二个输入,请停止计时器。

1 个答案:

答案 0 :(得分:1)

您可以创建一个自定义的 Timer 类,并在不同的线程中启动它。一旦发生超时(10秒后),您可以将SIGINT信号发送回父线程,这将引发我们在KeyboardInterrupt函数中捕获的main()异常。否则,您可以在用户在正确的时间输入第二个输入后停止 Timer ,这将停止 Timer 线程。此外,我们可以检查KeyboardInterrupt是否由于超时或用户操作而发生。

注意:在将信号发送到主进程时,我们还需要检查在哪个平台上运行程序。参见signal.CTRL_C_EVENT and signal.SIGINT

演示:https://repl.it/repls/StandardBuoyantProtools

解决方案:

import time
import threading
import os
import signal


class Timer(threading.Thread):
    _timeout = False
    _timer = 0
    _stopped = False

    def __init__(self, delay):
        super(Timer, self).__init__()
        self.restart(delay)

    def is_timeout(self):
        return self._timeout

    def stop(self):
        self._stopped = True

    def restart(self, delay):
        self._stopped = False
        self._timer = time.time() + delay

    def run(self):

        while not self._stopped:
            time.sleep(0.1)
            if time.time() >= self._timer:
                break
        if not self._stopped:
            self._timeout = True

            # check os name
            if os.name == 'nt':
                # we are on Windows
                os.kill(os.getpid(), signal.CTRL_C_EVENT)
            else:
                # we are on a Posix/Unix (or very unlikely on java) system
                os.kill(os.getpid(), signal.SIGINT)


def main():
    first_input = input('First input:')

    delay = 10
    timer = Timer(delay)
    timer.daemon = True

    try:
        print('\nStarting the timer for the second input %r second(s)' % delay)
        timer.start()

        second_input = input('Second input:')

        print('\nWell done. Stopping the timer!\n')
        timer.stop()

        print('Input values: %r %r\n' % (first_input, second_input))

        # do your stuff here...

    except KeyboardInterrupt:
        if timer.is_timeout():
            print("\nTimeout!")
        else:
            print("\nUser interrupted the input")


main()