5.9 enum--支持枚举类型

enum模块提供了枚举类型的支持。枚举类型是由一个名称和一个统一值来组成，值是常量的值，它们之间可以通过名称进行比较和引用，还可以迭代访问。

5.9.1 模块内容

本模块主要定义了两种枚举类型：Enum和IntEnum。定义了一个装饰器函数unique()，以便保证所有枚举常量的唯一性。

class enum.Enum 

构造枚举常量的基类。

例子：

#python 3.4

import enum

 

class Color(enum.Enum):

    red = 1

    green = 2

    blue = 3

 

print(Color.red, repr(Color.red))

print(Color.green)

print(Color.blue)

结果输出如下：

Color.red <Color.red: 1>

Color.green

Color.blue

 

class enum.IntEnum 

构造以整数值为枚举值的基类。本类里的枚举值可以与整数相比较，普通的枚举类是不能与整数相比较的。此类主要用为了旧的代码兼容，或者大大地提高了便利性的场合，否则还是使用普通的枚举类会比较好，因为它不与普通类型相互发生作用，避免了一些编程上失误。

例子：

#python 3.4

import enum

 

class Color(enum.IntEnum):

    red = 1

    green = 2

    blue = 3

 

print(Color.red, repr(Color.red))

print(Color.green)

print(Color.blue)

print(Color.red == 1)

结果输出如下：

Color.red <Color.red: 1>

Color.green

Color.blue

True

 

enum.unique() 

本装饰器函数保证了每个名称只能绑定一个值，达到枚举值的唯一性。

例子：

#python 3.4

import enum

 

class Color(enum.IntEnum):

    red = 1

    green = 2

    blue = 3

    red = 5

    

print(Color.red, repr(Color.red))

print(Color.green)

print(Color.blue)

结果输出如下：

Traceback (most recent call last):

  File "F:/temp/pywin/enum1.py", line 4, in <module>

    class Color(enum.IntEnum):

  File "F:/temp/pywin/enum1.py", line 8, in Color

    red = 5

  File "C:\Python34\lib\enum.py", line 66, in __setitem__

    raise TypeError('Attempted to reuse key: %r' % key)

TypeError: Attempted to reuse key: 'red'

 

枚举常量可以迭代遍历：

例子：

#python 3.4

import enum

 

class Color(enum.IntEnum):

    red = 1

    green = 2

    blue = 3

 

for i in Color:

    print(i)

结果输出如下：

Color.red

Color.green

Color.blue

 

枚举常量访问方式：

例子：

#python 3.4

import enum

 

class Color(enum.IntEnum):

    red = 1

    green = 2

    blue = 3

 

print(Color(2))

print(Color['red'])

red = Color.red

print(red.name, red.value)

结果输出如下：

Color.green

Color.red

red 1

在这里提供三种访问方式：Color(2),  Color[‘red’],  red.value。

 

通过函数API方式创建枚举常量：

例子：

#python 3.4

import enum

 

Animal = enum.Enum('Animal', 'ant bee cat dog')

print(Animal)

print(list(Animal))

结果输出如下：

<enum 'Animal'>

[<Animal.ant: 1>, <Animal.bee: 2>, <Animal.cat: 3>, <Animal.dog: 4>]

 

避免每个枚举常量都需要指定值：

例子：

#python 3.4

import enum

 

class AutoNumber(enum.Enum):

     def __new__(cls):

         value = len(cls.__members__) + 1

         obj = object.__new__(cls)

         obj._value_ = value

         return obj

 

class Color(AutoNumber):

     red = ()

     green = ()

     blue = ()

for i in Color:

    print(repr(i))

结果输出如下：

<Color.red: 1>

<Color.green: 2>

<Color.blue: 3>

蔡军生  QQ:9073204  深圳