import time
import board
import digitalio
import pwmio

led1 = digitalio.DigitalInOut(board.GP2)
led1.direction = digitalio.Direction.OUTPUT

led2 = digitalio.DigitalInOut(board.GP3)
led2.direction = digitalio.Direction.OUTPUT

led3 = digitalio.DigitalInOut(board.GP4)
led3.direction = digitalio.Direction.OUTPUT

led4 = digitalio.DigitalInOut(board.GP5)
led4.direction = digitalio.Direction.OUTPUT

pb1 = digitalio.DigitalInOut(board.GP18)
pb1.direction = digitalio.Direction.INPUT
pb1.pull = digitalio.Pull.DOWN

pb2 = digitalio.DigitalInOut(board.GP19)
pb2.direction = digitalio.Direction.INPUT
pb2.pull = digitalio.Pull.UP

buzzer = pwmio.PWMOut(board.GP15, duty_cycle=0, frequency=1000)

while True:

    if pb1.value == True:

        for i in range(6):

            led1.value = led2.value = led3.value = led4.value = True
            time.sleep(0.5)

            led1.value = led2.value = led3.value = led4.value = False
            time.sleep(0.5)

        buzzer.duty_cycle = 30000
        time.sleep(1)
        buzzer.duty_cycle = 0

    elif pb2.value == False:

        for i in range(4):

            led1.value = True; time.sleep(0.2); led1.value = False
            led2.value = True; time.sleep(0.2); led2.value = False
            led3.value = True; time.sleep(0.2); led3.value = False
            led4.value = True; time.sleep(0.2); led4.value = False

            led4.value = True; time.sleep(0.2); led4.value = False
            led3.value = True; time.sleep(0.2); led3.value = False
            led2.value = True; time.sleep(0.2); led2.value = False
            led1.value = True; time.sleep(0.2); led1.value = False

    else:

        led1.value = led2.value = led3.value = led4.value = False