python_strenum_string_comparable_enum_quickstart.py

1from enum import StrEnum
2
3class Direction(StrEnum):
4    NORTH = "north"
5    SOUTH = "south"
6    EAST = "east"
7    WEST = "west"
8
9# Members are strings
10print(Direction.NORTH == "north")  # True
11
12# They work with string methods
13print(Direction.SOUTH.upper())     # "SOUTH"
14
15# Useful for APIs that expect string values
16def move(direction: str):
17    print(f"Moving {direction}")
18
19move(Direction.EAST)               # Moving east