from machine import Pin
import time

# 簡化按鈕的接腳設置
def button(input_pin):
    return Pin(input_pin, Pin.IN, Pin.PULL_UP)

# 簡化 led 的接腳設置
def led(output_pin):
    return Pin(output_pin, Pin.OUT)

# 設定不同顏色按鈕和 led 的接腳
pins = {
    'green' : (button(32), led(21)),
    'yellow' : (button(33), led(17)),
    'red' : (button(25), led(16))
}

# 讓每個 led 在按下對應顏色的按鈕時發亮
while True:
    for color, (button, led) in pins.items():
        # print(button.value())
        led.value(not button.value())
        time.sleep(0.01)