如何让Python urwid程序支持管道输入并正常启动UI？

问题原因 #

当通过管道（如ls | python script.py）向程序输入数据时，stdin会被数据填满并进入EOF状态。urwid的MainLoop启动时默认会从stdin读取用户输入，此时读到空字符串会触发TypeError: ord() expected a character, but string of length 0 found错误。

解决方案 #

核心思路是先读取管道数据，再让urwid从终端读取用户输入，以下是两种可行实现方式：

方法一：重置stdin为终端设备 #

先读取全部管道数据，再将stdin重新指向终端，恢复urwid的输入源：

import urwid
import sys
import os

# 读取管道中的所有数据
pipe_data = sys.stdin.read()

# 重新打开stdin为终端设备（Linux/macOS用/dev/tty，Windows用con）
sys.stdin = open('/dev/tty', 'r')

# 初始化urwid UI元素
txt = urwid.Text(f"已读取管道数据，长度: {len(pipe_data)}")
filler = urwid.Filler(txt)

def quitter(key):
    if key == "q":
        raise urwid.ExitMainLoop()

# 启动UI循环
loop = urwid.MainLoop(filler, unhandled_input=quitter)
loop.run()

# UI关闭后可处理管道数据并输出
# print(pipe_data)

方法二：指定urwid的输入源 #

通过自定义RawInputHandler，让urwid直接从终端读取输入，绕过已EOF的stdin：

import urwid
import sys
import os

# 读取管道数据
pipe_data = sys.stdin.read()

# 打开终端设备作为输入源
tty_fd = os.open('/dev/tty', os.O_RDWR)
input_handler = urwid.raw_display.RawInputHandler(tty_fd)

# 初始化UI
txt = urwid.Text("管道数据读取完成，可进行交互")
filler = urwid.Filler(txt)

def quitter(key):
    if key == "q":
        raise urwid.ExitMainLoop()

# 传入自定义输入源启动循环
loop = urwid.MainLoop(filler, unhandled_input=quitter, input_handler=input_handler)
loop.run()

# 后续处理管道数据

注意事项 #

• 管道数据必须在urwid启动前读取完毕，否则会触发原始报错。
• 跨平台适配：Linux/macOS使用/dev/tty，Windows系统替换为con。
• UI关闭后，可对pipe_data进行过滤、排序等处理，再输出到stdout，实现类似fzf的筛选功能。

内容的提问来源于stack exchange，提问作者polm23