from machine import Pin
from time import sleep

r = Pin(2, Pin.OUT)
g = Pin(3, Pin.OUT)
b = Pin(4, Pin.OUT)


while True:
    switch = Pin(20, Pin.IN)
    #switch_state = switch.value
    print(switch.value())

    if switch.value() == 1:
        for i in range(3):
            r.value(1)
            g.value(0)
            b.value(0)
            sleep(0.5)
            r.value(0)
            g.value(1)
            b.value(0)
            sleep(0.5)
            r.value(0)
            g.value(0)
            b.value(1)
            sleep(0.5)
    else:
        for i in range(3):
            r.value(1)
            g.value(0)
            b.value(1)
            sleep(0.5)
            r.value(0)
            g.value(1)
            b.value(1)
            sleep(0.5)
            r.value(1)
            g.value(1)
            b.value(0)
            sleep(0.5)