Python中如何在input()运行时检测箭头键实现REPL历史输入循环

Python自带的input()函数是行缓冲式读取，无法捕获上下箭头这类特殊按键，要实现类似官方REPL的历史切换功能，最直接的方案是用标准库的readline模块（Windows平台需用pyreadline3替代）。

方法一：使用标准库readline（Linux/macOS） #

readline模块是Python官方REPL的底层依赖，导入后会自动增强输入功能，支持上下箭头遍历历史、Tab补全等特性。以下是完整示例：

import readline

# 可选：设置历史记录持久化文件
HISTORY_FILE = ".my_repl_history"

# 加载已保存的历史记录
try:
    readline.read_history_file(HISTORY_FILE)
except FileNotFoundError:
    pass

# 设置历史记录最大保留条数
readline.set_history_length(1000)

# REPL主循环
while True:
    try:
        code = input(" > ")
        if not code.strip():
            continue
        
        # 这里添加你的代码处理逻辑
        print(f"执行: {code}")
        
        # 将有效输入加入历史记录
        readline.add_history(code)
    except KeyboardInterrupt:
        print("\n退出REPL")
        break
    except EOFError:
        print("\n退出REPL")
        break

# 退出时保存历史记录
readline.write_history_file(HISTORY_FILE)

方法二：Windows平台兼容方案 #

Windows系统的Python默认没有readline模块，需要安装第三方兼容库pyreadline3，API和标准readline完全一致：

1. 安装依赖：

pip install pyreadline3

2. 代码和上面完全相同，导入readline即可，pyreadline3会自动接管输入处理。

原理说明 #

input()函数仅读取到回车为止的文本，不会处理终端的按键事件；而readline模块会切换终端到原始模式，直接捕获上下箭头等特殊按键，通过维护历史输入列表，实现按键切换历史的功能。

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