enum —- 对枚举的支持

3.4 新版功能.

源代码: Lib/enum.py


枚举是与多个唯一常量值绑定的一组符号名(即成员)。枚举中的成员可以进行身份比较,并且枚举自身也可迭代。

注解

枚举成员名称的大小写

枚举表示的是常量,因此,建议枚举成员名称使用大写字母,本篇的示例将用此种风格。

模块内容

本模块定义了四个枚举类,用来定义名称与值的唯一组合: EnumIntEnumFlagIntFlag。此外,还定义了一个装饰器,unique(), 和一个辅助类,auto

class enum.Enum

创建枚举常量的基类。 如需了解另一种构建语法,请参阅 Functional API 小节。

class enum.IntEnum

创建 int 子类枚举常量的基类。

class enum.IntFlag

创建可与位运算符搭配使用,又不失去 IntFlag 成员资格的枚举常量的基类。IntFlag 成员也是 int 的子类。

class enum.Flag

创建可与位运算符搭配使用,又不会失去 Flag 成员资格的枚举常量的基类。

enum.unique()

确保一个名称只绑定一个值的 Enum 类装饰器。

class enum.auto

以合适的值代替 Enum 成员的实例。 初始值默认从 1 开始。

3.6 新版功能: Flag, IntFlag, auto

创建 Enum

枚举是由 class 句法创建的,这种方式易读、易写。 另一种创建方法请参阅 Functional APIEnum 以如下定义枚举:

  1. >>> from enum import Enum
  2. >>> class Color(Enum):
  3. ... RED = 1
  4. ... GREEN = 2
  5. ... BLUE = 3
  6. ...

注解

Enum 成员值

成员值可以是 intstr 等。若无需设定确切值,auto 实例可以自动为成员分配合适 的值。将 auto 与其他值混用时必须要慎重。

注解

命名法

  • Color枚举 (或称为 enum )。

  • Color.REDColor.GREEN 等属性是 枚举成员 (或 enum 成员),也是常量。

  • 枚举成员具有 名称 (例如 Color.RED 的名称为 REDColor.BLUE 的值为 3 等等)

注解

虽然 Enum 由 class 语法创建,但 Enum 并不是常规的 Python 类。详见 How are Enums different?

枚举成员的字符串表现形式更容易理解:

  1. >>> print(Color.RED)
  2. Color.RED

同时,它的 repr 包含更多信息:

  1. >>> print(repr(Color.RED))
  2. <Color.RED: 1>

枚举成员的 类型 就是它所属的枚举:

  1. >>> type(Color.RED)
  2. <enum 'Color'>
  3. >>> isinstance(Color.GREEN, Color)
  4. True
  5. >>>

Enum 成员还包含 name 属性:

  1. >>> print(Color.RED.name)
  2. RED

枚举按定义的顺序进行迭代:

  1. >>> class Shake(Enum):
  2. ... VANILLA = 7
  3. ... CHOCOLATE = 4
  4. ... COOKIES = 9
  5. ... MINT = 3
  6. ...
  7. >>> for shake in Shake:
  8. ... print(shake)
  9. ...
  10. Shake.VANILLA
  11. Shake.CHOCOLATE
  12. Shake.COOKIES
  13. Shake.MINT

枚举成员可哈希,可用于字典和集合:

  1. >>> apples = {}
  2. >>> apples[Color.RED] = 'red delicious'
  3. >>> apples[Color.GREEN] = 'granny smith'
  4. >>> apples == {Color.RED: 'red delicious', Color.GREEN: 'granny smith'}
  5. True

枚举成员及其属性的编程访问

有时,要在程序中访问枚举成员(如,开发时不知道颜色的确切值,Color.RED 不适用的情况)。Enum 支持如下访问方式:

  1. >>> Color(1)
  2. <Color.RED: 1>
  3. >>> Color(3)
  4. <Color.BLUE: 3>

name 访问枚举成员时,可使用项目名称:

  1. >>> Color['RED']
  2. <Color.RED: 1>
  3. >>> Color['GREEN']
  4. <Color.GREEN: 2>

可访问枚举成员的 namevalue

  1. >>> member = Color.RED
  2. >>> member.name
  3. 'RED'
  4. >>> member.value
  5. 1

重复的枚举成员和值

两个枚举成员的名称不能相同:

  1. >>> class Shape(Enum):
  2. ... SQUARE = 2
  3. ... SQUARE = 3
  4. ...
  5. Traceback (most recent call last):
  6. ...
  7. TypeError: Attempted to reuse key: 'SQUARE'

但是,两个枚举成员可以有相同的值。假设,成员 A 和 B 的值相同(先定义的是 A),则 B 是 A 的别名。按值查找 A 和 B 的值返回的是 A。按名称查找 B,返回的也是 A:

  1. >>> class Shape(Enum):
  2. ... SQUARE = 2
  3. ... DIAMOND = 1
  4. ... CIRCLE = 3
  5. ... ALIAS_FOR_SQUARE = 2
  6. ...
  7. >>> Shape.SQUARE
  8. <Shape.SQUARE: 2>
  9. >>> Shape.ALIAS_FOR_SQUARE
  10. <Shape.SQUARE: 2>
  11. >>> Shape(2)
  12. <Shape.SQUARE: 2>

注解

不支持创建与已定义属性(其他成员、方法等)同名的成员,也不支持创建与现有成员同名的属性。

确保唯一枚举值

默认情况下,枚举允许多个名称作为一个值的别名。如需禁用此行为,下述装饰器可以确保枚举中的值仅能只用一次:

@``enum.unique

专用于枚举的 class 装饰器。 它会搜索一个枚举的 __members__ 并收集所找到的任何别名;只要找到任何别名就会引发 ValueError 并附带相关细节信息:

  1. >>> from enum import Enum, unique
  2. >>> @unique
  3. ... class Mistake(Enum):
  4. ... ONE = 1
  5. ... TWO = 2
  6. ... THREE = 3
  7. ... FOUR = 3
  8. ...
  9. Traceback (most recent call last):
  10. ...
  11. ValueError: duplicate values found in <enum 'Mistake'>: FOUR -> THREE

使用自动设定的值

如果确切的值不重要,你可以使用 auto:

  1. >>> from enum import Enum, auto
  2. >>> class Color(Enum):
  3. ... RED = auto()
  4. ... BLUE = auto()
  5. ... GREEN = auto()
  6. ...
  7. >>> list(Color)
  8. [<Color.RED: 1>, <Color.BLUE: 2>, <Color.GREEN: 3>]

值将由 _generate_next_value_() 来选择,该函数可以被重载:

  1. >>> class AutoName(Enum):
  2. ... def _generate_next_value_(name, start, count, last_values):
  3. ... return name
  4. ...
  5. >>> class Ordinal(AutoName):
  6. ... NORTH = auto()
  7. ... SOUTH = auto()
  8. ... EAST = auto()
  9. ... WEST = auto()
  10. ...
  11. >>> list(Ordinal)
  12. [<Ordinal.NORTH: 'NORTH'>, <Ordinal.SOUTH: 'SOUTH'>, <Ordinal.EAST: 'EAST'>, <Ordinal.WEST: 'WEST'>]

注解

默认 _generate_next_value_() 方法的目标是提供所给出的最后一个 int 所在序列的下一个 int,但这种行为方式属于实现细节并且可能发生改变。

注解

_generate_next_value_() 方法定义必须在任何其他成员之前。

迭代

对枚举成员的迭代不会给出别名:

  1. >>> list(Shape)
  2. [<Shape.SQUARE: 2>, <Shape.DIAMOND: 1>, <Shape.CIRCLE: 3>]

特殊属性 __members__ 是一个从名称到成员的只读有序映射。 它包含枚举中定义的所有名称,包括别名:

  1. >>> for name, member in Shape.__members__.items():
  2. ... name, member
  3. ...
  4. ('SQUARE', <Shape.SQUARE: 2>)
  5. ('DIAMOND', <Shape.DIAMOND: 1>)
  6. ('CIRCLE', <Shape.CIRCLE: 3>)
  7. ('ALIAS_FOR_SQUARE', <Shape.SQUARE: 2>)

__members__ 属性可被用于对枚举成员进行详细的程序化访问。 例如,找出所有别名:

  1. >>> [name for name, member in Shape.__members__.items() if member.name != name]
  2. ['ALIAS_FOR_SQUARE']

比较运算

枚举成员是按标识号进行比较的:

  1. >>> Color.RED is Color.RED
  2. True
  3. >>> Color.RED is Color.BLUE
  4. False
  5. >>> Color.RED is not Color.BLUE
  6. True

枚举值之间的排序比较 不被 支持。 Enum 成员不属于整数 (另请参阅下文的 IntEnum):

  1. >>> Color.RED < Color.BLUE
  2. Traceback (most recent call last):
  3. File "<stdin>", line 1, in <module>
  4. TypeError: '<' not supported between instances of 'Color' and 'Color'

相等比较的定义如下:

  1. >>> Color.BLUE == Color.RED
  2. False
  3. >>> Color.BLUE != Color.RED
  4. True
  5. >>> Color.BLUE == Color.BLUE
  6. True

与非枚举值的比较将总是不相等(同样地,IntEnum 被显式设计成不同的行为,参见下文):

  1. >>> Color.BLUE == 2
  2. False

允许的枚举成员和属性

以上示例使用整数作为枚举值。 使用整数相当简洁方便(并由 Functional API 默认提供),但并不强制要求使用。 在大部分用例中,开发者都关心枚举的实际值是什么。 但如果值 确实 重要,则枚举可以使用任意的值。

枚举属于 Python 的类,并可具有普通方法和特殊方法。 如果我们有这样一个枚举:

  1. >>> class Mood(Enum):
  2. ... FUNKY = 1
  3. ... HAPPY = 3
  4. ...
  5. ... def describe(self):
  6. ... # self is the member here
  7. ... return self.name, self.value
  8. ...
  9. ... def __str__(self):
  10. ... return 'my custom str! {0}'.format(self.value)
  11. ...
  12. ... @classmethod
  13. ... def favorite_mood(cls):
  14. ... # cls here is the enumeration
  15. ... return cls.HAPPY
  16. ...

那么:

  1. >>> Mood.favorite_mood()
  2. <Mood.HAPPY: 3>
  3. >>> Mood.HAPPY.describe()
  4. ('HAPPY', 3)
  5. >>> str(Mood.FUNKY)
  6. 'my custom str! 1'

对于允许内容的规则如下:以单下划线开头和结尾的名称是由枚举保留而不可使用;在枚举中定义的所有其他属性将成为该枚举的成员,例外项则包括特殊方法成员 (__str__(), __add__() 等),描述符 (方法也属于描述符) 以及在 _ignore_ 中列出的变量名。

注意:如果你的枚举定义了 __new__() 和/或 __init__() 那么给予枚举成员的任何值都会被传入这些方法。 请参阅示例 Planet

受限的 Enum 子类化

一个新的 Enum 类必须基于一个 Enum 类,至多一个实体数据类型以及出于实际需要的任意多个基于 object 的 mixin 类。 这些基类的顺序为:

  1. class EnumName([mix-in, ...,] [data-type,] base-enum):
  2. pass

另外,仅当一个枚举未定义任何成员时才允许子类化该枚举。 因此禁止这样的写法:

  1. >>> class MoreColor(Color):
  2. ... PINK = 17
  3. ...
  4. Traceback (most recent call last):
  5. ...
  6. TypeError: MoreColor: cannot extend enumeration 'Color'

但是允许这样的写法:

  1. >>> class Foo(Enum):
  2. ... def some_behavior(self):
  3. ... pass
  4. ...
  5. >>> class Bar(Foo):
  6. ... HAPPY = 1
  7. ... SAD = 2
  8. ...

允许子类化定义了成员的枚举将会导致违反类型与实例的某些重要的不可变规则。 在另一方面,允许在一组枚举之间共享某些通用行为也是有意义的。 (请参阅示例 OrderedEnum 。)

封存

枚举可以被封存与解封:

  1. >>> from test.test_enum import Fruit
  2. >>> from pickle import dumps, loads
  3. >>> Fruit.TOMATO is loads(dumps(Fruit.TOMATO))
  4. True

封存的常规限制同样适用:可封存枚举必须在模块的最高层级中定义,因为解封操作要求它们可以从该模块导入。

注解

使用 pickle 协议版本 4 可以方便地封存嵌套在其他类中的枚举。

通过在枚举类中定义 __reduce_ex__() 可以对 Enum 成员的封存/解封方式进行修改。

功能性 API

Enum 类属于可调用对象,它提供了以下功能性 API:

  1. >>> Animal = Enum('Animal', 'ANT BEE CAT DOG')
  2. >>> Animal
  3. <enum 'Animal'>
  4. >>> Animal.ANT
  5. <Animal.ANT: 1>
  6. >>> Animal.ANT.value
  7. 1
  8. >>> list(Animal)
  9. [<Animal.ANT: 1>, <Animal.BEE: 2>, <Animal.CAT: 3>, <Animal.DOG: 4>]

该 API 的主义类似于 namedtuple。 调用 Enum 的第一个参数是枚举的名称。

第二个参数是枚举成员名称的 来源。 它可以是一个用空格分隔的名称字符串、名称序列、键/值对 2 元组的序列,或者名称到值的映射(例如字典)。 最后两种选项使得可以为枚举任意赋值;其他选项会自动以从 1 开始递增的整数赋值(使用 start 形参可指定不同的起始值)。 返回值是一个派生自 Enum 的新类。 换句话说,以上对 Animal 的赋值就等价于:

  1. >>> class Animal(Enum):
  2. ... ANT = 1
  3. ... BEE = 2
  4. ... CAT = 3
  5. ... DOG = 4
  6. ...

默认以 1 而以 0 作为起始数值的原因在于 0 的布尔值为 False,但所有枚举成员都应被求值为 True

对使用功能性 API 创建的枚举执行封存可能会很麻烦,因为要使用帧堆栈的实现细节来尝试并找出枚举是在哪个模块中创建的(例如当你使用了另一个模块中的工具函数就可能失败,在 IronPython 或 Jython 上也可能无效)。 解决办法是显式地指定模块名称,如下所示:

  1. >>> Animal = Enum('Animal', 'ANT BEE CAT DOG', module=__name__)

警告

如果未提供 module,且 Enum 无法确定是哪个模块,新的 Enum 成员将不可被解封;为了让错误尽量靠近源头,封存将被禁用。

新的 pickle 协议版本 4 在某些情况下同样依赖于 __qualname__ 被设为特定位置以便 pickle 能够找到相应的类。 例如,类是否存在于全局作用域的 SomeData 类中:

  1. >>> Animal = Enum('Animal', 'ANT BEE CAT DOG', qualname='SomeData.Animal')

完整的签名为:

  1. Enum(value='NewEnumName', names=<...>, *, module='...', qualname='...', type=<mixed-in class>, start=1)

将被新 Enum 类将记录为其名称的数据。

names

Enum 的成员。 这可以是一个空格或逗号分隔的字符串 (起始值将为 1,除非另行指定):

  1. 'RED GREEN BLUE' | 'RED,GREEN,BLUE' | 'RED, GREEN, BLUE'

或是一个名称的迭代器:

  1. ['RED', 'GREEN', 'BLUE']

或是一个 (名称, 值) 对的迭代器:

  1. [('CYAN', 4), ('MAGENTA', 5), ('YELLOW', 6)]

或是一个映射:

  1. {'CHARTREUSE': 7, 'SEA_GREEN': 11, 'ROSEMARY': 42}

module — 模块

新 Enum 类所在模块的名称。

qualname

新 Enum 类在模块中的具体位置。

type — 类型

要加入新 Enum 类的类型。

start

当只传入名称时要使用的起始数值。

在 3.5 版更改: 增加了 start 形参。

派生的枚举

IntEnum

所提供的第一个变种 Enum 同时也是 int 的一个子类。 IntEnum 的成员可与整数进行比较;通过扩展,不同类型的整数枚举也可以相互进行比较:

  1. >>> from enum import IntEnum
  2. >>> class Shape(IntEnum):
  3. ... CIRCLE = 1
  4. ... SQUARE = 2
  5. ...
  6. >>> class Request(IntEnum):
  7. ... POST = 1
  8. ... GET = 2
  9. ...
  10. >>> Shape == 1
  11. False
  12. >>> Shape.CIRCLE == 1
  13. True
  14. >>> Shape.CIRCLE == Request.POST
  15. True

不过,它们仍然不可与标准 Enum 枚举进行比较:

  1. >>> class Shape(IntEnum):
  2. ... CIRCLE = 1
  3. ... SQUARE = 2
  4. ...
  5. >>> class Color(Enum):
  6. ... RED = 1
  7. ... GREEN = 2
  8. ...
  9. >>> Shape.CIRCLE == Color.RED
  10. False

IntEnum 值在其他方面的行为都如你预期的一样类似于整数:

  1. >>> int(Shape.CIRCLE)
  2. 1
  3. >>> ['a', 'b', 'c'][Shape.CIRCLE]
  4. 'b'
  5. >>> [i for i in range(Shape.SQUARE)]
  6. [0, 1]

IntFlag

所提供的下一个 Enum 的变种 IntFlag 同样是基于 int 的,不同之处在于 IntFlag 成员可使用按位运算符 (&, |, ^, ~) 进行组合且结果仍然为 IntFlag 成员。 如果,正如名称所表明的,IntFlag 成员同时也是 int 的子类,并能在任何使用 int 的场合被使用。 IntFlag 成员进行除按位运算以外的其他运算都将导致失去 IntFlag 成员资格。

3.6 新版功能.

示例 IntFlag 类:

  1. >>> from enum import IntFlag
  2. >>> class Perm(IntFlag):
  3. ... R = 4
  4. ... W = 2
  5. ... X = 1
  6. ...
  7. >>> Perm.R | Perm.W
  8. <Perm.R|W: 6>
  9. >>> Perm.R + Perm.W
  10. 6
  11. >>> RW = Perm.R | Perm.W
  12. >>> Perm.R in RW
  13. True

对于组合同样可以进行命名:

  1. >>> class Perm(IntFlag):
  2. ... R = 4
  3. ... W = 2
  4. ... X = 1
  5. ... RWX = 7
  6. >>> Perm.RWX
  7. <Perm.RWX: 7>
  8. >>> ~Perm.RWX
  9. <Perm.-8: -8>

IntFlagEnum 的另一个重要区别在于如果没有设置任何旗标(值为 0),则其布尔值为 False:

  1. >>> Perm.R & Perm.X
  2. <Perm.0: 0>
  3. >>> bool(Perm.R & Perm.X)
  4. False

由于 IntFlag 成员同时也是 int 的子类,因此它们可以相互组合:

  1. >>> Perm.X | 8
  2. <Perm.8|X: 9>

旗标

最后一个变种是 Flag。 与 IntFlag 类似,Flag 成员可使用按位运算符 (&, |, ^, ~) 进行组合,与 IntFlag 不同的是它们不可与任何其它 Flag 枚举或 int 进行组合或比较。 虽然可以直接指定值,但推荐使用 auto 作为值以便让 Flag 选择适当的值。

3.6 新版功能.

IntFlag 类似,如果 Flag 成员的某种组合导致没有设置任何旗标,则其布尔值为 False:

  1. >>> from enum import Flag, auto
  2. >>> class Color(Flag):
  3. ... RED = auto()
  4. ... BLUE = auto()
  5. ... GREEN = auto()
  6. ...
  7. >>> Color.RED & Color.GREEN
  8. <Color.0: 0>
  9. >>> bool(Color.RED & Color.GREEN)
  10. False

单个旗标的值应当为二的乘方 (1, 2, 4, 8, …),旗标的组合则无此限制:

  1. >>> class Color(Flag):
  2. ... RED = auto()
  3. ... BLUE = auto()
  4. ... GREEN = auto()
  5. ... WHITE = RED | BLUE | GREEN
  6. ...
  7. >>> Color.WHITE
  8. <Color.WHITE: 7>

对 “no flags set” 条件指定一个名称并不会改变其布尔值:

  1. >>> class Color(Flag):
  2. ... BLACK = 0
  3. ... RED = auto()
  4. ... BLUE = auto()
  5. ... GREEN = auto()
  6. ...
  7. >>> Color.BLACK
  8. <Color.BLACK: 0>
  9. >>> bool(Color.BLACK)
  10. False

注解

对于大多数新代码,强烈推荐使用 EnumFlag,因为 IntEnumIntFlag 打破了枚举的某些语义约定(例如可以同整数进行比较,并因而导致此行为被传递给其他无关的枚举)。 IntEnumIntFlag 的使用应当仅限于 EnumFlag 无法使用的场合;例如,当使用枚举替代整数常量时,或是与其他系统进行交互操作时。

其他事项

虽然 IntEnumenum 模块的一部分,但要独立实现也应该相当容易:

  1. class IntEnum(int, Enum):
  2. pass

这里演示了如何定义类似的派生枚举;例如一个混合了 str 而不是 intStrEnum

几条规则:

  1. 当子类化 Enum 时,在基类序列中的混合类型必须出现于 Enum 本身之前,如以上 IntEnum 的例子所示。

  2. 虽然 Enum 可以拥有任意类型的成员,不过一旦你混合了附加类型,则所有成员必须为相应类型的值,如在上面的例子中即为 int。 此限制不适用于仅添加方法而未指定另一数据类型的混合类。

  3. 当混合了另一数据类型时,value 属性会 不同于 枚举成员自身,但它们仍保持等价且比较结果也相等。

  4. %-style formatting: %s 和 %r 会分别调用 Enum 类的 __str__()__repr__();其他代码 (例如表示 IntEnum 的 %i 或 %h) 会将枚举成员视为对应的混合类型。

  5. 格式化字符串字面值, str.format()format() 将使用混合类型的 __format__() 除非在子类中重载了 __str__()__format__(),在这种情况下将使用被重载的方法或 Enum 的方法。 请使用 !s 和 !r 格式代码来强制使用 Enum 类的 __str__()__repr__() 方法。

何时使用 __new__()__init__()

当你想要定制 Enum 成员的实际值时必须使用 __new__()。 任何其他修改可以用 __new__() 也可以用 __init__(),应优先使用 __init__()

举例来说,如果你要向构造器传入多个条目,但只希望将其中一个作为值:

  1. >>> class Coordinate(bytes, Enum):
  2. ... """
  3. ... Coordinate with binary codes that can be indexed by the int code.
  4. ... """
  5. ... def __new__(cls, value, label, unit):
  6. ... obj = bytes.__new__(cls, [value])
  7. ... obj._value_ = value
  8. ... obj.label = label
  9. ... obj.unit = unit
  10. ... return obj
  11. ... PX = (0, 'P.X', 'km')
  12. ... PY = (1, 'P.Y', 'km')
  13. ... VX = (2, 'V.X', 'km/s')
  14. ... VY = (3, 'V.Y', 'km/s')
  15. ...
  16. >>> print(Coordinate['PY'])
  17. Coordinate.PY
  18. >>> print(Coordinate(3))
  19. Coordinate.VY

有趣的示例

虽然 Enum, IntEnum, IntFlagFlag 预期可覆盖大多数应用场景,但它们无法覆盖全部。 这里有一些不同类型枚举的方案,它们可以被直接使用,或是作为自行创建的参考示例。

省略值

在许多应用场景中人们都不关心枚举的实际值是什么。 有几个方式可以定义此种类型的简单枚举:

  • 使用 auto 的实例作为值

  • 使用 object 的实例作为值

  • 使用描述性的字符串作为值

  • 使用元组作为值并用自定义的 __new__() 以一个 int 值来替代该元组

使用以上任何一种方法均可向用户指明值并不重要,并且使人能够添加、移除或重排序成员而不必改变其余成员的数值。

无论你选择何种方法,你都应当提供一个 repr() 并且它也需要隐藏(不重要的)值:

  1. >>> class NoValue(Enum):
  2. ... def __repr__(self):
  3. ... return '<%s.%s>' % (self.__class__.__name__, self.name)
  4. ...

使用 auto

使用 auto 的形式如下:

  1. >>> class Color(NoValue):
  2. ... RED = auto()
  3. ... BLUE = auto()
  4. ... GREEN = auto()
  5. ...
  6. >>> Color.GREEN
  7. <Color.GREEN>

使用 object

使用 object 的形式如下:

  1. >>> class Color(NoValue):
  2. ... RED = object()
  3. ... GREEN = object()
  4. ... BLUE = object()
  5. ...
  6. >>> Color.GREEN
  7. <Color.GREEN>

使用描述性字符串

使用字符串作为值的形式如下:

  1. >>> class Color(NoValue):
  2. ... RED = 'stop'
  3. ... GREEN = 'go'
  4. ... BLUE = 'too fast!'
  5. ...
  6. >>> Color.GREEN
  7. <Color.GREEN>
  8. >>> Color.GREEN.value
  9. 'go'

使用自定义的 __new__()

使用自动编号 __new__() 的形式如下:

  1. >>> class AutoNumber(NoValue):
  2. ... def __new__(cls):
  3. ... value = len(cls.__members__) + 1
  4. ... obj = object.__new__(cls)
  5. ... obj._value_ = value
  6. ... return obj
  7. ...
  8. >>> class Color(AutoNumber):
  9. ... RED = ()
  10. ... GREEN = ()
  11. ... BLUE = ()
  12. ...
  13. >>> Color.GREEN
  14. <Color.GREEN>
  15. >>> Color.GREEN.value
  16. 2

要实现更通用的 AutoNumber,请添加 *args 到签名中:

  1. >>> class AutoNumber(NoValue):
  2. ... def __new__(cls, *args): # this is the only change from above
  3. ... value = len(cls.__members__) + 1
  4. ... obj = object.__new__(cls)
  5. ... obj._value_ = value
  6. ... return obj
  7. ...

这样当你从 AutoNumber 继承时你将可以编写你自己的 __init__ 来处理任何附加参数:

  1. >>> class Swatch(AutoNumber):
  2. ... def __init__(self, pantone='unknown'):
  3. ... self.pantone = pantone
  4. ... AUBURN = '3497'
  5. ... SEA_GREEN = '1246'
  6. ... BLEACHED_CORAL = () # New color, no Pantone code yet!
  7. ...
  8. >>> Swatch.SEA_GREEN
  9. <Swatch.SEA_GREEN>
  10. >>> Swatch.SEA_GREEN.pantone
  11. '1246'
  12. >>> Swatch.BLEACHED_CORAL.pantone
  13. 'unknown'

注解

如果定义了 __new__() 则它会在创建 Enum 成员期间被使用;随后它将被 Enum 的 __new__() 所替换,该方法会在类创建后被用来查找现有成员。

OrderedEnum

一个有序枚举,它不是基于 IntEnum,因此保持了正常的 Enum 不变特性(例如不可与其他枚举进行比较):

  1. >>> class OrderedEnum(Enum):
  2. ... def __ge__(self, other):
  3. ... if self.__class__ is other.__class__:
  4. ... return self.value >= other.value
  5. ... return NotImplemented
  6. ... def __gt__(self, other):
  7. ... if self.__class__ is other.__class__:
  8. ... return self.value > other.value
  9. ... return NotImplemented
  10. ... def __le__(self, other):
  11. ... if self.__class__ is other.__class__:
  12. ... return self.value <= other.value
  13. ... return NotImplemented
  14. ... def __lt__(self, other):
  15. ... if self.__class__ is other.__class__:
  16. ... return self.value < other.value
  17. ... return NotImplemented
  18. ...
  19. >>> class Grade(OrderedEnum):
  20. ... A = 5
  21. ... B = 4
  22. ... C = 3
  23. ... D = 2
  24. ... F = 1
  25. ...
  26. >>> Grade.C < Grade.A
  27. True

DuplicateFreeEnum

如果发现重复的成员名称则将引发错误而不是创建别名:

  1. >>> class DuplicateFreeEnum(Enum):
  2. ... def __init__(self, *args):
  3. ... cls = self.__class__
  4. ... if any(self.value == e.value for e in cls):
  5. ... a = self.name
  6. ... e = cls(self.value).name
  7. ... raise ValueError(
  8. ... "aliases not allowed in DuplicateFreeEnum: %r --> %r"
  9. ... % (a, e))
  10. ...
  11. >>> class Color(DuplicateFreeEnum):
  12. ... RED = 1
  13. ... GREEN = 2
  14. ... BLUE = 3
  15. ... GRENE = 2
  16. ...
  17. Traceback (most recent call last):
  18. ...
  19. ValueError: aliases not allowed in DuplicateFreeEnum: 'GRENE' --> 'GREEN'

注解

这个例子适用于子类化 Enum 来添加或改变禁用别名以及其他行为。 如果需要的改变只是禁用别名,也可以选择使用 unique() 装饰器。

Planet

如果定义了 __new__()__init__() 则枚举成员的值将被传给这些方法:

  1. >>> class Planet(Enum):
  2. ... MERCURY = (3.303e+23, 2.4397e6)
  3. ... VENUS = (4.869e+24, 6.0518e6)
  4. ... EARTH = (5.976e+24, 6.37814e6)
  5. ... MARS = (6.421e+23, 3.3972e6)
  6. ... JUPITER = (1.9e+27, 7.1492e7)
  7. ... SATURN = (5.688e+26, 6.0268e7)
  8. ... URANUS = (8.686e+25, 2.5559e7)
  9. ... NEPTUNE = (1.024e+26, 2.4746e7)
  10. ... def __init__(self, mass, radius):
  11. ... self.mass = mass # in kilograms
  12. ... self.radius = radius # in meters
  13. ... @property
  14. ... def surface_gravity(self):
  15. ... # universal gravitational constant (m3 kg-1 s-2)
  16. ... G = 6.67300E-11
  17. ... return G * self.mass / (self.radius * self.radius)
  18. ...
  19. >>> Planet.EARTH.value
  20. (5.976e+24, 6378140.0)
  21. >>> Planet.EARTH.surface_gravity
  22. 9.802652743337129

TimePeriod

一个演示如何使用 _ignore_ 属性的例子:

  1. >>> from datetime import timedelta
  2. >>> class Period(timedelta, Enum):
  3. ... "different lengths of time"
  4. ... _ignore_ = 'Period i'
  5. ... Period = vars()
  6. ... for i in range(367):
  7. ... Period['day_%d' % i] = i
  8. ...
  9. >>> list(Period)[:2]
  10. [<Period.day_0: datetime.timedelta(0)>, <Period.day_1: datetime.timedelta(days=1)>]
  11. >>> list(Period)[-2:]
  12. [<Period.day_365: datetime.timedelta(days=365)>, <Period.day_366: datetime.timedelta(days=366)>]

各种枚举有何区别?

枚举具有自定义的元类,它会影响所派生枚举类及其实例(成员)的各个方面。

枚举类

EnumMeta 元类负责提供 __contains__(), __dir__(), __iter__() 及其他方法以允许用户通过 Enum 类来完成一般类做不到的事情,例如 list(Color) 或 some_enum_var in Color。 EnumMeta 会负责确保最终 Enum 类中的各种其他方法是正确的 (例如 __new__(), __getnewargs__(), __str__()__repr__())。

枚举成员(即实例)

有关枚举成员最有趣的特点是它们都是单例对象。 EnumMeta 会在创建 Enum 类本身时将它们全部创建完成,然后准备好一个自定义的 __new__(),通过只返回现有的成员实例来确保不会再实例化新的对象。

细节要点

支持的 __dunder__ 名称

__members__ 是一个 member_name:member 条目的只读有序映射。 它只在类上可用。

如果指定了 __new__(),它必须创建并返回枚举成员;相应地设定成员的 _value_ 也是一个很好的主意。 一旦所有成员都创建完成它就不会再被使用。

支持的 _sunder_ 名称

  • _name_ — 成员的名称

  • _value_ — 成员的值;可以在 __new__ 中设置 / 修改

  • _missing_ — 当未发现某个值时所使用的查找函数;可被重载

  • _ignore_ — 一个名称列表,可以为 liststr,它不会被转化为成员,并将从最终类中被移除

  • _order_ — 用于 Python 2/3 代码以确保成员顺序一致(类属性,在类创建期间会被移除)

  • _generate_next_value_ — 用于 Functional API 并通过 auto 为枚举成员获取适当的值;可被重载

3.6 新版功能: _missing_, _order_, _generate_next_value_

3.7 新版功能: _ignore_

用来帮助 Python 2 / Python 3 代码保持同步提供 _order_ 属性。 它将与枚举的实际顺序进行对照检查,如果两者不匹配则会引发错误:

  1. >>> class Color(Enum):
  2. ... _order_ = 'RED GREEN BLUE'
  3. ... RED = 1
  4. ... BLUE = 3
  5. ... GREEN = 2
  6. ...
  7. Traceback (most recent call last):
  8. ...
  9. TypeError: member order does not match _order_

注解

在 Python 2 代码中 _order_ 属性是必须的,因为定义顺序在被记录之前就会丢失。

_Private__names

私有名称在 Python 3.10 中将成为普通属性而不再是错误或成员(具体取决于该名称是否以一个下划线结束)。 在 3.9 中使用这种名称将引发 DeprecationWarning

Enum 成员类型

Enum 成员是其 Enum 类的实例,一般通过 EnumClass.member 的形式来访问。 在特定情况下它们也可通过 EnumClass.member.member 的形式来访问,但你绝对不应这样做,因为查找可能失败,或者更糟糕地返回你所查找的 Enum 成员以外的对象(这也是成员应使用全大写名称的另一个好理由):

  1. >>> class FieldTypes(Enum):
  2. ... name = 0
  3. ... value = 1
  4. ... size = 2
  5. ...
  6. >>> FieldTypes.value.size
  7. <FieldTypes.size: 2>
  8. >>> FieldTypes.size.value
  9. 2

在 3.5 版更改.

Enum 类和成员的布尔值

混合了非 Enum 类型(例如 int, str 等)的 Enum 成员会按所混合类型的规则被求值;在其他情况下,所有成员都将被求值为 True。 要使你的自定义 Enum 的布尔值取决于成员的值,请在你的类中添加以下代码:

  1. def __bool__(self):
  2. return bool(self.value)

Enum 类总是会被求值为 True

带有方法的 Enum

如果你为你的 Enum 子类添加了额外的方法,如同上述的 Planet 类一样,这些方法将在对成员执行 dir() 时显示出来,但对类执行时则不会显示:

  1. >>> dir(Planet)
  2. ['EARTH', 'JUPITER', 'MARS', 'MERCURY', 'NEPTUNE', 'SATURN', 'URANUS', 'VENUS', '__class__', '__doc__', '__members__', '__module__']
  3. >>> dir(Planet.EARTH)
  4. ['__class__', '__doc__', '__module__', 'mass', 'name', 'radius', 'surface_gravity', 'value']

组合 Flag 的成员

如果 Flag 成员的某种组合未被命名,则 repr() 将包含所有已命名的旗标和值中所有已命名的旗标组合:

  1. >>> class Color(Flag):
  2. ... RED = auto()
  3. ... GREEN = auto()
  4. ... BLUE = auto()
  5. ... MAGENTA = RED | BLUE
  6. ... YELLOW = RED | GREEN
  7. ... CYAN = GREEN | BLUE
  8. ...
  9. >>> Color(3) # named combination
  10. <Color.YELLOW: 3>
  11. >>> Color(7) # not named combination
  12. <Color.CYAN|MAGENTA|BLUE|YELLOW|GREEN|RED: 7>