Python StrEnum基本概念和使用场景分析

from enum import StrEnum
class Color(StrEnum):
    RED = "red"
    GREEN = "green"
    BLUE = "blue"
# 使用StrEnum
print(Color.RED)  # 输出: red
print(Color.GREEN == "green")  # 输出: True
print(list(Color))  # 输出: [<Color.RED: 'red'>, <Color.GREEN: 'green'>, <Color.BLUE: 'blue'>]

from enum import StrEnum
class LogLevel(StrEnum):
    DEBUG = "debug"
    INFO = "info"
    WARNING = "warning"
    ERROR = "error"
def configure_logging(level: LogLevel):
    print(f"Configuring logging with level: {level}")
# 使用
configure_logging(LogLevel.DEBUG)
configure_logging("info")  # 这也是有效的,因为StrEnum可以与字符串直接比较

from enum import StrEnum
class APIStatus(StrEnum):
    SUCCESS = "success"
    FAILURE = "failure"
    PENDING = "pending"
def process_api_response(status: str):
    if status == APIStatus.SUCCESS:
        print("Request successful")
    elif status == APIStatus.FAILURE:
        print("Request failed")
    elif status == APIStatus.PENDING:
        print("Request is pending")
    else:
        print("Unknown status")
# 使用
process_api_response("success")
process_api_response(APIStatus.PENDING)

from enum import StrEnum
class Fruit(StrEnum):
    APPLE = "apple"
    BANANA = "banana"
    ORANGE = "orange"
def process_fruit(fruit: Fruit):
    print(f"Processing {fruit}")
# 有效使用
process_fruit(Fruit.APPLE)
process_fruit("banana")
# 无效使用会引发ValueError
try:
    process_fruit("grape")
except ValueError as e:
    print(f"Error: {e}")