import time
from machine import Pin as pin
time.sleep(0.1) # Wait for USB to become ready

#       North-South Traffic Lights
#First Traffic Light
traff_1_R = pin(0, pin.OUT)
traff_1_Y = pin(1, pin.OUT)
traff_1_G = pin(2, pin.OUT)

#Second Traffic Light
traff_2_R = pin(3, pin.OUT)
traff_2_Y = pin(4, pin.OUT)
traff_2_G = pin(5, pin.OUT)

#First Pedestrian Traffic Light
ped_1_R = pin(6, pin.OUT)
ped_1_G = pin(7, pin.OUT)

#Second Pedestrian Traffic Light
ped_2_R = pin(8, pin.OUT)
ped_2_G = pin(9, pin.OUT)

#       East-West Traffic Lights
#Third Traffic Light
traff_3_R = pin(10, pin.OUT)
traff_3_Y = pin(11, pin.OUT)
traff_3_G = pin(12, pin.OUT)

#Fourth Traffic Light
traff_4_R = pin(13, pin.OUT)
traff_4_Y = pin(14, pin.OUT)
traff_4_G = pin(15, pin.OUT)

#Third Pedestrian Traffic Light
ped_3_R = pin(16, pin.OUT)
ped_3_G = pin(17, pin.OUT)

#Fourth Pedestrian Traffic Light
ped_4_R = pin(18, pin.OUT)
ped_4_G = pin(19, pin.OUT)

delay = 1 #Sets a delay of 1 sec

while True:
    
    #       North-South Pattern
    #--------------------------------------------
    #North-South
    traff_1_R.value(0)
    traff_1_Y.value(0)
    traff_1_G.value(1)

    traff_2_R.value(0)
    traff_2_Y.value(0)
    traff_2_G.value(1)
    
    ped_1_R.value(0)
    ped_1_G.value(1)

    ped_2_R.value(0)
    ped_2_G.value(1)
    
    #East-West
    traff_3_R.value(1)
    traff_3_Y.value(0)
    traff_3_G.value(0)
    
    traff_4_R.value(1)
    traff_4_Y.value(0)
    traff_4_G.value(0)
    
    ped_3_R.value(1)
    ped_3_G.value(0)

    ped_4_R.value(1)
    ped_4_G.value(0)

    time.sleep(delay)
#--------------------------------------------
    #North-South
    traff_1_R.value(0)
    traff_1_Y.value(1)
    traff_1_G.value(0)

    traff_2_R.value(0)
    traff_2_Y.value(1)
    traff_2_G.value(0)
    
    ped_1_R.value(0)
    ped_1_G.value(1)

    ped_2_R.value(0)
    ped_2_G.value(1)
    
    #East-West
    traff_3_R.value(1)
    traff_3_Y.value(0)
    traff_3_G.value(0)
    
    traff_4_R.value(1)
    traff_4_Y.value(0)
    traff_4_G.value(0)
    
    ped_3_R.value(1)
    ped_3_G.value(0)

    ped_4_R.value(1)
    ped_4_G.value(0)

    time.sleep(delay)
#--------------------------------------------
    #North-South
    traff_1_R.value(1)
    traff_1_Y.value(0)
    traff_1_G.value(0)

    traff_2_R.value(1)
    traff_2_Y.value(0)
    traff_2_G.value(0)
    
    ped_1_R.value(1)
    ped_1_G.value(0)

    ped_2_R.value(1)
    ped_2_G.value(0)
    
    #East-West
    traff_3_R.value(0)
    traff_3_Y.value(0)
    traff_3_G.value(1)
    
    traff_4_R.value(0)
    traff_4_Y.value(0)
    traff_4_G.value(1)
    
    ped_3_R.value(0)
    ped_3_G.value(1)

    ped_4_R.value(0)
    ped_4_G.value(1)

    time.sleep(delay)
#--------------------------------------------
    #North-South
    traff_1_R.value(1)
    traff_1_Y.value(0)
    traff_1_G.value(0)

    traff_2_R.value(1)
    traff_2_Y.value(0)
    traff_2_G.value(0)
    
    ped_1_R.value(1)
    ped_1_G.value(0)

    ped_2_R.value(1)
    ped_2_G.value(0)
    
    #East-West
    traff_3_R.value(0)
    traff_3_Y.value(1)
    traff_3_G.value(0)
    
    traff_4_R.value(0)
    traff_4_Y.value(1)
    traff_4_G.value(0)
    
    ped_3_R.value(0)
    ped_3_G.value(1)

    ped_4_R.value(0)
    ped_4_G.value(1)

    time.sleep(delay)