Python 的枚举类型 enum 实现

Python 的枚举类型 enum 实现

C 语言中有个枚举类型 enum，很适合在定义一系列相关的常量时，由程序自动为常量分配 id，解决人工分配 id 可能导致的 id 冲突问题。在 Python 这种脚本语言中，当定义大量相关的常量时，如果由程序员人工分配，要么容易 id 冲突，要么不好进行 id 分段管理。其实，可以很简单地实现 Python 版的 enum（当然是没有类型检测的 enum 版本）。

实现代码如下：

1. def enum(module, str_enum, sep=None):  
2.     """把用特定分隔符隔开的 str_enum 字符串实现为 module 的枚举值. 
3.     @param module 宿主对象, 可以是 module, class, ... 
4.     eg1: enum(test, "A B C") => test.A,test.B,test.C == 0,1,2 
5.     eg2: enum(CTest, 'one=1,two,three,ten=0x0A,eleven', sep=',') => 
6.         CTest.one,CTest.two,CTest.ten, CTest.eleven == 1,2,10,11 
7.     """  
8.     idx = 0  
9.     for name in str_enum.split(sep):  
10.         if '=' in name:  
11.             name,val = name.rsplit('=', 1)  
12.             if val.isalnum():  
13.                 idx = eval(val)  
14.         setattr(module, name.strip(), idx)  
15.         idx += 1