Skip to content
Python

Enum

Define enum types using Enum.

By EZ4Code Team
enumenum

Code

from enum import Enum, auto

class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

class Status(Enum):
    PENDING = auto()
    RUNNING = auto()
    DONE = auto()

print(Color.RED.name, Color.RED.value)

# Get by value
c = Color(1)
print(c is Color.RED)

# Iterate
for s in Status:
    print(s)

# IntEnum supports comparison
class Priority(int, Enum):
    LOW = 1
    HIGH = 10

Explanation

Enum is used to define named constant sets, improving code readability.

More Python Snippets