from machine import Pin
import time

red= Pin(5, Pin.OUT)
yellow = Pin(4, Pin.OUT)
green = Pin(2, Pin.OUT)
button = Pin(18, Pin.IN, Pin.PULL_UP)

mode = 0 # 0 - night (yellow blink)/ 1 - day
debounce_time = 0.5
last_state_button = button.value()


while True:
    state_button = button.value()
    if state_button != last_state_button:
        time.sleep(debounce_time)
        if button.value() == state_button:
            last_state_button = state_button
    

    # mode = 0

    if state_button == 0:
        mode = 0
    else:
        mode = 1
            
    if mode == 0:
        print("Night mode")
        red.on()
        yellow.off()
        green.off()
        time.sleep(3)
        red.on()
        yellow.on()
        time.sleep(3)
        red.off()
        yellow.off()
        green.on()
        time.sleep(3)
        for i in range(5):
            green.off()
            time.sleep(1)
            green.on()
            time.sleep(1)
            i += 1
    else:
        print("Day mode")
        red.off()
        green.off()
        yellow.on()
        for i in range(10):
            yellow.off()
            time.sleep(1)
            yellow.on()
            time.sleep(1)