百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

让Python枚举更强大!教你玩转Enum扩展

off999 2025-06-15 18:36 91 浏览 0 评论

为什么你需要关注Enum?在日常开发中,你是否经常遇到这样的代码?

if status == 1:
    print("开始处理")
elif status == 2:
    print("处理中")
elif status == 3:
    print("已完成")

这些“魔法数字”虽然能用,但可读性和维护性极差。 Python 3 的Enum(枚举类)就是为此而生——它能将数字变成有意义的符号,让代码瞬间“说人话”。

一、基础操作:5分钟上手Python枚举

定义第一个枚举类

from enum import Enum

class OrderStatus(Enum):
    PENDING = 1    # 待处理
    PROCESSING = 2 # 处理中
    COMPLETED = 3  # 已完成

用法示例

current_status = OrderStatus.PROCESSING
print(current_status.value)  # 输出:2
print(current_status.name)   # 输出:'PROCESSING'

优势

  • o 避免直接操作数字,代码一目了然
  • o 防止无效值(比如status=4会直接报错)

二、进阶技巧1:动态扩展枚举成员

场景

现有OrderStatus需要新增状态CANCELLED=4,但不想修改原代码(比如枚举类来自第三方库)。

解决方案:虽然无法直接继承,但可通过组合枚举成员实现:

class ExtendedOrderStatus(Enum):
    # 直接包含原成员
    PENDING = OrderStatus.PENDING.value    # 待处理
    PROCESSING = OrderStatus.PROCESSING.value # 处理中
    COMPLETED = OrderStatus.COMPLETED.value  # 已完成
    
    CANCELLED = 4     # 新增“已取消”状态
    REFUNDED = 5      # 新增“已退款”状态
# 使用新状态
print(ExtendedOrderStatus.CANCELLED.value)  # 输出:4

三、进阶技巧2:给枚举“添加超能力”

场景

订单状态需要关联更多信息,比如状态描述、对应的操作按钮。

解决方案:在子类中添加方法!

class ExtendedOrderStatus(OrderStatus):
    CANCELLED = 4
    REFUNDED = 5

    def get_action(self):
        # 为不同状态返回操作提示
        if self == ExtendedOrderStatus.CANCELLED:
            return "联系客服申诉"
        elif self == ExtendedOrderStatus.REFUNDED:
            return "查看退款详情"
        else:
            return super().get_action()  # 调用父类逻辑(如有)

调用示例

class ExtendedOrderStatus(Enum):
    # 直接包含原成员
    PENDING = OrderStatus.PENDING.value    # 待处理
    PROCESSING = OrderStatus.PROCESSING.value # 处理中
    COMPLETED = OrderStatus.COMPLETED.value  # 已完成
    
    CANCELLED = 4     # 新增“已取消”状态
    REFUNDED = 5      # 新增“已退款”状态
    
    def get_action(self):
        # 为不同状态返回操作提示
        if self == ExtendedOrderStatus.CANCELLED:
            return "联系客服申诉"
        elif self == ExtendedOrderStatus.REFUNDED:
            return "查看退款详情"
        else:
            return super().get_action()  # 调用父类逻辑(如有)
        
status = ExtendedOrderStatus.CANCELLED
print(status.get_action())  # 输出:"联系客服申诉" 

四、避坑指南:扩展枚举的3个注意事项

1、成员名冲突
子类成员名若与父类重复,会直接报错!确保名称唯一。

2、 值无需连续
枚举值可以是任意类型(字符串、元组等),不一定用数字:

    class Priority(Enum):
        HIGH = (1, "红色")
        MEDIUM = (2, "黄色")
        LOW = (3, "绿色")

3、慎用auto()
自动赋值虽方便,但扩展时可能导致值重复,建议显式定义。

五、实战案例:枚举在业务中的妙用

场景:电商订单状态流转

from enum import Enum
class OrderStatus(Enum):
    CREATED = {"code": 10, "desc": "订单已创建"}
    PAID = {"code": 20, "desc": "已支付"}
    SHIPPED = {"code": 30, "desc": "已发货"}

    def can_cancel(self):
        # 判断当前状态是否允许取消
        return self in (OrderStatus.CREATED, OrderStatus.PAID)
        
status = OrderStatus.PAID
print(status.can_cancel())  # True

status = OrderStatus.SHIPPED
print(status.can_cancel())  # False        

通过扩展Enum,你可以:
灵活新增状态,无需修改原有代码
封装复杂逻辑,告别满屏的if-else
统一管理常量,降低维护成本

动手试试:在你的下一个项目中,用枚举替换那些“魔法数字”,感受代码质的提升吧!

相关推荐

不小心抹掉了mac内置的磁盘(不小心抹掉了mac内置的磁盘怎么恢复)

因为你进入的是恢复磁盘,恢复磁盘是基于mac的。自己不能抹掉自己吧。所以抹不掉。如果你要彻底抹掉的话,开机按option+command+r,wifi下载网络恢复磁盘,就可以抹掉了。您可以按住COMM...

开机界面切换用户不见了win7
开机界面切换用户不见了win7

开始--运行(或者WIN+R),键入“msconfig”,单击“确定”或回车,运行“系统配置实用程序”。在“启动”里把“Ctfmon.exe”选中,单击“确定”开始--运行,输入“ctfmon.exe”回车,再输入“msconfig”启动,...

2025-12-13 00:03 off999

win7旗舰版进入安全模式(win7旗舰版进入安全模式无法启动)

win7安全模式怎么进?  一、开机时按F8键  当你重启或开机时,在进入Windows系统启动画面之前按下F8键,会出现系统多操作启动菜单,有三个版本的安全模式可以选择,回车就直接进入安全模式。  ...

ps手机版下载中文免费版2025

ps2023对电脑要求高。ps2022版本对电脑配置要求是:1、内存要求在一32G和64G之间。2、硬盘要求选择7200转机械硬盘,4k以上分辨,就选择SSD固态硬盘。3、CPU要求睿频在4.1GHZ...

苹果电脑如何下载windows系统

下载后安装BootCamp,之后在[应用程序]的[实用程序]中找到BootCamp助理,下面就可以开始安装Windows前的准备工作了。  第一步是要用BootCamp助理刻录一张驱动程序的CD...

efi系统分区什么意思(efi系统分区的作用)

ud空间是指存放PE启动文件的分区;efi空间是指存放UEFI启动文件的分区。系统U盘(PE启动盘)的制作方法,一共分为三种。三分区法是其中的一种,三分区法就是一个U盘被分成了三个区,两边的分区是隐藏...

无需u盘重装系统(无需u盘安装系统)

可以通过以下步骤在不使用U盘的情况下重装Win7系统:首先需要备份您的电脑中的重要数据,以免在系统重装时丢失。进入系统的“控制面板”,找到“系统与安全”选项并单击进入。在“系统与安全”页面中,找到“备...

改无线网密码的网址(改无线网密码的网站)

  移动宽带无线网改密码,可以在移动网上营业厅办理,但更简单的办法是通过短信办理(详见如下第2条):  移动无线宽带短信操作方式:  1)开通业务:编写短信“KTWLAN”发送至10086。  注:短...

英伟达驱动怎么更新(英伟达驱动在哪儿更新)
  • 英伟达驱动怎么更新(英伟达驱动在哪儿更新)
  • 英伟达驱动怎么更新(英伟达驱动在哪儿更新)
  • 英伟达驱动怎么更新(英伟达驱动在哪儿更新)
  • 英伟达驱动怎么更新(英伟达驱动在哪儿更新)
c盘满了却不敢删(c盘满了清不掉)

1、运用磁盘清理软件清理C盘,大约可为C盘释放50M-800M空间。2、关闭休眠功能,在开始菜单的运行里输入powercfg-hoff指令,关闭休眠,此文件实际大小和物理内存是一样的,大约可以...

联想笔记本电脑怎么打开摄像头
联想笔记本电脑怎么打开摄像头

联想笔记本怎么打开摄像头?1、点击电脑开始菜单栏,在弹出的窗口中点击【设置】。2、在”隐私和安全性“面板中点击【相机】选项。3、将【摄像头访问】右侧的开关打开。材料/工具:以联想Y460笔记本为例。操作步骤如下:1、在电脑的桌面,打开我...

2025-12-12 19:51 off999

win732纯净版系统gho(win732纯净版下载)

一,win7x32gho指的是32位的Windows7操作系统,gho指的是ghost方式封装的镜像文件,一般这种都是盗版系统。二,如果自己需要安装64位系统就不要选这个,内存低于3GB可以安装win...

pscs6激活账户(pscs6激活账户和序列号免费)

pscs6破解补丁是AdobePhotoShopCs6版本的破解激活补丁,使用后不会弹出购买窗口,实现完美破解,无任何限制破解版软件扩展功能好像不好用!要用需要的扩展功能可以在网上下载需要的插件...

键盘皮肤下载(键盘皮肤下载软件免费有音效)

1、vivo自带的输入法不能更换键盘皮肤;2、建议可以下载第三方输入法更换皮肤,比如:搜狗输入法等;3、下载第三方输入法之后需要在设置-更多设置-输入法中设置为第三方的输入法使用;4、然后在输入法界面...

win11如何格式化硬盘(win11硬盘分区怎么改成gpt)

win11系统格式化电脑和硬盘的方法如下1、打开本地磁盘管理页面,右键点击需要格式化的磁盘,选择【格式化】。2、在弹出的窗口中,勾选【快速格式化】,点击【开始】。3、最后等待格式化完成即可。1.首先我...

取消回复欢迎 发表评论: