python中枚举类的使用样例

一、枚举类的定义

要定义一个枚举类,需要导入 enum 模块并继承 enum.Enum 类。

  • 枚举类用于表示一组命名的常量,提升代码的可读性和可维护性。
  • 可以通过名称和值访问枚举成员,并且可以迭代枚举类的所有成员。
  • 枚举类可以与自动值、混合类型和方法结合使用,以增强其实用性。

二、枚举类的样例使用

from enum import Enum, auto

class OrderStatus(Enum):
    PENDING = auto()
    PROCESSING = auto()
    SHIPPED = auto()
    DELIVERED = auto()

def update_order_status(order_id: int, status: OrderStatus):
    # 模拟更新订单状态的函数
    print(f"Order {order_id} status updated to {status.name}")

def check_order_status(order_id: int) -> OrderStatus:
    # 模拟检查订单状态的函数,假设所有订单都是 SHIPPED 状态
    return OrderStatus.SHIPPED

# 示例使用
order_id = 123
status = check_order_status(order_id)
print(f"Order {order_id} current status: {status.name}")

update_order_status(order_id, OrderStatus.DELIVERED)

三、枚举类的其他高级用法

可以将枚举与其他类型结合使用,例如将枚举与字符串、整数混合

你可能感兴趣的:(Python,python)