Python Enum 使用的幾點注意事項

songofhawk發表於2022-02-22
Python

Enum 是個類

所以

基本的類操作都可以用

也就是我們可以新增自己的方法

class Mood(Enum):
    FUNKY = 1
    HAPPY = 3

    def describe(self):
        # self is the member here
        return self.name, self.value

    def __str__(self):
        return 'my custom str! {0}'.format(self.value)

    @classmethod
    def favorite_mood(cls):
        # cls here is the enumeration
        return cls.HAPPY

>>> Mood.favorite_mood()
<Mood.HAPPY: 3>
>>> Mood.HAPPY.describe()
('HAPPY', 3)
>>> str(Mood.FUNKY)
'my custom str! 1'
Enum的每個類成員,都會自動被轉換成當前類的一個例項
from enum import Enum
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3
>>> type(Color.RED)
<enum 'Color'>
>>> isinstance(Color.GREEN, Color)
True

這就意味著，我們不能用列舉成員直接當成它的value來用：

>>> Color.RED == 1
False
>>> Color.RED.value == 1
False

列舉成員還有個name屬性，跟它的變數名相同

>>> Color.RED.name == "RED"
True

列舉類有個大坑：父類有成員的時候，就不能定義子類

所以，對於上面的Color類，如果還想定義一個子類，就會出錯：

>>> class MoreColor(Color):
...     PINK = 17
...
Traceback (most recent call last):
...
TypeError: MoreColor: cannot extend enumeration 'Color'

但父類沒有列舉成員，僅僅定義了函式是可以的：

class Foo(Enum):
    def some_behavior(self):
        pass

class Bar(Foo):
    HAPPY = 1
    SAD = 2

這顯然大大限制了列舉類的擴充套件，一個不能訪問成員的函式，用處也就不大了。

相關文章