from machine import Pin
from time import sleep

switch = Pin(20, Pin.IN)
red = Pin(2, Pin.OUT)
green = Pin(3, Pin.OUT)
blue = Pin(4, Pin.OUT)

while True:
  sw_state = switch.value()

  print("Switch  state is: ", sw_state)
    
  if sw_state == 1:
    for i in range(1,5,1):
       led = Pin(i, Pin.OUT)
       led.value(1)
       sleep(0.5)
       led.value(0)
       sleep(0.5)

  else:
    red.value(1)
    blue.value(1)
    sleep(0.5)
    red.value(0)
    blue.value(0)
    sleep(0.5)
    green.value(1)
    blue.value(1)
    sleep(0.5)
    green.value(0)
    blue.value(0)
    sleep(0.5)
    red.value(1)
    green.value(1)
    sleep(0.5)
    green.value(0)
    red.value(0)
    sleep(0.5)