Python3:select.select()中的readline等价物

时间:2013-11-21 15:02:09

标签: python python-3.x gnu readline

在Python脚本中,程序员可以导入readline,然后提供input()扩展功能(readline有许多其他用途)。我想在我的脚本中使用select.select()而不是input(),因为我喜欢超时功能。但是,当导入readline时,我无法使用input()从readline获取的功能。我所指的“扩展功能”的一个示例是能够按下向上键并查看上一个输入或使用左右箭头键移动内联光标以更改输入。

问题:如何使select.select()具有GNU-readline功能?这甚至可能吗?

编辑:为了防止你们有什么好奇我想要完成什么,我做了一个基于终端的聊天机器人(有点像Alicebot)。如果钻头在一定时间内没有收到任何输入,我希望机器人感到无聊并做其他事情。 (https://launchpad.net/neobot

2 个答案:

答案 0 :(得分:4)

您可以使用readline模块的readline.set_pre_input_hook([function])机制来完成此任务。

这是一个在没有输入10秒后超时的示例 - 未实现的是禁用警报的机制,如果提供了输入。

这必须以线程的方式完成,因为信号不能遍历线程。但是,你得到了基本的想法..

我为此代码的推进而道歉,我正在我的笔记本电脑上的一家咖啡店,只是有点打耳光。这是python2.7代码,但基本上应与python3兼容 - 概念是重要的部分。

如果你想让每一行输入都超时,我想你会想要在input_loop()函数的开头处把警报禁用。

您还应该查看Python模块树中的readline.c源代码以获取更多想法。

#!/usr/bin/python

import readline
import logging
import signal
import os

LOG_FILENAME = '/tmp/completer.log'
HISTORY_FILENAME = '/tmp/completer.hist'

logging.basicConfig(filename=LOG_FILENAME, level=logging.DEBUG,)

class YouAreTooSlow(Exception): pass


def get_history_items():
    return [ readline.get_history_item(i)
             for i in xrange(1, readline.get_current_history_length() + 1)
             ]

class HistoryCompleter(object):

    def __init__(self):
        self.matches = []
        return

    def complete(self, text, state):
        response = None
        if state == 0:
            history_values = get_history_items()
            logging.debug('history: %s', history_values)
            if text:
                self.matches = sorted(h 
                                      for h in history_values 
                                      if h and h.startswith(text))
            else:
                self.matches = []
            logging.debug('matches: %s', self.matches)
        try:
            response = self.matches[state]
        except IndexError:
            response = None
        logging.debug('complete(%s, %s) => %s', 
                      repr(text), state, repr(response))
        return response

def input_loop():
    if os.path.exists(HISTORY_FILENAME):
        readline.read_history_file(HISTORY_FILENAME)
    print 'Max history file length:', readline.get_history_length()
    print 'Startup history:', get_history_items()
    try:
        while True:
            line = raw_input('Prompt ("stop" to quit): ')
            if line == 'stop':
                break
            if line:
                print 'Adding "%s" to the history' % line
    finally:
        print 'Final history:', get_history_items()
        readline.write_history_file(HISTORY_FILENAME)

# Register our completer function

def slow_handler(signum, frame):
    print 'Signal handler called with signal', signum
    raise YouAreTooSlow()

def pre_input_hook():
    signal.signal(signal.SIGALRM, slow_handler)
    signal.alarm(10)

readline.set_pre_input_hook(pre_input_hook)
readline.set_completer(HistoryCompleter().complete)

# Use the tab key for completion
readline.parse_and_bind('tab: complete')

# Prompt the user for text
input_loop()

答案 1 :(得分:-1)

似乎readline的设计考虑了这种能力(link)。

  

普通readline()可以使用备用接口。某些应用程序需要将键盘I / O与文件,设备或窗口系统I / O交错,通常使用主循环在各种文件描述符上选择()。为了满足这种需要,readline也可以作为“回调”来调用。来自事件循环的函数。有一些功能可以使这很容易。

然而,似乎这些绑定并未在Python中实现。快速搜索一下,here的一个小伙子通过使用CType加载函数调用来完成POC。这不是理想的,但缺乏任何替代方案,也许这是你唯一的行动方案。