from machine import Pin
from time import sleep


led1 = Pin(0, Pin.OUT)
led2 = Pin(1, Pin.OUT)
led3 = Pin(2, Pin.OUT)
led4 = Pin(3, Pin.OUT)
led5 = Pin(4, Pin.OUT)

push_button = Pin(12, Pin.IN, Pin.PULL_UP)


delay = 0.13


# control 5 leds by 1 for On and 0 for Off
# example: number 11000 - 2 lefts led turning on and 3 right - off
def control_leds(leds_state):
    led1.value(leds_state % 10)
    leds_state = int(leds_state / 10)
    
    led2.value(leds_state % 10)
    leds_state = int(leds_state / 10)

    led3.value(leds_state % 10)
    leds_state = int(leds_state / 10)

    led4.value(leds_state % 10)
    leds_state = int(leds_state / 10)

    led5.value(leds_state % 10)


# init state
leds_state = 00001
control_leds(leds_state)
sleep(delay)

# Push and hold button for keeping leds running
while True:
    for i in range(0, 4):
        while push_button.value():
            sleep(0.01)

        leds_state *= 10
        control_leds(leds_state)
        sleep(delay)

    for i in range(0, 4):
        while push_button.value():
            sleep(0.01)

        leds_state /= 10
        control_leds(leds_state)
        sleep(delay)