Skip to content
Python

Enum

使用 Enum 定义枚举类型。

#enum#enum

Code

python
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