Python中如何实现等待按键释放后消费stdin剩余输入？

问题描述 #

目标是编写程序，等待按键释放后消费标准输入（stdin），避免后续程序获取按键按住产生的输入。现有WaitForRelease.py脚本如下：

#!/usr/bin/env python

from pynput import keyboard
import sys
import select

def on_key_release(key):
    print('Released Key %s' % key)
    return False


with keyboard.Listener(on_release = on_key_release) as listener:
    listener.join()

while True:
    if select.select([sys.stdin,],[],[],0.5)[0]:
        x = sys.stdin.read()
        print("Got: ", x)
    else:
        break

运行命令：

stty -echo; ./WaitForRelease.py; stty echo; cat

操作步骤：按住f键5秒后释放，再按Control-D给cat发EOF。

预期输出：

Released key 'f'

Got: ffffffffffffffffffffff

实际输出：

Released Key 'f'

fffffffffffffffffffffffffffffff

核心问题：cat获取了多余的f字符，Python程序未消费这些输入；同时需要支持处理程序启动前就按住的按键。

原因分析 #

1. 终端缓冲机制冲突：按住按键时，终端会先将输入字符存入自身缓冲区，不会立刻传递给程序。pynput的键盘监听直接与系统输入设备交互，和终端stdin缓冲区是两套独立机制——监听捕获释放事件时，终端缓冲区已存储大量f，但Python的select+read因终端行缓冲模式未触发读取，导致字符残留，后续cat启动后直接读取。
2. sys.stdin.read()行为限制：默认终端为行缓冲模式，read()需等待换行符或EOF才返回。你按下的Control-D是发给cat的EOF，而非Python程序，因此Python的read()未读取到内容，超时退出后缓冲区字符全留给了cat。
3. 启动前按键未被捕获：pynput的Listener启动后才开始监听，无法捕获程序启动前已按住的按键释放事件。

解决方案 #

修改思路 #

• 切换终端为无缓冲raw模式，让输入字符直接传递给程序，避免终端缓存。
• 直接操作终端属性，替代外部stty命令，确保程序运行期间终端模式正确。
• 处理启动前按住的按键：先检查当前是否有按键按住，等待所有按键释放后再消费缓冲区内容。

修改后的脚本 #

#!/usr/bin/env python
import sys
import termios
import tty
from pynput import keyboard

def get_terminal_attr():
    # 获取当前终端属性，用于后续恢复
    return termios.tcgetattr(sys.stdin.fileno())

def set_raw_mode(fd):
    # 设置终端为无缓冲raw模式
    attrs = termios.tcgetattr(fd)
    tty.setraw(fd)
    return attrs

def restore_terminal_attr(fd, old_attrs):
    # 恢复终端原有属性
    termios.tcsetattr(fd, termios.TCSADRAIN, old_attrs)

def wait_for_any_key_release():
    # 等待所有按键释放（包括程序启动前按住的）
    pressed_keys = set()
    def on_press(key):
        pressed_keys.add(key)
    def on_release(key):
        if key in pressed_keys:
            pressed_keys.remove(key)
        # 所有按键释放后退出监听
        if not pressed_keys:
            print(f'Released all keys')
            return False

    with keyboard.Listener(on_press=on_press, on_release=on_release) as listener:
        listener.join()

def consume_stdin():
    # 非阻塞消费stdin中所有剩余字符
    fd = sys.stdin.fileno()
    old_attrs = termios.tcgetattr(fd)
    try:
        # 设置终端为非阻塞读取模式
        attrs = termios.tcgetattr(fd)
        attrs[6][termios.VMIN] = 0
        attrs[6][termios.VTIME] = 0
        termios.tcsetattr(fd, termios.TCSADRAIN, attrs)
        
        content = b''
        while True:
            try:
                chunk = sys.stdin.read(1024)
                if not chunk:
                    break
                content += chunk.encode() if isinstance(chunk, str) else chunk
            except:
                break
        if content:
            print(f"Got: {content.decode()}")
    finally:
        termios.tcsetattr(fd, termios.TCSADRAIN, old_attrs)

def main():
    fd = sys.stdin.fileno()
    old_attrs = get_terminal_attr()
    try:
        # 先设置raw模式，避免终端缓存输入
        set_raw_mode(fd)
        # 等待所有按键释放
        wait_for_any_key_release()
        # 消费所有剩余输入
        consume_stdin()
    finally:
        # 无论程序是否异常，都恢复终端属性
        restore_terminal_attr(fd, old_attrs)

if __name__ == "__main__":
    main()

调用方式 #

无需额外stty命令，直接运行：

./WaitForRelease.py; cat

效果说明 #

1. 若启动脚本前按住f，脚本会等待你释放f后，自动消费终端缓冲区里所有的f，输出Got: ffffffffffffffffffffff。
2. 后续启动的cat不会再获取这些残留的f字符，仅会输出你之后输入的内容。

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