Python程序退出前如何清除未消费的用户输入？

问题背景 #

当Python程序在消费完所有输入前退出时，剩余输入会被发送至Shell，引发不必要的命令错误。例如运行以下代码：

import sys
for line in sys.stdin:
    sys.exit()

粘贴多行输入foo、bar、baz后，程序读取foo即退出，bar会被Shell当作命令执行导致报错。且不能修改示例中的for循环（代表复杂业务逻辑），此前用atexit注册sys.stdin.read()的方案会导致程序挂起，直到用户输入空行，因此需要无暂停的剩余输入清除方案。

解决方案 #

类Unix系统（Linux/macOS）：非阻塞读取标准输入 #

通过修改标准输入的文件状态标志为非阻塞模式，在程序退出前循环读取剩余输入，无数据时直接抛出异常终止读取，避免挂起。

import sys
import atexit
import fcntl
import os

def clear_remaining_stdin():
    fd = sys.stdin.fileno()
    # 保存原始文件状态标志
    original_flags = fcntl.fcntl(fd, fcntl.F_GETFL)
    try:
        # 将stdin设置为非阻塞模式
        fcntl.fcntl(fd, fcntl.F_SETFL, original_flags | os.O_NONBLOCK)
        # 循环读取直到无数据（抛出BlockingIOError）
        while True:
            try:
                sys.stdin.read()
            except BlockingIOError:
                break
    finally:
        # 恢复原始标志（可选，程序即将退出，不恢复也不影响）
        fcntl.fcntl(fd, fcntl.F_SETFL, original_flags)

# 注册退出时的清理函数
atexit.register(clear_remaining_stdin)

# 原有逻辑（不可修改）
for line in sys.stdin:
    sys.exit()

Windows系统：读取控制台输入缓冲区 #

使用pywin32库直接读取控制台输入缓冲区的剩余记录，无需等待用户输入。

1. 先安装依赖：

pip install pywin32

2. 示例代码：

import sys
import atexit
import win32console

def clear_remaining_stdin():
    stdin_handle = win32console.GetStdHandle(win32console.STD_INPUT_HANDLE)
    # 获取输入缓冲区中待处理的记录数
    record_count = stdin_handle.GetNumberOfConsoleInputRecords()
    if record_count > 0:
        # 读取所有剩余输入记录
        stdin_handle.ReadConsoleInput(record_count)

atexit.register(clear_remaining_stdin)

# 原有逻辑（不可修改）
for line in sys.stdin:
    sys.exit()

方案说明 #

• 类Unix方案通过非阻塞IO实现无等待读取，一旦输入缓冲区为空就停止，不会挂起程序。
• Windows方案直接读取控制台输入缓冲区的所有剩余内容，无需依赖非阻塞模式，兼容性更好。

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