Python基于msvcrt实现命令行输入时如何按上方向键调取历史输入

这个功能完全可以实现，核心逻辑是维护持久化的输入历史栈，在方向键触发时清空重绘当前行内容即可。你原有代码存在两个核心问题导致功能无法正常运行：

• 方向键判断逻辑错误：上方向键的第二返回值是字节类型b'H'，原有代码写的str(char) == "H"判断永远不会触发
• 缺少行重绘逻辑：切换历史输入时如果不清除当前行已有内容，会出现新旧文字重叠的显示错乱问题

具体实现逻辑如下：

• 维护一个持久生效的输入历史列表，搭配位置指针标记当前浏览到的历史条目，指针值为-1时代表当前处于未提交的新输入状态
• 用户按回车提交输入时，非空内容存入历史列表，同时重置指针到新输入状态
• 检测到上方向键按下时，指针向前移动（指向更早的输入条目），清空当前行后打印对应历史内容，同步更新当前正在编辑的字符串变量
• 检测到下方向键按下时，指针向后移动（指向更近的输入条目），指针回到-1位置时切换为空的新输入状态，同步重绘当前行
• 退格、普通字符输入操作也要同步重绘当前行，保证屏幕显示内容和实际存储的编辑内容一致

修改后可直接运行的代码如下：

import msvcrt

def test_input(prompt="> ", end="\n"):
    # 第一次调用时初始化历史存储和位置指针
    if not hasattr(test_input, "input_history"):
        test_input.input_history = []
        test_input.history_pos = -1
    current_str = ""
    print(prompt, end="", flush=True)
    
    while True:
        key = msvcrt.getch()
        # 回车提交输入
        if key == b"\r":
            print(end)
            # 非空输入才存入历史，避免冗余空行
            if current_str.strip():
                test_input.input_history.append(current_str)
            # 重置指针到新输入状态
            test_input.history_pos = -1
            return current_str
        # 处理功能键（方向键前缀为0xe0）
        elif key == b'\xe0':
            arrow_key = msvcrt.getch()
            # 上方向键：翻更早的历史
            if arrow_key == b'H':
                if test_input.history_pos < len(test_input.input_history) - 1:
                    test_input.history_pos += 1
                    current_str = test_input.input_history[-(test_input.history_pos + 1)]
                    # 清空当前行重绘内容
                    print(f"\r\x1b[K{prompt}{current_str}", end="", flush=True)
            # 下方向键：翻更近的历史/回到新输入状态
            elif arrow_key == b'P':
                if test_input.history_pos > -1:
                    test_input.history_pos -= 1
                    current_str = "" if test_input.history_pos == -1 else test_input.input_history[-(test_input.history_pos + 1)]
                    print(f"\r\x1b[K{prompt}{current_str}", end="", flush=True)
        # 退格键处理
        elif key == b'\x08':
            if current_str:
                current_str = current_str[:-1]
                print(f"\r\x1b[K{prompt}{current_str}", end="", flush=True)
        # 普通可打印字符处理
        else:
            try:
                # Windows终端默认GBK编码，解决中文输入解码报错
                char = key.decode("gbk")
                current_str += char
                print(char, end="", flush=True)
            except UnicodeDecodeError:
                # 忽略其他无处理逻辑的特殊功能键
                pass

补充说明：

• 上述代码的历史记录存储在函数静态属性中，多次调用test_input()时历史会持续保留，和Windows CMD的行为完全一致
• 逐字符校验逻辑可以直接添加在普通可打印字符的处理分支中，不符合规则的字符直接跳过，不追加到编辑字符串也不打印即可
• 如果需要支持左右方向键移动光标、指定位置插入/删除字符，可以在现有逻辑基础上新增光标位置变量，重绘时按照光标位置渲染内容即可

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