从 raw_input() 读取输入，而不会让 Python 中的其他线程覆盖提示符

Question

我试图让用户使用 raw_input() 在控制台输入命令，这工作正常。问题是我的后台线程偶尔会向屏幕输出日志信息，当它们这样做时，它们会弄乱输入提示（因为输出会转到光标当前所在的位置）。

这是一个小的 Python 程序，它说明了我的意思。

#!/usr/bin/env python
import threading
import time

def message_loop():
    while True:
        time.sleep(1)
        print "Hello World"

thread = threading.Thread(target = message_loop)
thread.start()

while True:
    input = raw_input("Prompt> ")
    print "You typed", input

这是我运行它时的示例：

Prompt> Hello World
Hello World
Hello World
Hello World
test
You typed test
Prompt> Hello World
Hello World
Hello World
hellHello World
o
You typed hello
Prompt> Hello World
Hello World
Hello World
Hello World

我想要的是提示符随线程的输出一起移动。就像这样：

Hello World
Hello World
Prompt> test
You typed test
Hello World
Hello World
Hello World
Hello World
Hello World
Prompt> hello
You typed hello
Hello World
Hello World
Hello World
Hello World
Prompt> 

关于如何在不诉诸丑陋的黑客的情况下实现这一目标的任何想法？ :)

Answer 1

我最近遇到了这个问题，想把这个解决方案留在这里以供将来参考。
这些解决方案从终端清除待处理的 raw_input（readline）文本，打印新文本，然后将 raw_input 缓冲区中的内容重新打印到终端。

第一个程序非常简单，但只有当只有 1 行文本等待 raw_input 时才能正常工作：

#!/usr/bin/python

import time,readline,thread,sys

def noisy_thread():
    while True:
        time.sleep(3)
        sys.stdout.write('\r'+' '*(len(readline.get_line_buffer())+2)+'\r')
        print 'Interrupting text!'
        sys.stdout.write('> ' + readline.get_line_buffer())
        sys.stdout.flush()

thread.start_new_thread(noisy_thread, ())
while True:
    s = raw_input('> ')

输出：

$ ./threads_input.py
Interrupting text!
Interrupting text!
Interrupting text!
> WELL, PRINCE, Genoa and Lucca are now no more than private estates of the Bo
Interrupting text!
> WELL, PRINCE, Genoa and Lucca are now no more than private estates of the Bo
naparte family. No, I warn you, that if you do not tell me we are at war,

第二个程序正确处理 2 个或更多缓冲行，但具有更多（标准）模块依赖性，并且需要一点终端黑客：

#!/usr/bin/python

import time,readline,thread
import sys,struct,fcntl,termios

def blank_current_readline():
    # Next line said to be reasonably portable for various Unixes
    (rows,cols) = struct.unpack('hh', fcntl.ioctl(sys.stdout, termios.TIOCGWINSZ,'1234'))

    text_len = len(readline.get_line_buffer())+2

    # ANSI escape sequences (All VT100 except ESC[0G)
    sys.stdout.write('\x1b[2K')                         # Clear current line
    sys.stdout.write('\x1b[1A\x1b[2K'*(text_len/cols))  # Move cursor up and clear line
    sys.stdout.write('\x1b[0G')                         # Move to start of line


def noisy_thread():
    while True:
        time.sleep(3)
        blank_current_readline()
        print 'Interrupting text!'
        sys.stdout.write('> ' + readline.get_line_buffer())
        sys.stdout.flush()          # Needed or text doesn't show until a key is pressed


if __name__ == '__main__':
    thread.start_new_thread(noisy_thread, ())
    while True:
        s = raw_input('> ')

输出。正确清除以前的阅读行：

$ ./threads_input2.py
Interrupting text!
Interrupting text!
Interrupting text!
Interrupting text!
> WELL, PRINCE, Genoa and Lucca are now no more than private estates of the Bo
naparte family. No, I warn you, that if you do not tell me we are at war,

---

有用的来源：

How to get Linux console window width在Python中

apt像列输出 - python库
（此代码示例展示了如何获取 Unix 或 Windows 的终端宽度）

http://en.wikipedia.org /wiki/ANSI_escape_code

Answer 2

我认为您需要一些可以让您从终端窗口动态打印/删除/覆盖文本的东西，例如 UNIX watch 或 top 命令如何工作。

我认为在你的情况下你会打印“Prompt>”但是当你收到“Hello World”时，你会覆盖“Prompt>”与“Hello World”，然后打印“Prompt>”在下面一行。我认为你不能通过定期输出打印到终端来做到这一点。

您也许可以使用 Python 的 curses 库来做您想做的事情。我从未使用过它，所以我无法告诉您如何解决您的问题（或者该模块是否能够解决您的问题），但我认为值得研究一下。搜索“pythoncurses教程”提供了PDF教程文档这似乎有帮助。

Answer 3

你需要从单个线程更新 stdout，而不是从多个线程......否则你无法控制交错 I/O。

您将需要创建一个用于输出写入的线程。

您可以在线程中使用队列，并让所有其他线程将其输出日志记录信息写入其中。然后从此队列中读取数据，并在适当的时间连同提示消息一起写入标准输出。

Answer 4

我认为这是不可能的。无论如何，那应该如何表现？在用户按下 Enter 之前什么都不会显示？如果是这样，只有当用户发出命令（或系统期望的任何命令）时才会出现输出，这听起来并不理想。

我认为你的线程应该输出到另一个文件。