from machine import Pin
from utime import sleep

pins = [
    Pin(9, Pin.OUT),  # A
    Pin(10, Pin.OUT), # B
    Pin(0, Pin.OUT),  # C
    Pin(1, Pin.OUT),  # D
    Pin(2, Pin.OUT),  # E
    Pin(8, Pin.OUT),  # F
    Pin(7, Pin.OUT),  # G
]


led_pins = [11,12,13, 15, 16, 17]
leds = []

for l in led_pins:
    leds.append(Pin(l, Pin.OUT))

red = [0, 0, 1]
yellow = [0, 1, 0]
green = [1, 0, 0]

timer = [7, 2, 2, 7, 2, 2]
a = [green, yellow, red, red, red, red]
b = [red,red, red, green, yellow, red ]

digits = [
    [0, 0, 0, 0, 0, 0, 1, 1], # 0
    [1, 0, 0, 1, 1, 1, 1, 1], # 1
    [0, 0, 1, 0, 0, 1, 0, 1], # 2 
    [0, 0, 0, 0, 1, 1, 0, 1], # 3
    [1, 0, 0, 1, 1, 0, 0, 1], # 4
    [0, 1, 0, 0, 1, 0, 0, 1], # 5
    [0, 1, 0, 0, 0, 0, 0, 1], # 6
    [0, 0, 0, 1, 1, 1, 1, 1], # 7
    [0, 0, 0, 0, 0, 0, 0, 1], # 8
    [0, 0, 0, 1, 1, 0, 0, 1], # 9
]

def reset():
    for pin in pins:
        pin.value(1)

reset()

while True:
    for t in range(len(timer)):
        leds[0].value(a[t][0])
        leds[1].value(a[t][1])
        leds[2].value(a[t][2])
        leds[3].value(b[t][0])
        leds[4].value(b[t][1])
        leds[5].value(b[t][2])

        for i in range(timer[t], -1, -1):
            for j in range(len(pins)):
                pins[j].value(digits[i][j])
            sleep(1)