python入门——面向对象编程基础到进阶之枚举类Enum

文章目录

面向对象编程
@property
使用枚举类

面向对象编程

@property

为了避免实例对象属性的值被随意修改，我们可以通过定义一个方法了限制值的范围。如：

推荐一下我建的python学习交流qun：850973621，群里有免费的视频教程，开发工具、
电子书籍、项目源码分享。一起交流学习，一起进步！

class Student(object):

    def get_score(self):
         return self._score

    def set_score(self, value):
        if not isinstance(value, int):                             # 如果值不是整数，显示错误
            raise ValueError('score must be an integer!')
        if value < 0 or value > 100:                               # 如果值超出范围，显示错误
            raise ValueError('score must between 0 ~ 100!')
        self._score = value

这样就可以避免输入无效的值

>>> s = Student()
>>> s.set_score(50)                                                # 是整数且在范围之内
>>> s.get_score()
50
>>> s.set_score(1000)                                              # 超出规定范围，无法输入
ValueError: score must between 0 ~ 100!

这样每次输入值的时候都要调用方法显得十分麻烦。而装饰器@property可以解决这一问题。@property的用途是可以把方法变成属性来调用，用法如下：

class Student(object):

    @property                                              # 在@property接着的方法可以变成属性调用
    def score(self):                     
        return self._score

    @score.setter                                          # 装饰器@score.setter的用途是可以给score赋值
    def score(self, value):
        if not isinstance(value, int):
            raise ValueError('score must be an integer!')
        if value < 0 or value > 100:
            raise ValueError('score must between 0 ~ 100!')
        self._score = value

这样就可以直接把score()方法当属性来调用

>>> s = Student()
>>> s.score = 50                                           # 实际上调用了@score.setter后的方法
>>> s.score                                                # 实际上调用了@property后的方法
50

注意到，实例s的分数属性名实际上是_score，而score是方法名。

装饰器@property本身又创建了一个装饰器@xxx.setter。如果该方法只有@property，那么就转化成只读属性；如果该方法添加@property之后，还有@xxx.setter，则转化为读写属性。

使用枚举类

一般情况下，我们会通过字母大写来表示常量，但实际上这样定义的常量还是变量，可以随时被修改。

如果让每一个常量都是class类型的唯一一个实例，那么就可以让定义的常量不被修改。Python提供了Enum类实现了这一功能。

如我们需要定义有周一到周日的枚举类型Month，则定义Month类继承Enum类，而后定义Month类的成员

from enum import Enum, unique

@unique                                        # @unique装饰器的作用是检查是否有重复值
class Month(Enum):                             # 定义一个Month类继承Enum类
    Jan = 1                                    # 定义成员name为Jan，value为1
    Feb = 2
    Mar = 3
    Apr = 4
    May = 5
    Jun = 6
    JUl = 7
    Aug = 8
    Sep = 9
    Oct = 10
    Nov = 11
    Dec = 12

其中@unique装饰器可以帮助我们检查保证没有重复值。

这样我们可以通过for…in语句来访问枚举类Month

>>>for name, member in Month.__members__.items():      # name取得是成员名称，member取的是枚举常量
...     print(name, '=>', member)
Jan ==> Month.Jan 
Feb ==> Month.Feb 
Mar ==> Month.Mar 
Apr ==> Month.Apr 
May ==> Month.May 
Jun ==> Month.Jun 
JUl ==> Month.JUl 
Aug ==> Month.Aug 
Sep ==> Month.Sep 
Oct ==> Month.Oct 
Nov ==> Month.Nov 
Dec ==> Month.Dec 

当然我们可以单一的访问枚举常量，根据其name或value都可以。

>>>print(Month.Jan)
Month.Jan
>>>print(Month['Jan'])
Month.Jan
>>>print(Month(1))
Month.Jan

此外，可以直接用一行语句来定义枚举类，像这样：

from enum import Enum

Month = Enum('Month', ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'))

使用这样的定义方法，默认第一个成员的value为1。

---

以上就是本节的全部内容，感谢你的阅读。