from enum import Enum

class Semaphore(Enum):

RED = 1

YELLOW = 2

GREEN = 3

# def handle_semaphore(light):

# if light is Semaphore.RED:

# print("You must stop!")

# elif light is Semaphore.YELLOW:

# print("Light will change to red, be careful!")

# elif light is Semaphore.GREEN:

# print("You can continue!")

def handle_semaphore(light):

match light:

case Semaphore.RED:

print("You must stop!")

case Semaphore.YELLOW:

print("Light will change to red, be careful!")

case Semaphore.GREEN:

print("You can continue!")

handle_semaphore(Semaphore.YELLOW)