import RPi.GPIO

# you must make relay PCB to control signal lamp
# DO NOT CONNECT SIGNAL LAMP TO GPIO DIRECTLY

GPIO.setup(17, GPIO.OUT)  # red
GPIO.setup(27, GPIO.OUT)  # yellow
GPIO.setup(22, GPIO.OUT)  # green

def signal_lamp(lamp):
    if lamp == "red":
        GPIO.output(17, 1)
        GPIO.output(27, 0)
        GPIO.output(22, 0)
    elif lamp == "yellow":
        GPIO.output(17, 0)
        GPIO.output(27, 1)
        GPIO.output(22, 0)
    elif lamp == "green":
        GPIO.output(17, 0)
        GPIO.output(27, 0)
        GPIO.output(22, 1)
    elif lamp == "test":
        GPIO.output(17, 1)
        GPIO.output(27, 1)
        GPIO.output(22, 1)
    elif lamp == "none":
        GPIO.output(17, 0)
        GPIO.output(27, 0)
        GPIO.output(22, 0)
    else:
        return 1