Ввод с клавиатуры с таймаутом в Python

Question

Как бы вы попросили пользователя ввести какой-либо ввод данных, но по истечении N секунд?

Google указывает на почтовую ветку об этом на http://mail.python.org/pipermail/python-list/2006-January/533215.html, но, похоже, она не работает. Оператор, в котором происходит таймаут, независимо от того, является ли это sys.input.readline или timer.sleep (), я всегда получаю:

: входные данные [raw_] ожидаются не более 1 аргумента, получено 2

, который каким-то образом исключение не может поймать.

Answer 1

Использование вызова select короче и должно быть намного более портативным

import sys, select

print "You have ten seconds to answer!"

i, o, e = select.select( [sys.stdin], [], [], 10 )

if (i):
  print "You said", sys.stdin.readline().strip()
else:
  print "You said nothing!"

Answer 2

Пример, на который вы ссылались, неверен, и на самом деле возникает исключение при вызове обработчика тревоги, а не при чтении блоков. Лучше попробуйте это:

import signal
TIMEOUT = 5 # number of seconds your want for timeout

def interrupted(signum, frame):
    "called when read times out"
    print 'interrupted!'
signal.signal(signal.SIGALRM, interrupted)

def input():
    try:
            print 'You have 5 seconds to type in your stuff...'
            foo = raw_input()
            return foo
    except:
            # timeout
            return

# set alarm
signal.alarm(TIMEOUT)
s = input()
# disable the alarm after success
signal.alarm(0)
print 'You typed', s

Answer 3

Не решение Python, но ...

Я столкнулся с этой проблемой с помощью скрипта, работающего под CentOS (Linux), и в моей ситуации мне удалось просто выполнить команду Bash "read -t" в подпроцессе. Зверский отвратительный взлом, я знаю, но я чувствую себя достаточно виноватым за то, как хорошо это сработало, и я хотел поделиться этим со всеми здесь.

import subprocess
subprocess.call('read -t 30', shell=True)

Все, что мне было нужно, это что-то, что ждет 30 секунд, пока не будет нажата клавиша ENTER. Это отлично сработало.

Answer 4

Ответ Павла не совсем сработал. Модифицированный код ниже, который работает для меня на

• windows 7 x64

• ванильная оболочка CMD (например, not git-bash или другая оболочка не M $)

  - ничто msvcrt не работает в git-bash.

• питон 3,6

(я публикую новый ответ, потому что редактирование ответа Пола напрямую изменит его с python 2.x -> 3.x, что кажется слишком большим для редактирования (py2 все еще используется)

import sys, time, msvcrt

def readInput( caption, default, timeout = 5):

    start_time = time.time()
    sys.stdout.write('%s(%s):'%(caption, default))
    sys.stdout.flush()
    input = ''
    while True:
        if msvcrt.kbhit():
            byte_arr = msvcrt.getche()
            if ord(byte_arr) == 13: # enter_key
                break
            elif ord(byte_arr) >= 32: #space_char
                input += "".join(map(chr,byte_arr))
        if len(input) == 0 and (time.time() - start_time) > timeout:
            print("timing out, using default value.")
            break

    print('')  # needed to move to next line
    if len(input) > 0:
        return input
    else:
        return default

# and some examples of usage
ans = readInput('Please type a name', 'john') 
print( 'The name is %s' % ans)
ans = readInput('Please enter a number', 10 ) 
print( 'The number is %s' % ans) 

Answer 5

А вот тот, который работает на Windows

Мне не удалось заставить ни один из этих примеров работать в Windows, поэтому я объединил несколько разных ответов StackOverflow, чтобы получить следующее:

import threading, msvcrt
import sys

def readInput(caption, default, timeout = 5):
    class KeyboardThread(threading.Thread):
        def run(self):
            self.timedout = False
            self.input = ''
            while True:
                if msvcrt.kbhit():
                    chr = msvcrt.getche()
                    if ord(chr) == 13:
                        break
                    elif ord(chr) >= 32:
                        self.input += chr
                if len(self.input) == 0 and self.timedout:
                    break    


    sys.stdout.write('%s(%s):'%(caption, default));
    result = default
    it = KeyboardThread()
    it.start()
    it.join(timeout)
    it.timedout = True
    if len(it.input) > 0:
        # wait for rest of input
        it.join()
        result = it.input
    print ''  # needed to move to next line
    return result

# and some examples of usage
ans = readInput('Please type a name', 'john') 
print 'The name is %s' % ans
ans = readInput('Please enter a number', 10 ) 
print 'The number is %s' % ans 

Answer 6

Следующий код работал для меня.

Я использовал два потока, один для получения raw_Input, а другой для ожидания в течение определенного времени. Если какой-либо поток завершается, оба потока завершаются и возвращаются.

def _input(msg, q):
    ra = raw_input(msg)
    if ra:
        q.put(ra)
    else:
        q.put("None")
    return

def _slp(tm, q):
    time.sleep(tm)
    q.put("Timeout")
    return

def wait_for_input(msg="Press Enter to continue", time=10):
    q = Queue.Queue()
    th = threading.Thread(target=_input, args=(msg, q,))
    tt = threading.Thread(target=_slp, args=(time, q,))

    th.start()
    tt.start()
    ret = None
    while True:
        ret = q.get()
        if ret:
            th._Thread__stop()
            tt._Thread__stop()
            return ret
    return ret

print time.ctime()    
t= wait_for_input()
print "\nResponse :",t 
print time.ctime()

Answer 7

Я потратил на это добрых двадцать минут или около того, поэтому подумал, что это стоит того, чтобы поставить это здесь. Тем не менее, он напрямую строит ответ пользователя 137673. Я нашел наиболее полезным сделать что-то вроде этого:

#! /usr/bin/env python

import signal

timeout = None

def main():
    inp = stdinWait("You have 5 seconds to type text and press <Enter>... ", "[no text]", 5, "Aw man! You ran out of time!!")
    if not timeout:
        print "You entered", inp
    else:
        print "You didn't enter anything because I'm on a tight schedule!"

def stdinWait(text, default, time, timeoutDisplay = None, **kwargs):
    signal.signal(signal.SIGALRM, interrupt)
    signal.alarm(time) # sets timeout
    global timeout
    try:
        inp = raw_input(text)
        signal.alarm(0)
        timeout = False
    except (KeyboardInterrupt):
        printInterrupt = kwargs.get("printInterrupt", True)
        if printInterrupt:
            print "Keyboard interrupt"
        timeout = True # Do this so you don't mistakenly get input when there is none
        inp = default
    except:
        timeout = True
        if not timeoutDisplay is None:
            print timeoutDisplay
        signal.alarm(0)
        inp = default
    return inp

def interrupt(signum, frame):
    raise Exception("")

if __name__ == "__main__":
    main()

Answer 8

Аналог Locane's для Windows:

import subprocess  
subprocess.call('timeout /T 30')

Answer 9

Вот переносимое и простое решение Python 3 с использованием потоков. Это единственный, который работал для меня, будучи кроссплатформенным.

У других вещей, которые я пробовал, были проблемы:

• Использование сигнала. SIGALRM: не работает в Windows
• Использование вызова вызова: не работает в Windows
• Использование принудительного завершения процесса (вместо потока): stdin нельзя использовать в новом процессе (stdin автоматически закрывается)
• Перенаправление stdin в StringIO и запись непосредственно в stdin: все равно будет писать в предыдущий stdin, если уже был вызван input () (см. https://stackoverflow.com/a/15055639/9624704)

    from threading import Thread
    class myClass:
        _input = None

        def __init__(self):
            get_input_thread = Thread(target=self.get_input)
            get_input_thread.daemon = True  # Otherwise the thread won't be terminated when the main program terminates.
            get_input_thread.start()
            get_input_thread.join(timeout=20)

            if myClass._input is None:
                print("No input was given within 20 seconds")
            else:
                print("Input given was: {}".format(myClass._input))


        @classmethod
        def get_input(cls):
            cls._input = input("")
            return

Answer 10

мое кроссплатформенное решение

def input_process(stdin_fd, sq, str):
    sys.stdin = os.fdopen(stdin_fd)
    try:
        inp = input (str)
        sq.put (True)
    except:
        sq.put (False)

def input_in_time (str, max_time_sec):
    sq = multiprocessing.Queue()
    p = multiprocessing.Process(target=input_process, args=( sys.stdin.fileno(), sq, str))
    p.start()
    t = time.time()
    inp = False
    while True:
        if not sq.empty():
            inp = sq.get()
            break
        if time.time() - t > max_time_sec:
            break
    p.terminate()
    sys.stdin = os.fdopen( sys.stdin.fileno() )
    return inp