import machine
import time


red_led = machine.Pin(21, machine.Pin.OUT)
yellow_led = machine.Pin(19,machine.Pin.OUT)
green_led = machine.Pin(18,machine.Pin.OUT)

a = machine.Pin(32, machine.Pin.OUT)
b = machine.Pin(33, machine.Pin.OUT)
c = machine.Pin(25, machine.Pin.OUT)
d = machine.Pin(26, machine.Pin.OUT)
e = machine.Pin(27, machine.Pin.OUT)
f = machine.Pin(14, machine.Pin.OUT)
g = machine.Pin(12, machine.Pin.OUT)
dot = machine.Pin(13, machine.Pin.OUT)

number_led = [a, b, c, d, e, f, g, dot]

number_dict = {
    0: "11111100",
    1: "01100000",
    2: "11011010",
    3: "11110010",
    4: "01100110",
    5: "10110110",
    6: "10111110",
    7: "11100000",
    8: "11111110",
    9: "11110110",
    "open": "11111111",
    "close": "00000000"
}

def show_number(number):
    if number_dict.get(number):
        i = 0
        for bit in number_dict.get(number):
            if bit == "1":
                number_led[i].value(1)
            else:
                number_led[i].value(0)
            i += 1

def main():
    while True:
        # 绿灯亮10秒
        green_led.value(1)
        for i in range(10) :
            show_number(i)
            time.sleep(1)
            print("绿灯亮，第{}秒".format(i + 1))
        green_led.value(0)
        
        # 黄灯亮3秒
        yellow_led.value(1)
        for i in range(3) :
            show_number("close")
            time.sleep(0.5)
            show_number("open")
            time.sleep(0.5)
            print("黄灯亮，第{}秒".format(i + 1))
        yellow_led.value(0)

        # 红灯亮10秒
        red_led.value(1)
        for i in range(10) :
            show_number(i)
            time.sleep(1)
            print("红灯亮，第{}秒".format(i + 1))
        red_led.value(0)

if __name__ == "__main__":
    main()