from machine import Pin
from time import sleep

red = Pin(7, Pin.OUT)
yellow = Pin(8, Pin.OUT)
green = Pin(9, Pin.OUT)

pins = [
    Pin(0, Pin.OUT),
    Pin(1, Pin.OUT),
    Pin(2, Pin.OUT),
    Pin(3, Pin.OUT),
    Pin(4, Pin.OUT),
    Pin(5, Pin.OUT),
    Pin(6, Pin.OUT)
]

def display(num, delay):
    for i in range(7):
        pins[i].value(num[i])
    sleep(delay)

digits = {
    0: [0, 0, 0, 0, 0, 0, 1],
    1: [1, 0, 0, 1, 1, 1, 1],
    2: [0, 0, 1, 0, 0, 1, 0],
    3: [0, 0, 0, 0, 1, 1, 0],
    4: [1, 0, 0, 1, 1, 0, 0],
    5: [0, 1, 0, 0, 1, 0, 0],
    6: [0, 1, 0, 0, 0, 0, 0],
    7: [0, 0, 0, 1, 1, 1, 1],
    8: [0, 0, 0, 0, 0, 0, 0],
    9: [0, 0, 0, 0, 1, 0, 0]
}


while True:
    red.on()
    yellow.off()
    green.off()
    for i in range(9, 0, -1):
        display(digits[i], 1)
    display(digits[0], 1)

    red.off()
    yellow.on()
    green.off()
    for i in range(1, 0, -1):
        display(digits[i], 1)
    display(digits[0], 1)

    red.off()
    yellow.off()
    green.on()
    for i in range(5, 0, -1):
        display(digits[i], 1)
    display(digits[0], 1)