Simple script to control a traffic light using a micro:bit and stop:bit

stopbit.py

from microbit import *
# Kitronik STOP:bit blocks

# turn each LED on or off individually
def stopBitLight(colour, illuminate):    
    if colour == "Red": 
        if illuminate == "On":
            pin0.write_digital(1)
        elif illuminate == "Off":
            pin0.write_digital(0)
    elif colour == "Yellow":
        if illuminate == "On":
            pin1.write_digital(1)
        elif illuminate == "Off":
            pin1.write_digital(0)  
    elif colour == "Green":
        if illuminate == "On":
            pin2.write_digital(1)
        elif illuminate == "Off":
            pin2.write_digital(0)     

# turn each LED on or off individually
def stopBitState(state): 
    if state == "stop":
        pin0.write_digital(1)
        pin1.write_digital(0)
        pin2.write_digital(0)
    elif state == "getReady":
        pin0.write_digital(1)
        pin1.write_digital(1)
        pin2.write_digital(0) 
    elif state == "go":
        pin0.write_digital(0)
        pin1.write_digital(0)
        pin2.write_digital(1)    
    elif state == "readyToStop":
        pin0.write_digital(0)
        pin1.write_digital(1)
        pin2.write_digital(0)

while True:
    stopBitLight("Red", "On")
    stopBitLight("Yellow", "Off")
    stopBitLight("Green", "Off")
    
    sleep(5000)
    
    stopBitLight("Red", "Off")
    stopBitLight("Yellow", "Off")
    stopBitLight("Green", "On") 
    
    # Tell vehicles to drive here
    
    sleep(5000)
    
    stopBitLight("Red", "Off")
    stopBitLight("Yellow", "On")
    stopBitLight("Green", "Off")
    
    sleep(1000)
    
    # Tell vehicles to stop here