如何用turtle的onkeypress与listen实现动态按键切换画笔颜色？

你正在用Python的turtle库开发网格绘图程序，箭头键移动功能已实现，需求是通过直接输入两位数字切换画笔颜色（无需在shell回车），比如输入02切换蓝色、21切换黄色。但当前代码存在以下问题：

• 按下数字后画笔直接变为函数中最后一个颜色（比如按0直接变成09对应的绿色）
• 非0/1/2的数字无响应
• 使用keyboard库的无限循环会导致程序崩溃

原代码问题分析 #

1. 函数绑定错误：在zero()等函数中，onkeypress(change_colour(colour), "1")是直接执行了change_colour函数，而非将函数绑定到按键事件，导致程序启动或按下第一个数字时就触发所有change_colour调用，最终只保留最后一次的颜色设置。
2. 状态管理缺失：没有区分「等待输入第一位数字」和「等待输入第二位数字」的状态，无法正确处理两位数字的输入流程。
3. 冗余依赖：keyboard库无需引入，turtle自身的事件循环足以处理按键事件，额外的循环会阻塞turtle的主循环导致程序无响应。

解决方案 #

通过状态变量跟踪当前输入阶段（等待第一位/第二位数字），配合颜色映射字典实现两位数字到颜色的对应，同时利用turtle的原生事件绑定机制处理按键，避免阻塞主循环。

完整代码示例 #

from turtle import Turtle, Screen, listen, onkeypress

# 颜色映射：两位数字对应颜色，可按需扩展
COLOR_MAP = {
    "02": "blue",
    "21": "yellow",
    "09": "green",
    "11": "red",
    "15": "purple"
}

# 全局状态变量：记录已输入的第一位数字，None表示等待输入第一位
current_prefix = None
pen = Turtle()
screen = Screen()

def reset_input_state():
    """重置输入状态，回到等待第一位数字的状态"""
    global current_prefix
    current_prefix = None
    # 重新绑定允许作为第一位的数字按键
    bind_first_digit_keys()

def handle_color_input(digit):
    """处理数字按键，根据当前状态判断是第一位还是第二位输入"""
    global current_prefix
    if current_prefix is None:
        # 记录第一位数字，切换到等待第二位的状态
        current_prefix = digit
        # 解绑第一位数字按键，绑定所有数字作为第二位输入
        unbind_all_digits()
        for d in "0123456789":
            onkeypress(lambda d=d: handle_color_input(d), d)
    else:
        # 拼接两位颜色代码，切换颜色
        color_code = current_prefix + digit
        if color_code in COLOR_MAP:
            pen.pencolor(COLOR_MAP[color_code])
            pen.fillcolor(COLOR_MAP[color_code])
        # 完成输入后重置状态
        reset_input_state()

def bind_first_digit_keys():
    """绑定允许作为第一位的数字按键（这里设置为0/1/2）"""
    unbind_all_digits()
    onkeypress(lambda: handle_color_input("0"), "0")
    onkeypress(lambda: handle_color_input("1"), "1")
    onkeypress(lambda: handle_color_input("2"), "2")

def unbind_all_digits():
    """解绑所有数字按键的响应"""
    for d in "0123456789":
        onkeypress(None, d)

# 初始化：绑定第一位数字按键
bind_first_digit_keys()
listen()

# 以下是你已实现的箭头键移动逻辑示例
def move_up():
    pen.setheading(90)
    pen.forward(20)

def move_down():
    pen.setheading(270)
    pen.forward(20)

def move_left():
    pen.setheading(180)
    pen.forward(20)

def move_right():
    pen.setheading(0)
    pen.forward(20)

onkeypress(move_up, "Up")
onkeypress(move_down, "Down")
onkeypress(move_left, "Left")
onkeypress(move_right, "Right")

# 启动turtle主事件循环
screen.mainloop()

代码说明 #

1. 颜色映射字典：COLOR_MAP定义了两位数字代码与对应颜色的映射关系，可根据需求自由添加或修改。
2. 状态管理：current_prefix变量跟踪输入进度，None表示等待输入第一位数字，非None时表示等待输入第二位数字。
3. 按键绑定逻辑：
   • 初始状态仅绑定0/1/2作为第一位数字输入
   • 输入第一位数字后，自动切换为绑定所有数字作为第二位输入
   • 完成两位输入后，自动重置状态，回到等待第一位数字的状态
4. 无阻塞事件处理：完全依赖turtle原生的listen()和onkeypress()，通过screen.mainloop()维持事件循环，避免程序崩溃。

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