from machine import Pin 
from time import sleep

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

while True:
    sw_state=sw.value()
    if sw_state==1:
        for i in range(3):
            r.value(1)
            sleep(.5)
            r.value(0)

            g.value(1)
            sleep(.5)
            g.value(0)
            
            b.value(1)
            sleep(.5)
            b.value(0)
    if sw_state==0:
        for i in range(3):
            r.value(1)
            b.value(1)
            sleep(.5)
            r.value(0)
            b.value(0)
            
            g.value(1)
            b.value(1)
            sleep(.5)
            g.value(0)
            b.value(0)
            
            r.value(1)
            g.value(1)
            sleep(.5)
            r.value(0)
            g.value(0)