#Traffic light LEDs using an array

from machine import Pin
import time

# Define GPIO pins for the LEDs
RED = Pin(2, Pin.OUT)
AMBER = Pin(6, Pin.OUT)
GREEN = Pin(10, Pin.OUT)

# Put them in an array
lights = [RED, AMBER, GREEN]

####################################################
# visual Thonny breakout.

#use this declaration for Pico1 & Pico2

# Set Pin25 as output (Pin 25 is where the onboard LED is connected)
LED = Pin(25, Pin.OUT)

# Print a message in Thonny to remind user to break out of code now.
print('booting... Press ctrl-c to stop')

# loop 20 times
for i in range (20):

# turn on LED, wait half a second.
#for Pico1 & Pico2
    LED.on()
    time.sleep(0.5)

# turn off LED, wait half a second.
    LED.off()  
    time.sleep(0.5)
####################################################

# Helper function to turn all lights off
def all_off():
    for light in lights:
        light.off()

while True:
    # Red
    all_off()
    RED.on()
    time.sleep(3)

    # Red + Amber
    AMBER.on()
    time.sleep(1)

    # Green
    all_off()
    GREEN.on()
    time.sleep(5)

    # Amber
    all_off()
    AMBER.on()
    time.sleep(1)