Linux下Python使用select处理ESC序列按键延迟问题求助

问题核心：以ESC开头的特殊按键序列（比如上箭头对应ESC [ A），用select单字符读取时，ESC后的后续字符不会触发select唤醒——因为终端默认会等待一段时间，判断是单独按下ESC还是特殊键序列的一部分。单独按下ESC时，read还会阻塞到下一次按键，导致输入处理延迟。

解决方案思路 #

读取到ESC字符后，临时修改终端的输入超时设置，用非阻塞方式尝试读取后续字符，通过短超时区分单独ESC和特殊键序列：

• 设置VTIME（等待时间，单位0.1秒）为1（即0.1秒），VMIN（最小读取字符数）为0，这样read会在超时前读取所有可用字符，不会阻塞。
• 读取完成后立刻恢复终端原始设置，避免影响后续输入逻辑。

修改后的代码 #

#!/usr/bin/env python

import sys
import tty
import termios
import select, time

old_settings = termios.tcgetattr(sys.stdin)
ts = time.time()

def read_special_sequence(fd):
    # 临时修改终端设置，读取ESC后续字符，超时0.1秒
    temp_settings = termios.tcgetattr(fd)
    try:
        # VMIN=0: 读取0或多个字符；VTIME=1: 等待0.1秒后超时
        temp_settings[6][termios.VMIN] = 0
        temp_settings[6][termios.VTIME] = 1
        termios.tcsetattr(fd, termios.TCSADRAIN, temp_settings)
        
        # 读取所有可用的后续字符
        sequence = []
        while True:
            rlist, _, _ = select.select([fd], [], [], 0)
            if rlist:
                char = sys.stdin.read(1)
                sequence.append(char)
            else:
                break
        return sequence
    finally:
        # 恢复原始终端设置
        termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)

try:
    tty.setraw(sys.stdin.fileno())
    while True:
        rlist, _, _ = select.select([sys.stdin], [], [], 2)
        if rlist:
            char = sys.stdin.read(1)
            if ord(char) == 27:
                seq = read_special_sequence(sys.stdin.fileno())
                if seq:
                    print(f'at {time.time()-ts:6.2f} ESC sequence: ESC{"".join(seq)}', '\x0d')
                    # 可在这里添加特殊键判断逻辑，比如识别上箭头
                    if seq == ['[', 'A']:
                        print(f'at {time.time()-ts:6.2f} 上箭头被按下', '\x0d')
                else:
                    print(f'at {time.time()-ts:6.2f} ESCAPE!', '\x0d')
            else:
                print(f'at {time.time()-ts:6.2f} got', char if ord(char) >32 else f'{{{ord(char)}}}', '\x0d')
                if char == 'x':
                    break
finally:
    termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_settings)
    print('byeeeee')

效果说明 #

• 按下上箭头这类特殊键时，会一次性读取完整的ESC序列，不会延迟到下一次按键才显示后续字符
• 单独按下ESC时，0.1秒后会被识别为单独按键，不会阻塞等待后续输入
• 无需root权限，兼容SSH连接的无桌面环境（如树莓派服务器）

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