from machine import Pin, Timer
from time import sleep

led_green = Pin(22, Pin.OUT)
led_blue = Pin(21, Pin.OUT)
led_red = Pin(23,Pin.OUT)

button_pin34 = Pin(34, Pin.IN)

def handler_2(tim2):
    global switch
    
    if switch == 0:
        tim2.deinit()
    elif switch == 1:
        led_green.value(not led_green.value())
    elif switch == 2:
        led_blue.value(not led_blue.value())
    elif switch == 3:
        led_red.value(not led_red.value())
    else:
        tim2.deinit()
        switch = 0

tim2=Timer(2)

switch=0
while True:
    if button_pin34.value() == 1:
        sleep(0.1)

        if button_pin34.value() == 1:
            sleep(0.1)
            switch = switch+1

        if switch == 1:
            led_blue.value(0)
            led_red.value(0)
            tim2.init(period=1500, mode=Timer.PERIODIC, callback=handler_2)
            sleep(0.1)

        elif switch == 2:
            led_green.value(0)
            led_blue.value(0)
            tim2.init(period=1000, mode=Timer.PERIODIC, callback=handler_2)
            sleep(0.1)

        elif switch == 3:
            led_red.value(0)
            led_blue.value(0)
            tim2.init(period=500, mode=Timer.PERIODIC, callback=handler_2)
            sleep(0.1)        

        else:
            switch=0
            tim2.deinit()
            led_red.value(0)
            led_blue.value(0)
            led_green.value(0)

    switch_status = "The value of switch is {} and button is {}"
    print(switch_status.format(switch, button_pin34.value()), end="\r")