import time
from machine import Pin
from time import sleep
sleep(0.1) # Wait for USB to become ready


seg_a = Pin(27, Pin.OUT)
seg_b = Pin(28, Pin.OUT)
seg_c = Pin(17, Pin.OUT)
seg_d = Pin(18, Pin.OUT)
seg_e = Pin(19, Pin.OUT)
seg_f = Pin(26, Pin.OUT)
seg_g = Pin(22, Pin.OUT)
segments = [seg_a, seg_b, seg_c, seg_d, seg_e, seg_f, seg_g]


button_increase = Pin(3, Pin.IN, Pin.PULL_DOWN)
button_decrease = Pin(12, Pin.IN, Pin.PULL_DOWN)


def set_segments(b):
    global segments
    for x,y in zip(segments,b):
        x.value(y)


def set_number(val):
    numbers = [
        [0,0,0,0,0,0,1],
        [1,0,0,1,1,1,1],
        [0,0,1,0,0,1,0],
        [0,0,0,0,1,1,0],
        [1,0,0,1,1,0,0],
        [0,1,0,0,1,0,0],
        [0,1,0,0,0,0,0],
        [0,0,0,1,1,1,1],
        [0,0,0,0,0,0,0],
        [0,0,0,0,1,0,0]
    ]
    set_segments(numbers[val])


current_number = 0
while True:
    if button_increase.value() == 1:
        while button_increase.value() == 1:
            pass

        if current_number < 9:
            current_number += 1
        else:
            current_number = 0

        set_number(current_number)

    if button_decrease.value() == 1:
        while button_decrease.value() == 1:
            pass

        if current_number > 0:
            current_number -= 1
        else:
            current_number = 9

        set_number(current_number)