enum.py

from enum import Enum

class Environment(Enum):
    DEVELOPMENT = "development"
    STAGING = "staging"
    PRODUCTION = "production"

def set_environment(env: Environment) -> str:
    return f"Environment set to {env.value}"

print(set_environment(Environment.DEVELOPMENT))  # Output: Environment set to development
print(set_environment(Environment.PRODUCTION))   # Output: Environment set to production