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

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

off999 2025-06-15 18:36 75 浏览 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
统一管理常量,降低维护成本

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

相关推荐

阿里云国际站ECS:阿里云ECS如何提高网站的访问速度?

TG:@yunlaoda360引言:速度即体验,速度即业务在当今数字化的世界中,网站的访问速度已成为决定用户体验、用户留存乃至业务转化率的关键因素。页面加载每延迟一秒,都可能导致用户流失和收入损失。对...

高流量大并发Linux TCP性能调优_linux 高并发网络编程

其实主要是手里面的跑openvpn服务器。因为并没有明文禁p2p(哎……想想那么多流量好像不跑点p2p也跑不完),所以造成有的时候如果有比较多人跑BT的话,会造成VPN速度急剧下降。本文所面对的情况为...

性能测试100集(12)性能指标资源使用率

在性能测试中,资源使用率是评估系统硬件效率的关键指标,主要包括以下四类:#性能测试##性能压测策略##软件测试#1.CPU使用率定义:CPU处理任务的时间占比,计算公式为1-空闲时间/总...

Linux 服务器常见的性能调优_linux高性能服务端编程

一、Linux服务器性能调优第一步——先搞懂“看什么”很多人刚接触Linux性能调优时,总想着直接改配置,其实第一步该是“看清楚问题”。就像医生看病要先听诊,调优前得先知道服务器“哪里...

Nginx性能优化实战:手把手教你提升10倍性能!

关注△mikechen△,十余年BAT架构经验倾囊相授!Nginx是大型架构而核心,下面我重点详解Nginx性能@mikechen文章来源:mikechen.cc1.worker_processe...

高并发场景下,Spring Cloud Gateway如何抗住百万QPS?

关注△mikechen△,十余年BAT架构经验倾囊相授!大家好,我是mikechen。高并发场景下网关作为流量的入口非常重要,下面我重点详解SpringCloudGateway如何抗住百万性能@m...

Kubernetes 高并发处理实战(可落地案例 + 源码)

目标场景:对外提供HTTPAPI的微服务在短时间内收到大量请求(例如每秒数千至数万RPS),要求系统可弹性扩容、限流降级、缓存减压、稳定运行并能自动恢复。总体思路(多层防护):边缘层:云LB...

高并发场景下,Nginx如何扛住千万级请求?

Nginx是大型架构的必备中间件,下面我重点详解Nginx如何实现高并发@mikechen文章来源:mikechen.cc事件驱动模型Nginx采用事件驱动模型,这是Nginx高并发性能的基石。传统...

Spring Boot+Vue全栈开发实战,中文版高清PDF资源

SpringBoot+Vue全栈开发实战,中文高清PDF资源,需要的可以私我:)SpringBoot致力于简化开发配置并为企业级开发提供一系列非业务性功能,而Vue则采用数据驱动视图的方式将程序...

Docker-基础操作_docker基础实战教程二

一、镜像1、从仓库获取镜像搜索镜像:dockersearchimage_name搜索结果过滤:是否官方:dockersearch--filter="is-offical=true...

你有空吗?跟我一起搭个服务器好不好?

来人人都是产品经理【起点学院】,BAT实战派产品总监手把手系统带你学产品、学运营。昨天闲的没事的时候,随手翻了翻写过的文章,发现一个很严重的问题。就是大多数时间我都在滔滔不绝的讲理论,却很少有涉及动手...

部署你自己的 SaaS_saas如何部署

部署你自己的VPNOpenVPN——功能齐全的开源VPN解决方案。(DigitalOcean教程)dockovpn.io—无状态OpenVPNdockerized服务器,不需要持久存储。...

Docker Compose_dockercompose安装

DockerCompose概述DockerCompose是一个用来定义和管理多容器应用的工具,通过一个docker-compose.yml文件,用YAML格式描述服务、网络、卷等内容,...

京东T7架构师推出的电子版SpringBoot,从构建小系统到架构大系统

前言:Java的各种开发框架发展了很多年,影响了一代又一代的程序员,现在无论是程序员,还是架构师,使用这些开发框架都面临着两方面的挑战。一方面是要快速开发出系统,这就要求使用的开发框架尽量简单,无论...

Kubernetes (k8s) 入门学习指南_k8s kubeproxy

Kubernetes(k8s)入门学习指南一、什么是Kubernetes?为什么需要它?Kubernetes(k8s)是一个开源的容器编排系统,用于自动化部署、扩展和管理容器化应用程序。它...

取消回复欢迎 发表评论: